Causal Foundation Models with Continuous Treatments
Causal inference, estimating causal effects from observational data, is a fundamental tool in many disciplines. Of particular importance across a variety of domains is the continuous treatment setting, where the variable of intervention has a continuous range. This setting is far less explored and represents a substantial shift from the binary treatment setting, with models needing to represent effects across a continuum of treatment values. In this paper, we present the first causal foundation model for the continuous treatment setting. Our model meta-learns the ability to predict causal effects across a wide variety of unseen tasks without additional training or fine-tuning. First, we d...
| Authors | Christopher Stith, Medha Barath, Vahid Balazadeh et al. |
|---|---|
| Primary category | cs.LG |
| Published | 14 May 2026 |
| Last updated at source | 13 July 2026 |
Connected in the knowledge graph
- Layer6/CCPFNModel · cited by
Models and datasets referencing this paper
- Layer6/CCPFNModel
Silytics shows public metadata from arXiv and links to the original. Information may be incomplete or out of date. Last refreshed 17 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.