Triadic Closure-Heterogeneity-Harmony GCN for Link Prediction
Ke-ke Shang, Junfan Yi, Michael Small, Yijie Zhou
TL;DR
TriHetGCN tackles link prediction by fusing physics-inspired topology cues with graph neural networks. It builds topology-aware node features via shortest-path distances to a small set of anchor nodes and integrates triadic closure (Common Neighbors) and network heterogeneity (degree differences) into the GCN propagation with learnable weights. The model delivers state-of-the-art performance across nine real-world datasets, including those with and without node attributes, and ablation experiments reveal complementary benefits of CN and HI signals. This approach bridges statistical physics and deep learning, offering robust generalization for diverse networks and enabling effective link prediction even when node attributes are absent. The work also discusses computational considerations and future extensions to dynamic graphs.
Abstract
Link prediction aims to estimate the likelihood of connections between pairs of nodes in complex networks, which is beneficial to many applications from friend recommendation to metabolic network reconstruction. Traditional heuristic-based methodologies in the field of complex networks typically depend on predefined assumptions about node connectivity, limiting their generalizability across diverse networks. While recent graph neural network (GNN) approaches capture global structural features effectively, they often neglect node attributes and intrinsic structural relationships between node pairs. To address this, we propose TriHetGCN, an extension of traditional Graph Convolutional Networks (GCNs) that incorporates explicit topological indicators -- triadic closure and degree heterogeneity. TriHetGCN consists of three modules: topology feature construction, graph structural representation, and connection probability prediction. The topology feature module constructs node features using shortest path distances to anchor nodes, enhancing global structure perception. The graph structural module integrates topological indicators into the GCN framework to model triadic closure and heterogeneity. The connection probability module uses deep learning to predict links. Evaluated on nine real-world datasets, from traditional networks without node attributes to large-scale networks with rich features, TriHetGCN achieves state-of-the-art performance, outperforming mainstream methods. This highlights its strong generalization across diverse network types, offering a promising framework that bridges statistical physics and graph deep learning.
