首页 | 本学科首页   官方微博 | 高级检索  
     

融合注意力机制和语义关联性的多标签图像分类
引用本文:薛丽霞, 江迪, 汪荣贵, 等. 融合注意力机制和语义关联性的多标签图像分类[J]. 光电工程, 2019, 46(9): 180468. doi: 10.12086/oee.2019.180468
作者姓名:薛丽霞  江迪  汪荣贵  杨娟
作者单位:合肥工业大学计算机与信息学院,安徽 合肥 230009
摘    要:卷积神经网络在单标签图像分类中表现出了良好的性能,但是,如何将其更好地应用到多标签图像分类仍然是一项重要的挑战。本文提出一种基于卷积神经网络并融合注意力机制和语义关联性的多标签图像分类方法。首先,利用卷积神经网络来提取特征;其次,利用注意力机制将数据集中的每个标签类别和输出特征图中的每个通道进行对应;最后,利用监督学习的方式学习通道之间的关联性,也就是学习标签之间的关联性。实验结果表明,本文方法可以有效地学习标签之间语义关联性,并提升多标签图像分类效果。

关 键 词:多标签图像分类   卷积神经网络   注意力机制   语义关联性
收稿时间:2018-09-01
修稿时间:2018-12-25

Multi-label classification based on attention mechanism and semantic dependencies
Xue Lixia, Jiang Di, Wang Ronggui, et al. Multi-label classification based on attention mechanism and semantic dependencies[J]. Opto-Electronic Engineering, 2019, 46(9): 180468. doi: 10.12086/oee.2019.180468
Authors:Xue Lixia  Jiang Di  Wang Ronggui  Yang Juan
Affiliation:School of Computer and Information, Hefei University of Technology, Hefei, Anhui 230009, China
Abstract:Multi-label image classification which is a generalization of the single-label image classification is aimed to assign multi-labels to the image to full express the specific visual concepts contained in the image. We propose a method based on convolutional neural networks, which combines attention mechanism and semantic relevance, to solve the multi label problem. Firstly, we use convolution neural network to extract features. Then, we apply the attention mechanism to obtain the correspondence between the label and channel of the feature map. Finally, we explore the channel-wise correlation which is essentially the semantic dependencies between labels by means of supervised learning. The experimental results show that the proposed method can exploit the dependencies between multiple tags to improve the performance of multi label image classification.
Keywords:multi-label classification  convolution neural network  attention mechanism  semantic dependencies
本文献已被 万方数据 等数据库收录!
点击此处可从《光电工程》浏览原始摘要信息
点击此处可从《光电工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号