EconBase
← All papers

Bayesian Double Machine Learning for Causal Inference

Francis J. DiTraglia, Laura Liu

arXiv 18 Aug 2025 · Econometrics

arXiv:2508.12688 · PDF · DOI · OpenAlex · Extracted main text

Abstract

This paper proposes a simple, novel, and fully-Bayesian approach for causal inference in partially linear models with high-dimensional control variables. Off-the-shelf machine learning methods can introduce biases in the causal parameter known as regularization-induced confounding. To address this, we propose a Bayesian Double Machine Learning (BDML) method, which modifies a standard Bayesian multivariate regression model and recovers the causal effect of interest from the reduced-form covariance matrix. Our BDML is related to the burgeoning frequentist literature on DML while addressing its limitations in finite-sample inference. Moreover, the BDML is based on a fully generative probability model in the DML context, adhering to the likelihood principle. We show that in high dimensional setups the naive estimator implicitly assumes no selection on observables--unlike our BDML. The BDML exhibits lower asymptotic bias and achieves asymptotic normality and semiparametric efficiency as established by a Bernstein-von Mises theorem, thereby ensuring robustness to misspecification. In simulations, our BDML achieves lower RMSE, better frequentist coverage, and shorter confidence interval width than alternatives from the literature, both Bayesian and frequentist.

Citation extraction

35
references
77
in-text mentions
35
distinct cited
0
self-citations
12,374
main-text words

appendix boundary found by appendix_command · 69% of the source is main text. Read the extracted text to check this.

Most heavily cited references

The works this paper leans on most, across its whole bibliography — not restricted to papers in our corpus. Ranked by composite intensity, which combines how often a work is mentioned, how many sections mention it, and how much of that falls in the main text rather than the appendix.

ReferenceIntensityMentionsSectionsMain text
1Hahn, P.R., Carvalho, C.M., Puelz, D., He, J (2018) Regularization and Confounding in Linear Regression for Treatment Effect Estimation1.000104100%
2Belloni, A., Chernozhukov, V., Hansen, C (2014) Inference on Treatment Effects after Selection among High-Dimensional Controls1.00053100%
3Linero, A.R (2023) In Nonparametric and High-Dimensional Models, Bayesian Ignorability is an Informative Prior0.96520690%
4Hahn, P.R., Murray, J.S., Carvalho, C.M (2020) Bayesian Regression Tree Models for Causal Inference: Regularization, Confounding, and Heterogeneous Effects (with Discussion)0.87452100%
5Chernozhukov, V., Chetverikov, D., Demirer, M., Duflo, E., Hansen, C… (2018) Double/debiased machine learning for treatment and structural parameters0.73732100%
6Ahrens, A., Chernozhukov, V., Hansen, C., Kozbur, D., Schaffer, M.,… (2025) An Introduction to Double/Debiased Machine Learning0.51121100%
7Breunig, C., Liu, R., Yu, Z (2024) Semiparametric Bayesian Difference-in-Differences0.51121100%
8Luo, Y., Graham, D.J., McCoy, E.J (2023) Semiparametric Bayesian doubly robust causal estimation0.51121100%
9Walker, C.D (2025) Parametrization, Prior Independence, and the Semiparametric Bernstein-von Mises Theorem for the Partially Linear Model0.51121100%
10Angrist, J.D., Frandsen, B (2022) Machine Labor0.40511100%

Showing the top 10 of 35 scored citations.

Cited by, within the corpus

arXiv econ.EM papers that cite this one, ranked by how heavily they lean on it.

Citing paperIntensityMentionsSections
1Debiased Bayesian Inference for High-dimensional Regression Models0.40511