Skip to main navigation Skip to search Skip to main content

Adversarially regularized graph attention networks for inductive learning on partially labeled graphs

  • Jiaren Xiao
  • , Quanyu Dai
  • , Xiaochen Xie
  • , James Lam
  • , Ka Wai Kwok*
  • *Corresponding author for this work
  • The University of Hong Kong
  • Hong Kong Polytechnic University
  • Harbin Institute of Technology Shenzhen
  • Guangdong Key Laboratory of Intelligent Morphing Mechanisms and Adaptive Robotics

Research output: Contribution to journalArticlepeer-review

Abstract

The high cost of data labeling often results in node label shortage in real applications. To improve node classification accuracy, graph-based semi-supervised learning leverages the ample unlabeled nodes to train together with the scarce available labeled nodes. However, most existing methods require the information of all nodes, including those to be predicted, during model training, which is not practical for dynamic graphs with newly added nodes. To address this issue, an adversarially regularized graph attention model is proposed to classify newly added nodes in a partially labeled graph. An attention-based aggregator is designed to generate the representation of a node by aggregating information from its neighboring nodes, thus naturally generalizing to previously unseen nodes. In addition, adversarial training is employed to improve the model's robustness and generalization ability by enforcing node representations to match a prior distribution. Experiments on real-world datasets demonstrate the effectiveness of the proposed method in comparison with the state-of-the-art methods. The code is available at https://github.com/JiarenX/AGAIN.

Original languageEnglish
Article number110456
JournalKnowledge-Based Systems
Volume268
DOIs
StatePublished - 23 May 2023
Externally publishedYes

Keywords

  • Adversarial regularization
  • Attention mechanism
  • Graph neural networks
  • Graph-based semi-supervised learning
  • Inductive learning

Fingerprint

Dive into the research topics of 'Adversarially regularized graph attention networks for inductive learning on partially labeled graphs'. Together they form a unique fingerprint.

Cite this