• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2024, Vol. 46 ›› Issue (09): 1667-1674.

• Artificial Intelligence and Data Mining • Previous Articles     Next Articles

A multimodal multi-label classification method based on hypergraph

LU Bin1,2,FAN Qiang2,3,ZHOU Xiao-lei2,3,YAN Hao 2,3,WANG Fang-xiao2,3   

  1. (1.School of Software,Nanjing University of Information Science and Technology,Nanjing 210044;
    2.The Sixty-third Research Institute,National University of Defense Technology,Nanjing 210007;
    3.Laboratory for Big Data and Decision,National University of Defense Technology,Changsha 410073,China)
  • Received:2023-09-01 Revised:2023-10-17 Accepted:2024-09-25 Online:2024-09-25 Published:2024-09-23

Abstract: Label classification aims to select the most relevant subset of labels from a set of labels to tag an instance, which has become a hot issue in the field of artificial intelligence. Traditional multi- label learning methods mainly focus on identifying single-modal data, with limited research on mining high-order  correlation between multi-modal data. To address the issue of insufficient representation of high-order correlations between multi-modal data in multi-label scenarios, this paper proposed a multi-modal multi-label classification method based on hypergraphs. The hypergraph model is introduced to model the high-order correlations of multi-modal data, and the fusion of multi-modal features and hyperedge convolution operation are utilized to achieve the mining of multi-modal data relationships and feature recognition, thus improving the performance of multi-modal multi-label classification. Experiments were conducted on the movie genre classification task, and the proposed method was compared with traditional methods. The experimental results show that the proposed method outperforms the comparison methods in terms of accuracy, precision, and F1 score, demonstrating the effectiveness of the method.

Key words: multi-label learning, data correlation, hypergraph, multi-modal