mlsanityData to 14 Oct 2026

Transformer Is Inherently a Causal Learner

arXiv 2601.05647 ↗First posted 9 Jan 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

We reveal that transformers trained in an autoregressive manner naturally encode time-delayed causal structures in their learned representations. When predicting future values in multivariate time series, the gradient sensitivities of transformer outputs with respect to past inputs directly recover the underlying causal graph, without any explicit causal objectives or structural constraints. We prove this connection theoretically under standard identifiability conditions and develop a practical extraction method using aggregated gradient attributions. On challenging cases such as nonlinear dynamics, long-term dependencies, and non-stationary systems, this approach greatly surpasses the performance of state-of-the-art discovery algorithms, especially as data heterogeneity increases, exhibiting scaling potential where causal accuracy improves with data volume and heterogeneity, a property traditional methods lack. This unifying view lays the groundwork for a future paradigm where causal discovery operates through the lens of foundation models, and foundation models gain interpretability and enhancement through the lens of causality.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 58 of its 89 references that are in the map. First posted on arXiv 9 Jan 2026.

Builds on

58 of 89 references.

References · 58

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2023Visual Instruction Tuningcited by 7,869
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2022Flow Matching for Generative Modelingcited by 5,702
  6. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  7. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  8. 2015Variational Inference with Normalizing Flowscited by 2,170
  9. 2021Is Space-Time Attention All You Need for Video Understanding?cited by 1,620
  10. 2021ViViT: A Video Vision Transformercited by 1,620
  11. 2015Show, Attend and Tell: Neural Image Caption Generation with Visual Attentioncited by 1,423
  12. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
38 more
  1. 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,323
  2. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  3. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,085
  4. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
  5. 2019Attention is not Explanationcited by 929
  6. 2020Quantifying Attention Flow in Transformerscited by 775
  7. 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 752
  8. 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
  9. 2019Variational Autoencoders and Nonlinear ICA: A Unifying Frameworkcited by 503
  10. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 407
  11. 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 349
  12. 2016From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classificationcited by 341
  13. 2017How to Escape Saddle Points Efficientlycited by 285
  14. 2013Causal Discovery with Continuous Additive Noise Modelscited by 255
  15. 2024Foundation Models for Time Series Analysis: A Tutorial and Surveycited by 239
  16. 2016Deep Learning without Poor Local Minimacited by 234
  17. 2016Understanding Neural Networks through Representation Erasurecited by 224
  18. 2020DYNOTEARS: Structure Learning from Time-Series Datacited by 200
  19. 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
  20. 2025Chronos-2: From Univariate to Universal Forecastingcited by 194
  21. 2020On the Role of Sparsity and DAG Constraints for Learning Linear DAGscited by 191
  22. 2019Causal Discovery from Heterogeneous/Nonstationary Data with Independent Changescited by 166
  23. 2018Pathologies of Neural Models Make Interpretations Difficultcited by 140
  24. 2020Discovering contemporaneous and lagged causal relations in autocorrelated nonlinear time series datasetscited by 137
  25. 2020The elephant in the interpretability room: Why use attention as explanation when we have saliency methods?cited by 122
  26. 2024AttnLRP: Attention-Aware Layer-Wise Relevance Propagation for Transformerscited by 120
  27. 2020High-recall causal discovery for autocorrelated time series with latent confounderscited by 84
  28. 2022Learning to Induce Causal Structurecited by 68
  29. 2022Causal Discovery in Heterogeneous Environments Under the Sparse Mechanism Shift Hypothesiscited by 55
  30. 2021On the Expressive Power of Self-Attention Matricescited by 47
  31. 2013Strong Completeness and Faithfulness in Bayesian Networkscited by 45
  32. 2023Structure Learning with Continuous Optimization: A Sober Look and Beyondcited by 36
  33. 2023CausalTime: Realistically Generated Time-series for Benchmarking of Causal Discoverycited by 36
  34. 2021NTS-NOTEARS: Learning Nonparametric DBNs With Prior Knowledgecited by 33
  35. 2020Reconstructing regime-dependent causal relationships from observational time seriescited by 28
  36. 2023Assumption violations in causal discovery and the robustness of score matchingcited by 28
  37. 2021Data Generating Process to Evaluate Causal Discovery Techniques for Time Series Datacited by 22
  38. 2024A Meta-Learning Approach to Bayesian Causal Discoverycited by 22

50 most cited of 58 All 58 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.