English

Structural Imbalance Aware Graph Augmentation Learning

Machine Learning 2023-03-27 v1 Artificial Intelligence

Abstract

Graph machine learning (GML) has made great progress in node classification, link prediction, graph classification and so on. However, graphs in reality are often structurally imbalanced, that is, only a few hub nodes have a denser local structure and higher influence. The imbalance may compromise the robustness of existing GML models, especially in learning tail nodes. This paper proposes a selective graph augmentation method (SAug) to solve this problem. Firstly, a Pagerank-based sampling strategy is designed to identify hub nodes and tail nodes in the graph. Secondly, a selective augmentation strategy is proposed, which drops the noisy neighbors of hub nodes on one side, and discovers the latent neighbors and generates pseudo neighbors for tail nodes on the other side. It can also alleviate the structural imbalance between two types of nodes. Finally, a GNN model will be retrained on the augmented graph. Extensive experiments demonstrate that SAug can significantly improve the backbone GNNs and achieve superior performance to its competitors of graph augmentation methods and hub/tail aware methods.

Keywords

Cite

@article{arxiv.2303.13757,
  title  = {Structural Imbalance Aware Graph Augmentation Learning},
  author = {Zulong Liu and Kejia-Chen and Zheng Liu},
  journal= {arXiv preprint arXiv:2303.13757},
  year   = {2023}
}

Comments

13 pages, 11 figures, 7 tables

R2 v1 2026-06-28T09:31:26.820Z