基于频域角度改进CBAM注意力机制

作者：张俊飞; 张贵英

来源：自动化与仪器仪表, 2021, (09): 87-95.

DOI：10.14016/j.cnki.1001-9227.2021.09.087

• 离散余弦变换
• CBAM
• ResNet50

CBAM注意力机制可以无缝嵌入神经网络模型中,以提高模型性能。当前国内外学者已在众多智能研究领域引用了CBAM机制。然而,CBAM机制内部池化操作并不能充分表达信号特征。为提高CBAM机制性能,提出利用离散余弦变换频率分量代替池化操作。CBAM机制通道注意力采用各通道前k个最高离散余弦变换频率分量代替池化操作;空间注意力采用通道维度离散余弦变换所有频率分量代替池化操作。通过六组原生和改进CBAM机制对比实验可知:改进CBAM机制和ResNet50的整合后心音分类效果更加精准;前16个最高离散余弦变换频率分量能够充分表达信息通道特征。

• 单位
  现代教育技术中心; 基础医学院; 广州医科大学