Opto-Electronic Engineering, Volume. 46, Issue 9, 180468(2019)

Multi-label classification based on attention mechanism and semantic dependencies

Xue Lixia, Jiang Di, Wang Ronggui, and Yang Juan*
Author Affiliations
  • [in Chinese]
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    Xue Lixia, Jiang Di, Wang Ronggui, Yang Juan. Multi-label classification based on attention mechanism and semantic dependencies[J]. Opto-Electronic Engineering, 2019, 46(9): 180468

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Article

    Received: Sep. 10, 2018

    Accepted: --

    Published Online: Oct. 14, 2019

    The Author Email: Juan Yang (yangjuan6985@163.com)

    DOI:10.12086/oee.2019.180468

    Topics