mlsanityData to 7 Oct 2026

Do causal predictors generalize better to new domains?

arXiv 2402.09891 ↗First posted 15 Feb 2024cs.LG stat.MLcited by 11

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 20 of its 93 references that are in the map. First posted on arXiv 15 Feb 2024.

Builds on

20 of 93 references.

References · 20

  1. 2019Invariant Risk Minimizationcited by 1,832
  2. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
  3. 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
  4. 2020In Search of Lost Domain Generalizationcited by 1,042
  5. 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
  6. 2020TabTransformer: Tabular Data Modeling Using Contextual Embeddingscited by 398
  7. 2015Return of Frustratingly Easy Domain Adaptationcited by 389
  8. 2021SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Trainingcited by 310
  9. 2020The Risks of Invariant Risk Minimizationcited by 262
  10. 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
  11. 2019Neural Oblivious Decision Ensembles for Deep Learning on Tabular Datacited by 200
  12. 2018Anchor regression: heterogeneous data meets causalitycited by 116
8 more
  1. 2022Causal Machine Learning: A Survey and Open Problemscited by 107
  2. 2014Domain-Adversarial Neural Networkscited by 78
  3. 2020Coping with Label Shift via Distributionally Robust Optimisationcited by 69
  4. 2022Invariant and Transportable Representations for Anti-Causal Domain Shiftscited by 37
  5. 2022Modeling the Data-Generating Process is Necessary for Out-of-Distribution Generalizationcited by 34
  6. 2022Provable Domain Generalization via Invariant-Feature Subspace Recoverycited by 32
  7. 2021An Online Learning Approach to Interpolation and Extrapolation in Domain Generalizationcited by 30
  8. 2018Preventing Failures Due to Dataset Shift: Learning Predictive Models That Transportcited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 6; 2026: 3.

11 papers.