LoyalDE: Improving the performance of Graph Neural Networks with loyal node discovery and emphasis.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Recent years have witnessed an increasing focus on graph-based semi-supervised learning with Graph Neural Networks (GNNs). Despite existing GNNs having achieved remarkable accuracy, research on the quality of graph supervision information has inadvertently been ignored. In fact, there are significant differences in the quality of supervision information provided by different labeled nodes, and treating supervision information with different qualities equally may lead to sub-optimal performance of GNNs. We refer to this as the graph supervision loyalty problem, which is a new perspective for improving the performance of GNNs. In this paper, we devise FT-Score to quantify node loyalty by considering both the local feature similarity and the local topology similarity, and nodes with higher loyalty are more likely to provide higher-quality supervision. Based on this, we propose LoyalDE (Loyal Node Discovery and Emphasis), a model-agnostic hot-plugging training strategy, which can discover potential nodes with high loyalty to expand the training set, and then emphasize nodes with high loyalty during model training to improve performance. Experiments demonstrate that the graph supervision loyalty problem will fail most existing GNNs. In contrast, LoyalDE brings about at most 9.1% performance improvement to vanilla GNNs and consistently outperforms several state-of-the-art training strategies for semi-supervised node classification.

Authors

  • Haotong Wei
    Shandong University, School of Mechanical, Electrical and Information Engineering, Weihai, 264209, China. Electronic address: weihaotong@mail.sdu.edu.cn.
  • Yinlin Zhu
    Sun Yat-sen University, School of Computer Science and Engineering, Guangzhou, 510275, China.
  • Xunkai Li
    Beijing Institute of Technology, School of Computer Science and Technology, Beijing, 100081, China.
  • Bin Jiang
    Department of Urology, Chinese People's Liberation Army General Hospital, Beijing, 100039 China.