MUG: Meta-path-aware Universal Heterogeneous Graph Pre-Training

Read full text
Authors: Association for Artificial Intelligence 2026, Zhiyong Feng, Dongxiao He, Yun Huang, Lianze Shan, Weixiong Zhang, Jitao Zhao
Cited by: 0

Abstract

Universal graph pre-training has emerged as a key paradigm in graph representation learning, offering a promising way to train encoders to learn transferable representations from unlabeled graphs and to effectively generalize across a wide range of downstream tasks. However, recent explorations in universal graph pre-training primarily focus on homogeneous graphs and it remains unexplored for heterogeneous graphs, which exhibit greater structural and semantic complexity. This heterogeneity makes it highly challenging to train a universal encoder for diverse heterogeneous graphs: (i) the diverse types with dataset-specific semantics hinder the construction of a unified representation space; (ii) the number and semantics of meta-paths vary across datasets, making encoding and aggregation patterns learned from one dataset difficult to apply to others. To address these challenges, we propose a novel Meta-path-aware Universal heterogeneous Graph pre-training (MUG) approach. Specifically, for challenge (i), MUG introduces a input unification module to bridge heterogeneous types by enriching original attributes with contextual structure information and aligning them within a unified representation space. Furthermore, for challenge (ii), MUG combines meta-path reconstruction with global scattering objective, enabling a shared encoder to capture transferable high-order relational patterns while mitigating dataset-specific biases. Extensive experiments demonstrate the effectiveness of MUG on some real datasets.

Fields of Study & Subjects

Advanced Graph Neural Networks Graph Theory and Algorithms Recommender Systems and Techniques

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN