Learning semantic dependencies with channel correlation for multi-label classification

被引:6
|
作者
Xue, Lixia [1 ]
Jiang, Di [1 ]
Wang, Ronggui [1 ]
Yang, Juan [1 ]
Hu, Min [1 ]
机构
[1] Hefei Univ Technol, Sch Comp & Informat, Hefei 230601, Peoples R China
来源
VISUAL COMPUTER | 2020年 / 36卷 / 07期
基金
中国国家自然科学基金;
关键词
Multi-label image classification; Attention; Convolutional neural network; Label correlation; IMAGE CLASSIFICATION; GRADIENTS;
D O I
10.1007/s00371-019-01731-5
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
Multi-label image classification is a fundamental and challenging task in computer vision. Although remarkable success has been achieved by applying CNN-RNN pattern, such method has a slow convergence rate due to the existence of RNN module. Instead of utilizing the RNN modules, this paper proposes a novel channel correlation network which is fully based on convolutional neural network (CNN) to model the label correlations with high training efficiency. By creating a new attention module, the image features obtained by CNN are further convoluted to obtain the correspondence between the label and the channel-wise feature map. Then we use the SE and the convolution operation alternately to eliminate the irrelevant information to better explore the label correlation. Experiments on PASCAL VOC 2007 and MIRFlickr25k show that our model can effectively exploit the dependencies between multiple tags to achieve better performance.
引用
收藏
页码:1325 / 1335
页数:11
相关论文
共 50 条
  • [31] Correlation concept-cognitive learning model for multi-label classification
    Wu, Jiaming
    Tsang, Eric C. C.
    Xu, Weihua
    Zhang, Chengling
    Yang, Lanzhen
    [J]. KNOWLEDGE-BASED SYSTEMS, 2024, 290
  • [32] Learning common and label-specific features for multi-Label classification with correlation information
    Li, Junlong
    Li, Peipei
    Hu, Xuegang
    Yu, Kui
    [J]. PATTERN RECOGNITION, 2022, 121
  • [33] Multi-label classification with Missing Labels using Label Correlation and Robust Structural Learning
    Rastogi, Reshma
    Mortaza, Sayed
    [J]. KNOWLEDGE-BASED SYSTEMS, 2021, 229
  • [34] Multi-Label Learning Based on Transfer Learning and Label Correlation
    Yang, Kehua
    She, Chaowei
    Zhang, Wei
    Yao, Jiqing
    Long, Shaosong
    [J]. CMC-COMPUTERS MATERIALS & CONTINUA, 2019, 61 (01): : 155 - 169
  • [35] MULTI-LABEL DEEP ACTIVE LEARNING WITH LABEL CORRELATION
    Ranganathan, Hiranmayi
    Venkateswara, Hemanth
    Chakraborty, Shayok
    Panchanathan, Sethuraman
    [J]. 2018 25TH IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2018, : 3418 - 3422
  • [36] Joint learning of multi-label classification and label correlations
    He, Zhi-Fen
    Yang, Ming
    Liu, Hui-Dong
    [J]. Ruan Jian Xue Bao/Journal of Software, 2014, 25 (09): : 1967 - 1981
  • [37] Multi-Label Learning with Global and Local Label Correlation
    Zhu, Yue
    Kwok, James T.
    Zhou, Zhi-Hua
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2018, 30 (06) : 1081 - 1094
  • [38] Learning Label Specific Features for Multi-Label Classification
    Huang, Jun
    Li, Guorong
    Huang, Qingming
    Wu, Xindong
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2015, : 181 - 190
  • [39] Enhancing Label Correlation Feedback in Multi-Label Text Classification via Multi-Task Learning
    Zhang, Ximing
    Zhang, Qian-Wen
    Yan, Zhao
    Liu, Ruifang
    Cao, Yunbo
    [J]. FINDINGS OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS, ACL-IJCNLP 2021, 2021, : 1190 - 1200
  • [40] Integrating Label Semantic Similarity Scores into Multi-label Text Classification
    Chen, Zihao
    Liu, Yang
    Cheng, Baitai
    Peng, Jing
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2022, PT II, 2022, 13530 : 234 - 245