TAG-DLM: Diffusion Language Models for Text-Attributed Graph Learning
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Text-attributed graphs (TAGs), where each node carries a natural language description, require models to jointly reason over text and graph topology. Existing approaches often handle the two modalities separately: graph neural networks operate on shallow text features, while hybrids of LLMs and graphs use the language model mainly as a text encoder and delegate structure learning to a separate graph module. We propose method that unifies textual reasoning and graph message passing within a masked diffusion language model, a language model with bidirectional attention and generative decoding. For each graph instance, method linearises a sampled local neighbourhood into a token sequence and injects graph structure through a topology attention mask, which realises message passing over the graph. Because the diffusion language model can both interpret and generate text, the method adapts to different tasks simply by changing the prompt, supporting node classification, link prediction, and cross-dataset transfer with no target-specific fine-tuning. Experiments show that method outperforms graph neural networks, graph transformers, and LLM-based baselines on all three TAG benchmarks across two tasks, improving over the strongest baseline by up to 3.9 points.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 13 of its 20 references that are in the map. First posted on arXiv 30 Jun 2026.
Builds on
13 of 20 references.
References · 13
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2020A Generalization of Transformer Networks to Graphscited by 661
- 2023NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classificationcited by 210
- 2024LLaGA: Large Language and Graph Assistantcited by 181
- 2023SGFormer: Simplifying and Empowering Transformers for Large-Graph Representationscited by 133
- 2023DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusioncited by 87
- 2018Rao-Blackwellized Stochastic Gradients for Discrete Distributionscited by 24
- 2024Can LLMs Convert Graphs to Text-Attributed Graphs?cited by 19
1 more
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Distilling Large Language Models for Text-Attributed Graph Learning
- 2026Teaching LLMs to See Graphs: Unifying Text and Structural Reasoning
- 2025Large Language Models Meet Text-Attributed Graphs: A Survey of Integration Frameworks and Applications
- 2024Verbalized Graph Representation Learning: A Fully Interpretable Graph Model Based on Large Language Models Throughout the Entire Process
- 2025SSTAG: Structure-Aware Self-Supervised Learning Method for Text-Attributed Graphs
- 2026What Gets Unmasked First? Trajectory Analysis of Diffusion Models for Graph-to-Text Generation
- 2026Improving LLM Reasoning with Homophily-aware Structural and Semantic Text-Attributed Graph Compression
- 2025Quantizing Text-attributed Graphs for Semantic-Structural Integration