mlsanityData to 14 Oct 2026

Attention Dispersion in Dynamic Graph Transformers: Diagnosis and a Transferable Fix

arXiv 2605.16112 ↗First posted 15 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformer-based architectures have become the dominant paradigm for Continuous-Time Dynamic Graph (CTDG) learning, yet their performance remains limited on temporally shifted datasets. In this work, we identify attention dispersion as a shared failure mode of dynamic graph Transformers under temporal distribution shift. Through controlled ablation contrasting structurally and temporally distinguished historical neighbors against random ones, we show that prediction depends on a class of critical nodes that carry consistently more predictive signal than arbitrary neighbors. However, existing Transformers fail to focus on these nodes even when they are present in the input, as temporal shift weakens attention contrast and produces overly dispersed attention distributions. This diagnosis suggests a simple and transferable fix: replace standard attention with differential attention, which suppresses common-mode attention and amplifies distinctive token-level signals. When added to three representative CTDG Transformer baselines, differential attention consistently improves performance, with gains concentrated on high-shift datasets. Attention-level measurements further confirm the mechanism, showing reduced attention entropy and increased attention mass on critical nodes. Building on these findings, we introduce DiffDyG, a reference implementation combining differential attention with standard input encodings. Across 9 benchmarks and three negative sampling protocols, DiffDyG achieves SOTA performance, with especially large gains on the most shifted datasets.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 33 references that are in the map. First posted on arXiv 15 May 2026.

Builds on

26 of 33 references.

References · 26

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  4. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
  5. 2020Temporal Graph Networks for Deep Learning on Dynamic Graphscited by 530
  6. 2020Inductive Representation Learning on Temporal Graphscited by 462
  7. 2019Predicting Dynamic Embedding Trajectory in Temporal Interaction Networkscited by 448
  8. 2019EvolveGCN: Evolving Graph Convolutional Networks for Dynamic Graphscited by 447
  9. 2021Inductive Representation Learning in Temporal Networks via Causal Anonymous Walkscited by 193
  10. 2023Towards Better Dynamic Graph Learning: New Architecture and Unified Librarycited by 157
  11. 2017Know-Evolve: Deep Temporal Reasoning for Dynamic Knowledge Graphscited by 149
  12. 2023Do We Really Need Complicated Model Architectures For Temporal Networks?cited by 143
14 more
  1. 2022ROLAND: Graph Learning Framework for Dynamic Graphscited by 132
  2. 2022Towards Better Evaluation for Dynamic Link Predictioncited by 122
  3. 2020APAN: Asynchronous Propagation Attention Network for Real-time Temporal Graph Embeddingcited by 78
  4. 2021TCL: Transformer-based Dynamic Graph Modelling via Contrastive Learningcited by 68
  5. 2022DGraph: A Large-Scale Financial Dataset for Graph Anomaly Detectioncited by 63
  6. 2022Provably expressive temporal graph networkscited by 56
  7. 2020Laplacian Change Point Detection for Dynamic Graphscited by 25
  8. 2024On the Feasibility of Simple Transformer for Dynamic Graph Modelingcited by 22
  9. 2024Improving Temporal Link Prediction via Temporal Walk Matrix Projectioncited by 21
  10. 2024Repeat-Aware Neighbor Sampling for Dynamic Graph Learningcited by 11
  11. 2020Graph Hawkes Neural Network for Forecasting on Temporal Knowledge Graphscited by 10
  12. 2024PRES: Toward Scalable Memory-Based Dynamic Graph Neural Networkscited by 10
  13. 2024Co-Neighbor Encoding Schema: A Light-cost Structure Encoding Method for Dynamic Link Predictioncited by 10
  14. 2025TIDFormer: Exploiting Temporal and Interactive Dynamics Makes A Great Dynamic Graph Transformercited by 5

Built on by

Not cited yet.