cs.LGNov 5, 2026

AtlasULP: Domain-aware Universal Link Prediction via Relation Atlas

Authors: Yujing Liu, Yixin Liu, Yu Zheng, Lianhua Chi, Alan Wee-Chung Liew, Heng Tao Shen, Shirui Pan

Organizations: School of Information and Communication Technology, Griffith University, Australia · Department of Computer Science and Information Technology, La Trobe University, Australia · School of Computer Science and Technology, Tongji University, China

Abstract

Link prediction (LP) is a widely applied task in graph learning. Conventional LP approaches typically follow a dataset-specific paradigm, requiring independent training for each graph and incurring high computational and maintenance costs in large-scale applications. Motivated by these limitations, recent work explores Universal Link Prediction (ULP), aiming to enable training-free inference on arbitrary unseen graphs. However, existing methods primarily rely on subgraph sampling strategies to construct universal representations, which are unable to capture higher-order information due to the exponential growth of higher-order neighborhoods. As a result, they often under-perform on graphs where long-range signals are essential, such as biological networks. Moreover, their prediction models are typically domain-agnostic and lack the ability to adapt to different graph domains, leading to limited generalization. To address these challenges, we propose an Atlas-based Universal Link Prediction framework (AtlasULP), which introduces the relation atlas to capture structural associations between node pairs from both global and local perspectives, enabling a comprehensive characterization of connectivity signals across diverse graph domains. Building upon this representation, we develop a domain-guided prompting ULP model, which generates domain-aware prompt tokens from contextual structures and performs prompt-guided in-context prediction for adaptive link inference. Extensive experiments demonstrate that AtlasULP consistently outperforms state-of-the-art methods across diverse graph domains.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Instance Discrimination for Link Prediction

    May 18, 2026Valentin Cuzin-Rambaud, Mathieu Lefort, Rémy CazabetData AugmentationLink Prediction

  2. LEDA: Latent Semantic Distribution Alignment for Multi-domain Graph Pre-training

    Feb 26, 2026Lianze Shan, Jitao Zhao, Dongxiao He +3Graph Foundation ModelsCross-Domain Generalization

  3. Link prediction on multi-relational graphs from an influence propagation perspective

    Aug 5, 2026Zidu Yin, Yuankai Qi, Dong Gong +3Graph Neural NetworksLink Prediction