Login / Signup

Finding core labels for maximizing generalization of graph neural networks.

Sichao FuXueqi MaYibing ZhanFanyu YouQinmu PengTongliang LiuJames BaileyDanilo Mandic
Published in: Neural networks : the official journal of the International Neural Network Society (2024)
Graph neural networks (GNNs) have become a popular approach for semi-supervised graph representation learning. GNNs research has generally focused on improving methodological details, whereas less attention has been paid to exploring the importance of labeling the data. However, for semi-supervised learning, the quality of training data is vital. In this paper, we first introduce and elaborate on the problem of training data selection for GNNs. More specifically, focusing on node classification, we aim to select representative nodes from a graph used to train GNNs to achieve the best performance. To solve this problem, we are inspired by the popular lottery ticket hypothesis, typically used for sparse architectures, and we propose the following subset hypothesis for graph data: "There exists a core subset when selecting a fixed-size dataset from the dense training dataset, that can represent the properties of the dataset, and GNNs trained on this core subset can achieve a better graph representation". Equipped with this subset hypothesis, we present an efficient algorithm to identify the core data in the graph for GNNs. Extensive experiments demonstrate that the selected data (as a training set) can obtain performance improvements across various datasets and GNNs architectures.
Keyphrases
  • neural network
  • electronic health record
  • machine learning
  • big data
  • convolutional neural network
  • squamous cell carcinoma
  • lymph node
  • radiation therapy
  • artificial intelligence
  • early stage
  • body composition
  • high speed