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