融合注意力机制和语义关联性的多标签图像分类

作者：薛丽霞; 江迪; 汪荣贵; 杨娟^*

来源：光电工程, 2019, 46(09): 22-30.

• 多标签图像分类
• 卷积神经网络
• 注意力机制
• 语义关联性

卷积神经网络在单标签图像分类中表现出了良好的性能,但是,如何将其更好地应用到多标签图像分类仍然是一项重要的挑战。本文提出一种基于卷积神经网络并融合注意力机制和语义关联性的多标签图像分类方法。首先,利用卷积神经网络来提取特征;其次,利用注意力机制将数据集中的每个标签类别和输出特征图中的每个通道进行对应;最后,利用监督学习的方式学习通道之间的关联性,也就是学习标签之间的关联性。实验结果表明,本文方法可以有效地学习标签之间语义关联性,并提升多标签图像分类效果。

• 单位
  合肥工业大学