Graph Open-Set Recognition Via Entropy Message Passing

Lina Yang,Bin Lu,Xiaoying Gan
DOI: https://doi.org/10.1109/icdm58522.2023.00193
2023-01-01
Abstract: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 state-of-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%.
What problem does this paper attempt to address?