English

Enhancing Transformer with GNN Structural Knowledge via Distillation: A Novel Approach

Machine Learning 2025-03-05 v1 Artificial Intelligence

Abstract

Integrating the structural inductive biases of Graph Neural Networks (GNNs) with the global contextual modeling capabilities of Transformers represents a pivotal challenge in graph representation learning. While GNNs excel at capturing localized topological patterns through message-passing mechanisms, their inherent limitations in modeling long-range dependencies and parallelizability hinder their deployment in large-scale scenarios. Conversely, Transformers leverage self-attention mechanisms to achieve global receptive fields but struggle to inherit the intrinsic graph structural priors of GNNs. This paper proposes a novel knowledge distillation framework that systematically transfers multiscale structural knowledge from GNN teacher models to Transformer student models, offering a new perspective on addressing the critical challenges in cross-architectural distillation. The framework effectively bridges the architectural gap between GNNs and Transformers through micro-macro distillation losses and multiscale feature alignment. This work establishes a new paradigm for inheriting graph structural biases in Transformer architectures, with broad application prospects.

Keywords

Cite

@article{arxiv.2503.01888,
  title  = {Enhancing Transformer with GNN Structural Knowledge via Distillation: A Novel Approach},
  author = {Zhihua Duan and Jialin Wang},
  journal= {arXiv preprint arXiv:2503.01888},
  year   = {2025}
}
R2 v1 2026-06-28T22:05:13.544Z