Partial Multi-Label Learning via Probabilistic Graph Matching Mechanism

被引:38
作者
Lyu, Gengyu [1 ]
Feng, Songhe [1 ]
Li, Yidong [2 ]
机构
[1] Beijing Jiaotong Univ, Beijing Key Lab Traff Data Anal & Min, Beijing 100044, Peoples R China
[2] Beijing Jiaotong Univ, Sch Comp & Informat Technol, Beijing 100044, Peoples R China
来源
KDD '20: PROCEEDINGS OF THE 26TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING | 2020年
基金
中国国家自然科学基金; 北京市自然科学基金;
关键词
partial multi-label learning; 'instance-to-label' matching; matching selection; graph matching; 'many-to-many' constraint;
D O I
10.1145/3394486.3403053
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Partial Multi-Label learning (PML) learns from the ambiguous data where each instance is associated with a candidate label set, where only a part is correct. The key to solve such problem is to disam-biguate the candidate label sets and identify the correct assignments between instances and their ground-truth labels. In this paper, we interpret such assignments as instance-to-label matchings, and formulate the task of PML as a matching selection problem. To model such problem, we propose a novel grapH mAtching based partial muLti-label lEarning (HALE) framework, where Graph Matching scheme is incorporated owing to its good performance of exploiting the instance and label relationship. Meanwhile, since conventional one-to-one graph matching algorithm does not satisfy the constraint of PML problem that multiple instances may correspond to multiple labels, we extend the traditional probabilistic graph matching algorithm from one-to-one constraint to many-to-many constraint, and make the proposed framework to accommodate to the PML problem. Moreover, to improve the performance of predictive model, both the minimum error reconstruction and k-nearest-neighbor weight voting scheme are employed to assign more accurate labels for unseen instances. Extensive experiments on various data sets demonstrate the superiority of our proposed method.
引用
收藏
页码:105 / 113
页数:9
相关论文
共 37 条
[1]   Learning multi-label scene classification [J].
Boutell, MR ;
Luo, JB ;
Shen, XP ;
Brown, CM .
PATTERN RECOGNITION, 2004, 37 (09) :1757-1771
[2]   Online multi-label dependency topic models for text classification [J].
Burkhardt, Sophie ;
Kramer, Stefan .
MACHINE LEARNING, 2018, 107 (05) :859-886
[3]   Spectral Symmetry Analysis [J].
Chertok, Michael ;
Keller, Yosi .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2010, 32 (07) :1227-1238
[4]  
Cour T, 2011, J MACH LEARN RES, V12, P1501
[5]  
Demsar J, 2006, J MACH LEARN RES, V7, P1
[6]   A Probabilistic Approach to Spectral Graph Matching [J].
Egozi, Amir ;
Keller, Yosi ;
Guterman, Hugo .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (01) :18-27
[7]  
Elisseeff A, 2002, ADV NEUR IN, V14, P681
[8]  
Fang J., 2019, 33 AAAI C ART INT, P3518
[9]  
Feng L, 2019, AAAI CONF ARTIF INTE, P3542
[10]  
Feng Lei, 2018, INT JOINT C ART INT, P2107