基于双重特征注意力的多标签图像分类模型

作者：邱凯星; 冯广^*

来源：计算机与现代化, 2023, (12): 41-47.

• 图像分类
• 多标签
• 注意力机制
• 深度学习
• 特征关联

针对目前多标签图像分类任务中存在的图像多区域特征信息提取不足、图像特征与标签语义关系构建难等问题，提出一种基于双重特征注意力的多标签图像分类模型。首先，构建图像特征注意力模块对图像信息进行全局多区域特征的注意力关联，增强图像特征提取能力；其次，通过构建联合特征注意力模块对图像特征信息和标签嵌入进行相关性表示，从而使标签与图像区域之间进行跨模态融合得到更优的映射关系。实验结果表明，该模型在VOC2007和COCO2014多标签图像分类数据集中均取得了较好的分类效果，其性能指标相比于现有算法有较大的提升，验证了该模型的有效性。

• 单位
  广东工业大学