mlsanityData to 14 Oct 2026

Does TabPFN Understand Causal Structures?

arXiv 2511.07236 ↗First posted 10 Nov 2025cs.LGcited by 5

Not on the map. Why?

Abstract (arXiv)

Causal discovery is fundamental for multiple scientific domains, yet extracting causal information from real world data remains a significant challenge. Given the recent success on real data, we investigate whether TabPFN, a transformer-based tabular foundation model pre-trained on synthetic datasets generated from structural causal models, encodes causal information in its internal representations. We develop an adapter framework using a learnable decoder and causal tokens that extract causal signals from TabPFN's frozen embeddings and decode them into adjacency matrices for causal discovery. Our evaluations demonstrate that TabPFN's embeddings contain causal information, outperforming several traditional causal discovery algorithms, with such causal information being concentrated in mid-range layers. These findings establish a new direction for interpretable and adaptable foundation models and demonstrate the potential for leveraging pre-trained tabular models for causal discovery.

Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 15 of its 26 references that are in the map. First posted on arXiv 10 Nov 2025.

Builds on

15 of 26 references.

References · 15

  1. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  2. 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 752
  3. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
  4. 2020Differentiable Causal Discovery from Interventional Datacited by 218
  5. 2022Amortized Inference for Causal Structure Learningcited by 90
  6. 2017Permutation-based Causal Inference Algorithms with Interventionscited by 78
  7. 2022Learning to Induce Causal Structurecited by 68
  8. 2025Do-PFN: In-Context Learning for Causal Effect Estimationcited by 51
  9. 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networkscited by 47
  10. 2019Scaling structural learning with NO-BEARS to infer causal transcriptome networkscited by 39
  11. 2025Foundation Models for Causal Inference via Prior-Data Fitted Networkscited by 32
  12. 2024A Meta-Learning Approach to Bayesian Causal Discoverycited by 22
3 more
  1. 2024Recursive Causal Discoverycited by 12
  2. 2025Early Stopping Tabular In-Context Learningcited by 12
  3. 2024FairPFN: Transformers Can do Counterfactual Fairnesscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 5.

5 papers.