Graph Open-Set Recognition via Entropy Message Passing

被引:2
作者
Yang, Lina [1 ]
Lu, Bin [1 ]
Gan, Xiaoying [1 ]
机构
[1] Shanghai Jiao Tong Univ, Shanghai, Peoples R China
来源
23RD IEEE INTERNATIONAL CONFERENCE ON DATA MINING, ICDM 2023 | 2023年
基金
国家重点研发计划;
关键词
Graph Neural Networks; Open-Set Recognition; OOD Detection; Node Classification;
D O I
10.1109/ICDM58522.2023.00193
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph Neural Networks (GNNs) have achieved great success in semi-supervised node classification. These methods usually assume the closed -set setting and classify unlabeled nodes to known classes. However, in reality, there exits some unknown classes due to biased -sampling, distribution shifts, anomaly, etc. Therefore, it is important to identify unknown classes while classifying known classes, which is defined as Graph Open -Set Recognition (GOSR). To alleviate this problem, we propose Entropy Message Passing (EMP) for GOSR, which takes into account the graph structure information when identifying unknown classes and automatically determines the discrimination threshold. To be specific, we calculate the likelihood of a node belonging to unknown class through entropy propagation. Then, we transform the threshold selection into entropy clustering to identify the unknown class nodes. Finally, we classify the remaining nodes. Experimental evaluations on six benchmark graph datasets demonstrate that our method outperforms stateof-the-art baseline methods in unknown class detection and graph open-set recognition tasks. Especially in unknown class detection, our method has achieved a significant reduction in FPR@95 ranging from 12.62% to 55.88%.
引用
收藏
页码:1469 / 1474
页数:6
相关论文
共 22 条
[1]   Towards Open Set Deep Networks [J].
Bendale, Abhijit ;
Boult, Terrance E. .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1563-1572
[2]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[3]  
Hendrycks Dan., 2016, BASELINE DETECTING M
[4]  
Huang TC, 2022, PROCEEDINGS OF THE THIRTY-FIRST INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2022, P2087
[5]  
Jang E., 2017, CATEGORICAL REPARAME
[6]  
Kendall Alex, WHAT UNCERTAINTIES W
[7]  
Kipf ThomasN., 2016, INT C LEARN REPR, DOI DOI 10.48550/ARXIV.1609.02907
[8]  
Kotz S., 2000, Extreme Value Distributions Theory and Applications
[9]  
Li ZL, 2022, ADV NEUR IN
[10]  
Liang S., 2018, INT C LEARNING REPRE