MG-GCN: Multi-Granularity Graph Convolutional Neural Network for Multi-Label Classification in Multi-Label Information System

被引:17
作者
Yu, Bin [1 ]
Xie, Hengjie [1 ]
Cai, Mingjie [2 ]
Ding, Weiping [3 ]
机构
[1] Hunan Normal Univ, Coll Informat Sci & Engn, Changsha 410081, Peoples R China
[2] Hunan Univ, Coll Math, Changsha 410082, Peoples R China
[3] Nantong Univ, Sch Informat Sci & Technol, Nantong, Peoples R China
来源
IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE | 2024年 / 8卷 / 01期
基金
中国国家自然科学基金;
关键词
Difference graph; MG-GCN; multi-label classification; multi-granularity; multi-label information system; RECOGNITION;
D O I
10.1109/TETCI.2023.3300303
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, many researchers have applied Graph Convolutional Neural Networks (GCN) to multi-label learning tasks by establishing relations among labels. However, these researchers have only used GCN as an auxiliary module and have lacked exploration of object relations, which has limited the inference capability of GCN. Applying GCN to multi-label classification in information systems poses significant challenges to address because information systems are considered as a graph with only nodes and no edges. This article proposes a multi-granularity graph convolutional neural network (MG-GCN) based on object relations to enhance the versatility of GCN and demonstrate its reasoning ability. The MG-GCN efficiently solves complex multi-label classification problems in information systems. Firstly, based on the multi-granularity thinking, the attribute set is divided into several attribute subsets based on granularity. Secondly, a graph of differences representing the relationship between objects is established under each attribute subset. Finally, based on the attention mechanism and multiple difference matrices of objects, MG-GCN is proposed and applied to multi-label classification within information systems. The effectiveness and significance of performance of MG-GCN has been verified through ranking-based indicators and hypothesis tests, respectively.
引用
收藏
页码:288 / 299
页数:12
相关论文
共 31 条
  • [1] HARAM: a Hierarchical ARAM neural network for large-scale text classification
    Benites, Fernando
    Sapozhnikova, Elena
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON DATA MINING WORKSHOP (ICDMW), 2015, : 847 - 854
  • [2] Bruna J., 2013, P INT C LEARN REPR B
  • [3] Learning Graph Convolutional Networks for Multi-Label Recognition and Applications
    Chen, Zhao-Min
    Wei, Xiu-Shen
    Wang, Peng
    Guo, Yanwen
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (06) : 6969 - 6983
  • [4] Multi-Label Image Recognition with Graph Convolutional Networks
    Chen, Zhao-Min
    Wei, Xiu-Shen
    Wang, Peng
    Guo, Yanwen
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 5172 - 5181
  • [5] Defferrard M, 2016, ADV NEUR IN, V29
  • [6] Feng YF, 2019, AAAI CONF ARTIF INTE, P3558
  • [7] Kipf T.N., 2016, P ICLR
  • [8] Influence maximization in social networks using graph embedding and graph neural network
    Kumar, Sanjay
    Mallik, Abhishek
    Khetarpal, Anavi
    Panda, B. S.
    [J]. INFORMATION SCIENCES, 2022, 607 : 1617 - 1636
  • [9] General Multi-label Image Classification with Transformers
    Lanchantin, Jack
    Wang, Tianlu
    Ordonez, Vicente
    Qi, Yanjun
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 16473 - 16483
  • [10] Learning label correlations for multi-label image recognition with graph networks
    Li, Qing
    Peng, Xiaojiang
    Qiao, Yu
    Peng, Qiang
    [J]. PATTERN RECOGNITION LETTERS, 2020, 138 : 378 - 384