Graph Learning Approaches for Graph with Noise: Application to Disease Prediction in Population Graph

Lang Chen,Yangmin Huang,Bin Liao,Kun Nie,Shoubin Dong,Jinlong Hu
DOI: https://doi.org/10.1109/bibm49941.2020.9313348
2020-01-01
Abstract:Graph neural networks have been developed for various node classification tasks in graph data. However, the noise in graph data would affect the effectiveness of training and prediction of these graph learning models. In this paper, we propose a graph learning approach, named GCN_CL, which introduces a confident learning method into the graph convolutional networks model to classify nodes in graph data with label noise. The proposed approach includes node classification module and confident learning module, where the confident learning module selects clean nodes with high confident labels for node classification module to train an accurate model in the graphs with label noise. Pseudo label method is further applied on unlabeled data to increase samples for confident learning and improve the classification performance of subsequent node classification module. We evaluated classification performance of GCN_CL and compared our approach against other models to classify autism spectrum disorders patients versus health controls in population graph, which is constructed from ABIDE I dataset. The experimental results show GCN_CL achieves achieves the best performance in the population graph with different artificial noise levels.
What problem does this paper attempt to address?