Do causal predictors generalize better to new domains?
Vivian Y. Nastl, Moritz Hardt
Abstract
We study how well machine learning models trained on causal features generalize across domains. We consider 16 prediction tasks on tabular datasets covering applications in health, employment, education, social benefits, and politics. Each dataset comes with multiple domains, allowing us to test how well a model trained in one domain performs in another. For each prediction task, we select features that have a causal influence on the target of prediction. Our goal is to test the hypothesis that models trained on causal features generalize better across domains. Without exception, we find that predictors using all available features, regardless of causality, have better in-domain and out-of-domain accuracy than predictors using causal features. Moreover, even the absolute drop in accuracy from one domain to the other is no better for causal predictors than for models that use all features. In addition, we show that recent causal machine learning methods for domain generalization do not perform better in our evaluation than standard predictors trained on the set of causal features. Likewise, causal discovery algorithms either fail to run or select causal variables that perform no better than our selection. Extensive robustness checks confirm that our findings are stable under variable misclassification.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7ad6845c-cacf-4924-b3e6-0ad80de7cea5Cited by top-tier papers5
- TabStruct: Measuring Structural Fidelity of Tabular DataXiangjian Jiang, Nikola Simidjievski, Mateja JamnikICLR 2026 · 10 citations
- When Shift Happens - Confounding Is to BlameAbbavaram Gowtham Reddy, Celia Rubio-Madrigal, Rebekka Burkholz, Krikamol MuandetICLR 2026 · 5 citations
- TCD-Arena: Assessing Robustness of Time Series Causal Discovery Methods Against Assumption ViolationsGideon Stein, Niklas Penzel, Tristan Piater, Joachim DenzlerICLR 2026 · 1 citation
- Boosting for Predictive SufficiencyAbbavaram Gowtham Reddy, Rajeev Verma, Celia Rubio-Madrigal, Krikamol Muandet et al.ICLR 2026
- Hedging on the frontier: Learning new tasks with few samplesTobias Wegel, Federico Di Gennaro, Geelon So, Fanny YangICML 2026
Builds on16
- Revisiting Deep Learning Models for Tabular DataYury Gorishniy, Ivan Rubachev, Valentin Khrulkov, Artem BabenkoNeurIPS 2021 · 1,847 citations
- In Search of Lost Domain GeneralizationIshaan Gulrajani, David Lopez-PazICLR 2021 · 1,416 citations
- Out-of-Distribution Generalization via Risk Extrapolation (REx)David Krueger, Ethan Caballero, Jörn-Henrik Jacobsen, Amy Zhang et al.ICML 2021 · 1,163 citations
- Retiring Adult: New Datasets for Fair Machine LearningFrances Ding, Moritz Hardt, John Miller, Ludwig SchmidtNeurIPS 2021 · 671 citations
- Adversarial Domain Adaptation with Domain MixupMinghao Xu, Jian Zhang, Bingbing Ni, Teng Li et al.AAAI 2020 · 499 citations
Related papers
- Out-of-distribution Generalization with Causal Invariant TransformationsRuoyu Wang, Mingyang Yi, Zhitang Chen, Shengyu ZhuCVPR 2022 · 40 citations
- Causal Transportability for Visual RecognitionChengzhi Mao, Kevin Xia, James Wang, Hao Wang et al.CVPR 2022 · 27 citations
- Anti-causal domain generalization: Leveraging unlabeled dataSorawit Saengkyongam, Juan L. Gamella, Andrew Miller, Jonas Peters et al.ICML 2026
- FairPFN: A Tabular Foundation Model for Causal FairnessJake Robertson, Noah Hollmann, Samuel Müller, Noor H. Awad et al.ICML 2025
- Environment Agnostic Invariant Risk Minimization for Classification of Sequential DatasetsPraveen Venkateswaran, Vinod Muthusamy, Vatche Isahagian, Nalini VenkatasubramanianKDD 2021 · 17 citations
