Training a Label-Noise-Resistant GNN with Reduced Complexity
Zhao, Rui, Shi, Bin, Liang, Zhiming, Ruan, Jianfei, Dong, Bo, Lin, Lu
–arXiv.org Artificial Intelligence
--Graph Neural Networks (GNNs) have been widely employed for semi-supervised node classification tasks on graphs. However, the performance of GNNs is significantly affected by label noise, that is, a small amount of incorrectly labeled nodes can substantially misguide model training. Mainstream solutions define node classification with label noise (NCLN) as a reliable labeling task, often introducing node similarity with quadratic computational complexity to more accurately assess label reliability. T o this end, in this paper, we introduce the Label Ensemble Graph Neural Network (LEGNN), a lower complexity method for robust GNNs training against label noise. Specifically, LEGNN conducts a two-step process: bootstrapping neighboring contexts and robust learning with gathered multiple labels. In the former step, we apply random neighbor masks for each node and gather the predicted labels as a high-probability label set. This mitigates the impact of inaccurately labeled neighbors and diversifies the label set. In the latter step, we utilize a partial label learning based strategy to aggregate the high-probability label information for model training. Additionally, we symmetrically gather a low-probability label set to counteract potential noise from the bootstrapped high-probability label set. Extensive experiments on six datasets demonstrate that LEGNN achieves outstanding performance while ensuring efficiency. Moreover, it exhibits good scalability on dataset with over one hundred thousand nodes and one million edges. Graph, as a ubiquitous data form in real world, can represent a variety of relational structures, such as social networks [1, 2], transportation systems [3, 4], taxpayer network [5, 6] and many more. Graph Neural Networks (GNNs) have emerged as powerful tools for capturing and analyzing graph-structured data, gaining significant attention in recent years [7, 8, 3, 9, 10, 11, 12, 13, 14, 15]. Figure 1: A comparison between our proposed label ensemble method and existing reliable labeling methods. The colors of the nodes represent their labels, while the symbols " " and " " indicate whether the nodes are correctly or incorrectly labeled, respectively.
arXiv.org Artificial Intelligence
Nov-17-2024
- Country:
- North America > United States (0.04)
- Asia > China
- Shaanxi Province > Xi'an (0.04)
- Genre:
- Research Report > New Finding (0.67)
- Industry:
- Information Technology (0.48)
- Transportation > Infrastructure & Services (0.34)
- Technology: