Does TabPFN Understand Causal Structures?
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
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 752
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
- 2020Differentiable Causal Discovery from Interventional Datacited by 218
- 2022Amortized Inference for Causal Structure Learningcited by 90
- 2017Permutation-based Causal Inference Algorithms with Interventionscited by 78
- 2022Learning to Induce Causal Structurecited by 68
- 2025Do-PFN: In-Context Learning for Causal Effect Estimationcited by 51
- 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networkscited by 47
- 2019Scaling structural learning with NO-BEARS to infer causal transcriptome networkscited by 39
- 2025Foundation Models for Causal Inference via Prior-Data Fitted Networkscited by 32
- 2024A Meta-Learning Approach to Bayesian Causal Discoverycited by 22
3 more
- 2024Recursive Causal Discoverycited by 12
- 2025Early Stopping Tabular In-Context Learningcited by 12
- 2024FairPFN: Transformers Can do Counterfactual Fairnesscited by 8
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Noise Immunity in In-Context Tabular Learning: An Empirical Robustness Analysis of TabPFN's Attention Mechanismscited by 6
- 2026Arrow: A Foundation Model for Causal Discoverycited by 5
- 2026CDFM: Towards a General-Purpose Causal Discovery Foundation Modelcited by 2
- 2026Topological Signatures of Context-Level Reliability in TabPFNcited by 0
- 2026Characterizing the Separability Gap in Prior-Data Fitted Networkscited by 0
- 2026Characterizing the Separability Gap in Prior-Data Fitted Networks10 Oct 2026
- 2026Topological Signatures of Context-Level Reliability in TabPFN20 Jul 2026
- 2026CDFM: Towards a General-Purpose Causal Discovery Foundation Model13 Jul 2026
- 2026Arrow: A Foundation Model for Causal Discovery8 May 2026
- 2026Noise Immunity in In-Context Tabular Learning: An Empirical Robustness Analysis of TabPFN's Attention Mechanisms6 Apr 2026