Your Autoregressive Model Already Reveals the Causal Graph
Not on the map. Why?
Abstract (arXiv)
Autoregressive models trained via next-token prediction implicitly learn the conditional independence structure of their data-generating process. We exploit this observation to perform scalable causal discovery from a single observed sequence of discrete events -- without any task-specific retraining. Such single-stream settings arise naturally in vehicle diagnostics, manufacturing systems, and patient trajectories, yet they remain largely unsolved: the absence of repeated samples, massive event vocabularies, and long-range temporal dependencies render existing methods either inaccurate or computationally intractable. We introduce TRACE, a framework that repurposes any pretrained autoregressive model as a density estimator for conditional mutual information, the fundamental primitive for conditional independence testing. By constructing parallelized CI tests on GPUs, TRACE recovers both the sample-level time causal graph and its summary projection, scaling linearly with the vocabulary size while naturally handling delayed causal effects. Crucially, we prove that minimizing the standard cross-entropy pretraining loss directly minimizes an upper bound on the causal identification error, establishing a duality between sequence prediction and causal discovery. On nonlinear SCMs (|X| = 8000) and real-world vehicle diagnostic logs (|X| = 29100), TRACE is the first applicable method at this scale, outperforming the strongest baseline by over 20 F1 points.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 30 of its 61 references that are in the map. First posted on arXiv 1 Feb 2026.
Builds on
30 of 61 references.
References · 30
- 2017Attention Is All You Needcited by 59,932
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2017Learning Important Features Through Propagating Activation Differencescited by 1,454
- 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 745
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
- 2023Causal Reasoning and Large Language Models: Opening a New Frontier for Causalitycited by 348
- 2017Detecting causal associations in large nonlinear time series datasetscited by 240
- 2020Transformer Hawkes Processcited by 230
- 2020The elephant in the interpretability room: Why use attention as explanation when we have saliency methods?cited by 122
- 2020Amortized Causal Discovery: Learning to Infer Causal Graphs from Time-Series Datacited by 99
- 2023Can large language models build causal graphs?cited by 72
18 more
- 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 43
- 2016Uncovering Causality from Multivariate Hawkes Integrated Cumulantscited by 38
- 2020CAUSE: Learning Granger Causality from Event Sequences using Attribution Methodscited by 36
- 2023A Survey on Causal Discovery Methods for I.I.D. and Time Series Datacited by 35
- 2023Causal Interpretation of Self-Attention in Pre-Trained Transformerscited by 30
- 2015Sparse and low-rank multivariate Hawkes processescited by 19
- 2021Causal Hidden Markov Model for Time Series Disease Forecastingcited by 19
- 2020BSODA: A Bipartite Scalable Framework for Online Disease Diagnosiscited by 14
- 2023Structural Hawkes Processes for Learning Causal Structure from Discrete-Time Event Sequencescited by 14
- 2022Causal Discovery in Hawkes Processes by Minimum Description Lengthcited by 11
- 2020Estimating Causal Effects with the Neural Autoregressive Density Estimatorcited by 8
- 2024Deep Autoregressive Models as Causal Inference Enginescited by 6
- 2024Harnessing Event Sensory Data for Error Pattern Prediction in Vehicles: A Language Model Approachcited by 6
- 2024Large-Scale Targeted Cause Discovery via Learning from Simulated Datacited by 5
- 2025Towards Practical Multi-label Causal Discovery in High-Dimensional Event Sequences via One-Shot Graph Aggregationcited by 4
- 2025One-Shot Multi-Label Causal Discovery in High-Dimensional Event Sequencescited by 4
- 2023Learning Causal Alignment for Reliable Disease Diagnosiscited by 3
- 2024Hawkes Models And Their Applicationscited by 2
Built on by
Citing papers per year: 2026: 2.
2 papers.