Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglement

📅 2026-06-18
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
Graph neural networks are prone to interference from irrelevant neighbors in decision boundary regions, leading to distorted embeddings of boundary nodes and degraded classification performance. This work addresses this structural entanglement issue—previously unexplored—and proposes Boundary Embedding Shaping (BES), a novel approach that leverages adaptive contrastive learning and a boundary-aware mechanism to selectively suppress spurious structural noise. Implemented as a lightweight plug-in module, BES achieves significant embedding refinement with minimal parameter perturbation. Experimental results demonstrate that BES substantially enhances model discriminability, yielding an average 3.3% absolute improvement over GCN on node classification tasks—with gains up to 5.0% on WikiCS—and also achieves superior accuracy in link prediction.
📝 Abstract
Graph neural networks (GNNs) excel at aggregating neighbor information for classification, yet their performance is hindered by graph structural entanglement, where spurious correlations from semantically irrelevant neighbors contaminate node embeddings. This challenge is most acute for nodes near class boundaries in the embedding space, where amplified structural noise blurs decision boundaries and destabilizes predictions. Existing robust GNN methods largely treat all nodes uniformly, ignoring boundary vulnerabilities. In this paper, to improve classification performance, we tackle graph structural disentanglement by identifying boundary-region entanglement as the primary bottleneck and propose Boundary Embedding Shaping (BES), an adaptive contrastive learning GNN plug-in module that selectively suppresses spurious structural noise at decision boundaries with minimal model parameter perturbation. Extensive experiments demonstrate that BES consistently improves boundary discrimination and outperforms existing leading methods. Notably, BES boosts GCN performance by an average of 3.3% in node classification (up to 5.0% on WikiCS) and achieves superior accuracy in link prediction.
Problem

Research questions and friction points this paper is trying to address.

graph structural entanglement
boundary nodes
spurious correlations
decision boundary
node classification
Innovation

Methods, ideas, or system contributions that make the work stand out.

Boundary Embedding Shaping
Adaptive Contrastive Learning
Graph Structural Disentanglement
Decision Boundary
Spurious Correlation
🔎 Similar Papers
No similar papers found.
💼 Related Jobs
No related jobs found.
J
Jiaqing Chen
School of Information Science and Technology, Yunnan Normal University, Kunming, China
Z
Zidu Yin
School of Information Science and Technology, Yunnan Normal University, Kunming, China
Y
Yichao Cai
Australian Institute for Machine Learning, Adelaide University, Adelaide, Australia
Yuhang Liu
Yuhang Liu
The University of Adelaide
Representation LearningLLMsLatent Variable ModelsResponsible AI
Zhen Zhang
Zhen Zhang
The University of Adelaide
CausationProbabilistic Graphical ModelsProbabilistic InferenceGraph Neural Networks
Dong Gong
Dong Gong
University of New South Wales (UNSW)
Computer VisionImage ProcessingMachine Learning
J
Javen Qinfeng Shi
Australian Institute for Machine Learning, Adelaide University, Adelaide, Australia