When and Why LLM Causal Priors Help: Closed-Loop Prior Selection for Amortized Causal Inference
Not on the map. Why?
Abstract (arXiv)
Causal effect estimation asks how an outcome would change under an intervention, and medicine, economics, and public policy all treat it as a foundational task. Prior-data fitted networks (PFNs) amortize the task: a model trained on large numbers of programmatically generated synthetic causal tasks reads a new problem's observational data into context and returns an interventional-effect estimate in a single forward pass. The capability of such models is largely determined by the synthetic training prior, which is currently designed by hand, a bottleneck acknowledged by both Do-PFN and CausalPFN. Large language models (LLMs) can now ``draw'' plausible causal graphs for a given domain, suggesting that LLM-distilled graphs could serve as prior material. Whether injecting such graphs helps at all, where any gain comes from, and when injection helps. Practice has so far relied on manual trial and error. We propose a closed-loop prior selection framework that casts prior injection as a budget-constrained optimization over a candidate prior pool. Candidates undergo cheap post-training and are scored by a composite metric dominated by real-domain generalization; the winner then receives full training and paired statistical validation. On a 7.34M-parameter Do-PFN, the framework's winner attains a formally significant 2.75× gain on the primary evaluation domain, and its error falls below that of the uninjected official base. Generalization on an adjacent monitoring domain improves significantly, and no monitored capability degrades. Mechanism experiments show that the gain depends on the semantic content of the distilled graph rather than its structural diversity alone does not produce it (directional evidence). With this framework and this regularity in hand, the use of LLM causal priors stops being manual trial and error and becomes an empirically verifiable selection problem.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 20 references that are in the map. First posted on arXiv 7 Sep 2026.
Builds on
11 of 20 references.
References · 11
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 699
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2021Transformers Can Do Bayesian Inferencecited by 360
- 2023Causal Reasoning and Large Language Models: Opening a New Frontier for Causalitycited by 353
- 2024V-STaR: Training Verifiers for Self-Taught Reasonerscited by 217
- 2025Do-PFN: In-Context Learning for Causal Effect Estimationcited by 51
- 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 49
- 2025Pushing Test-Time Scaling Limits of Deep Search with Asymmetric Verificationcited by 12
- 2026AREX: Towards a Recursively Self-Improving Agent for Deep Researchcited by 5
Built on by
Not cited yet.