Can predictive models be used for causal inference?
Not on the map. Why?
Abstract (arXiv)
Supervised machine learning (ML) and deep learning (DL) algorithms excel at predictive tasks, but it is commonly assumed that they often do so by exploiting non-causal correlations, which may limit both interpretability and generalizability. Here, we show that this trade-off between explanation and prediction is not as deep and fundamental as expected. Whereas ML and DL algorithms will indeed tend to use non-causal features for prediction when fed indiscriminately with all data, it is possible to constrain the learning process of any ML and DL algorithm by selecting features according to Pearl's backdoor adjustment criterion. In such a situation, some algorithms, in particular deep neural networks, can provide near unbiased effect estimates under feature collinearity. Remaining biases are explained by the specific algorithmic structures as well as hyperparameter choice. Consequently, optimal hyperparameter settings are different when tuned for prediction or inference, confirming the general expectation of a trade-off between prediction and explanation. However, the effect of this trade-off is small compared to the effect of a causally constrained feature selection. Thus, once the causal relationship between the features is accounted for, the difference between prediction and explanation may be much smaller than commonly assumed. We also show that such causally constrained models generalize better to new data with altered collinearity structures, suggesting generalization failure may often be due to a lack of causal learning. Our results not only provide a perspective for using ML for inference of (causal) effects but also help to improve the generalizability of fitted ML and DL models to new data.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 16 of its 38 references that are in the map. First posted on arXiv 18 Jun 2023.
Builds on
16 of 38 references.
References · 16
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2019Explainable Artificial Intelligence (XAI): Concepts, Taxonomies, Opportunities and Challenges toward Responsible AIcited by 1,552
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2019The many Shapley values for model explanationcited by 344
- 2019Interpretable machine learning: definitions, methods, and applicationscited by 305
- 2016Visualizing the Effects of Predictor Variables in Black Box Supervised Learning Modelscited by 213
- 2015ranger: A Fast Implementation of Random Forests for High Dimensional Data in C++ and Rcited by 172
- 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2019Learning Neural Causal Models from Unknown Interventionscited by 113
4 more
- 2019Unrestricted Permutation forces Extrapolation: Variable Importance Requires at least One More Model, or There Is No Free Variable Importancecited by 87
- 2013Correlation and variable importance in random forestscited by 41
- 2022Wide and Deep Neural Networks Achieve Optimality for Classificationcited by 12
- 2022Machine Learning and Deep Learning -- A review for Ecologistscited by 11
Built on by
Citing papers per year: 2024: 1; 2025: 1.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2024Causal Concept Graph Models: Beyond Causal Opacity in Deep Learning
- 2023Causal reasoning in typical computer vision tasks
- 2025Integrating Causality with Neurochaos Learning: Proposed Approach and Research Agenda
- 2025CausalProfiler: Generating Synthetic Benchmarks for Rigorous and Transparent Evaluation of Causal Machine Learning
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attention
- 2026Can Post-Training Transform LLMs into Causal Reasoners?
- 2024Explaining Time Series Classification Predictions via Causal Attributions
- 2026Causality as the Statistical Conscience of Artificial Intelligence: From Pearl's Ladder to Trustworthy Machines