Search papers, labs, and topics across Lattice.
This paper systematizes the emerging paradigm of Causal Foundation Models (CFMs), which reframe causal effect estimation as an in-context learning problem to eliminate bespoke, per-dataset causal discovery and estimator pipelines. By pretraining architectures across diverse data-generating processes, CFMs predict causal quantities like average treatment effects on entirely unseen empirical datasets via a single forward pass without parameter updates. The work provides a formal bridge between classical causal inference and large-scale meta-learning, accompanied by open-source implementations and canonical evaluation protocols.
Traditional causal inference requires hand-crafted pipelines and bespoke estimator training for every dataset, but causal foundation models can now infer treatment effects entirely zero-shot via in-context learning.
Causal inference is the practice of estimating the effect of a treatment or intervention from data. It traditionally requires a bespoke pipeline for every new problem: first proposing a causal mechanism, selecting a compatible estimator, and finally training it. Meanwhile, across diverse settings and modalities, much of machine learning has shifted to the paradigm of foundation models: networks pretrained once at scale and applied to new tasks without fine-tuning. Causal foundation models (CFMs) bring this paradigm to causal inference. CFMs are pretrained neural networks that estimate causal quantities, such as the average treatment effect, on entirely new datasets using in-context learning without requiring model updates. This work provides a practical introduction to this emerging area. We summarize the necessary background in causal inference and machine learning before discussing CFMs. Throughout, we include example code and Jupyter notebooks.
