Bridging Semantics and Structure: A Theoretically Grounded Framework for Link Prediction on TAGs
Organizations: Karlsruhe Institute of Technology (KIT), Karlsruhe, Germany · University of Zurich, Zurich, Zurich, Switzerland · University of Michigan, Ann Arbor, Michigan, USA
Abstract
Link prediction on text-attributed graphs (TAGs) is a fundamental problem with applications in knowledge discovery, recommendation and beyond. Recent progress has highlighted the promise of combining Graph Neural Networks (GNNs) with Language Models (LMs) in order to leverage the complementary structural and semantic signals. However, due to the distinct inductive biases of GNNs and LMs, the design of an optimal fusion strategy remains an open research question. In this work, we present Bridge-LP, an intuitive, simple, efficient and theoretically grounded framework that unifies graph and language representations. It is built around three theoretically-justified design components: (D1) jumping connection of textual proximity, which enables robust LM training; (D2) compressed embedding, which encodes neighborhood information long exploited by classical link prediction heuristics, thereby enriching structural signals in a learnable way; and (D3) cache embedding training strategy, which reduces computational complexity. Together, these mechanisms establish an end-to-end cross-modal framework that scales to large graphs while capturing richer graph-text interactions. We conduct an extensive empirical evaluation, comparing against 40 existing models across 9 diverse datasets. Experiments show that Bridge-LP consistently outperforms state-of-the-art approaches, delivering up to a 43.06% improvement in Mean Reciprocal Rank (MRR) and setting a new benchmark for language-graph integration.
Figures & tables
| Works | ZS | FT | ||
| MLP | GCN | MLP | GCN | |
| Graphformer ( Yang et al., 2021 ) | ✓ | |||
| GAINT ( Chien et al., 2022 ) | ✓ | ✓ | ||
| SimTeG ( Duan et al., 2023 ) | ✓ | ✓ | ||
| TAPE ( He et al., 2024 ) | ✓ | |||
| LEADING ( Xue et al., 2023 ) | ✓ | |||
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Models | Cora | PubMed | Arxiv 2023 | |||
| Hits@100 | AUC | Hits@100 | AUC | Hits@100 | AUC | |
| RotatE | 56.60 6.29 | 61.57 3.88 | 38.58 2.59 | 80.89 0.40 | 34.57 0.84 | 76.73 0.44 |
| TransE | 57.00 7.06 | 64.19 3.83 | 50.04 2.63 | 85.29 0.97 | 30.22 1.96 | 73.03 1.04 |
| ComplEx | 73.59 0.90 | 75.71 1.09 | 53.87 3.09 | 89.60 0.51 | 32.17 2.38 | 73.31 0.61 |
| DistMult | 73.12 0.28 | 76.53 0.80 | 54.82 2.95 | 89.91 0.53 | 32.37 1.62 | 73.25 0.39 |
| BERT-Inf | 56.90 3.26 | 65.09 1.41 | 48.73 1.43 | 88.96 0.31 | 48.74 1.15 | 86.37 0.27 |
| Models | Cora | PubMed | Arxiv 2023 | |||
| Hits@50 | MRR | Hits@50 | MRR | Hits@50 | MRR | |
| MiniLM-GAT-CT | 54.23 4.08 | 13.74 5.21 | 29.44 2.84 | 4.26 1.75 | 13.76 2.22 | 2.62 0.55 |
| MiniLM-SAGE-CT | 63.04 9.23 | 15.09 1.35 | 45.31 3.83 | 6.70 3.61 | 49.11 4.22 | 11.48 3.41 |
| mpnet-GIN-CT | 89.01 5.54 | 29.06 7.96 | 46.82 4.22 | 11.86 3.68 | 55.11 3.70 | 18.88 5.89 |
| mpnet-GAT-CT | 74.90 9.22 | 23.16 11.10 | 35.82 3.81 | 5.36 1.69 | 19.50 1.91 | 4.49 0.91 |
| mpnet-SAGE-CT | 82.01 4.19 | 25.34 8.06 | 57.58 3.78 | 11.91 3.58 | 52.97 5.05 | 14.51 3.37 |
Text-Attributed Graph Learning
Text-attributed graph learning combines node text and graph structure to improve classification, retrieval, anomaly detection, and reasoning.
See all 66 papers