mlsanityData to 7 Oct 2026

Your Autoregressive Model Already Reveals the Causal Graph

arXiv 2602.01135 ↗First posted 1 Feb 2026cs.LGcited by 2

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  3. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  4. 2017Learning Important Features Through Propagating Activation Differencescited by 1,454
  5. 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 745
  6. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
  7. 2023Causal Reasoning and Large Language Models: Opening a New Frontier for Causalitycited by 348
  8. 2017Detecting causal associations in large nonlinear time series datasetscited by 240
  9. 2020Transformer Hawkes Processcited by 230
  10. 2020The elephant in the interpretability room: Why use attention as explanation when we have saliency methods?cited by 122
  11. 2020Amortized Causal Discovery: Learning to Infer Causal Graphs from Time-Series Datacited by 99
  12. 2023Can large language models build causal graphs?cited by 72
18 more
  1. 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 43
  2. 2016Uncovering Causality from Multivariate Hawkes Integrated Cumulantscited by 38
  3. 2020CAUSE: Learning Granger Causality from Event Sequences using Attribution Methodscited by 36
  4. 2023A Survey on Causal Discovery Methods for I.I.D. and Time Series Datacited by 35
  5. 2023Causal Interpretation of Self-Attention in Pre-Trained Transformerscited by 30
  6. 2015Sparse and low-rank multivariate Hawkes processescited by 19
  7. 2021Causal Hidden Markov Model for Time Series Disease Forecastingcited by 19
  8. 2020BSODA: A Bipartite Scalable Framework for Online Disease Diagnosiscited by 14
  9. 2023Structural Hawkes Processes for Learning Causal Structure from Discrete-Time Event Sequencescited by 14
  10. 2022Causal Discovery in Hawkes Processes by Minimum Description Lengthcited by 11
  11. 2020Estimating Causal Effects with the Neural Autoregressive Density Estimatorcited by 8
  12. 2024Deep Autoregressive Models as Causal Inference Enginescited by 6
  13. 2024Harnessing Event Sensory Data for Error Pattern Prediction in Vehicles: A Language Model Approachcited by 6
  14. 2024Large-Scale Targeted Cause Discovery via Learning from Simulated Datacited by 5
  15. 2025Towards Practical Multi-label Causal Discovery in High-Dimensional Event Sequences via One-Shot Graph Aggregationcited by 4
  16. 2025One-Shot Multi-Label Causal Discovery in High-Dimensional Event Sequencescited by 4
  17. 2023Learning Causal Alignment for Reliable Disease Diagnosiscited by 3
  18. 2024Hawkes Models And Their Applicationscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.