IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2508.12688.html

Bayesian Double Machine Learning for Causal Inference

Author

Listed:
  • Francis J. DiTraglia
  • Laura Liu

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.

Suggested Citation

  • Francis J. DiTraglia & Laura Liu, 2025. "Bayesian Double Machine Learning for Causal Inference," Papers 2508.12688, arXiv.org.
  • Handle: RePEc:arx:papers:2508.12688
    as

    Download full text from publisher

    File URL: http://arxiv.org/pdf/2508.12688
    File Function: Latest version
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Achim Ahrens & Victor Chernozhukov & Christian Hansen & Damian Kozbur & Mark Schaffer & Thomas Wiemann, 2025. "An Introduction to Double/Debiased Machine Learning," Papers 2504.08324, arXiv.org, revised Feb 2026.
    2. Brett R. Gordon & Robert Moakler & Florian Zettelmeyer, 2023. "Close Enough? A Large-Scale Exploration of Non-Experimental Approaches to Advertising Measurement," Marketing Science, INFORMS, vol. 42(4), pages 768-793, July.
    3. Cotti, Chad & Courtemanche, Charles & Liang, Yang & Maclean, Johanna Catherine & Nesson, Erik & Sabia, Joseph J., 2025. "The effect of e-cigarette flavor bans on tobacco use," Journal of Health Economics, Elsevier, vol. 102(C).
    4. Yang, Jui-Chung & Chuang, Hui-Ching & Kuan, Chung-Ming, 2020. "Double machine learning with gradient boosting and its application to the Big N audit quality effect," Journal of Econometrics, Elsevier, vol. 216(1), pages 268-283.
    5. Joseph Antonelli & Georgia Papadogeorgou & Francesca Dominici, 2022. "Causal inference in high dimensions: A marriage between Bayesian modeling and good frequentist properties," Biometrics, The International Biometric Society, vol. 78(1), pages 100-114, March.
    6. Kaspar Wüthrich & Ying Zhu, 2023. "Omitted Variable Bias of Lasso-Based Inference Methods: A Finite Sample Analysis," The Review of Economics and Statistics, MIT Press, vol. 105(4), pages 982-997, July.
    7. Wüthrich, Kaspar & Zhu, Ying, 2023. "Omitted Variable Bias of Lasso-Based Inference Methods: A Finite Sample Analysis," University of California at San Diego, Economics Working Paper Series qt1gp6g9gm, Department of Economics, UC San Diego.
    8. Yan Dora Zhang & Brian P. Naughton & Howard D. Bondell & Brian J. Reich, 2022. "Bayesian Regression Using a Prior on the Model Fit: The R2-D2 Shrinkage Prior," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 117(538), pages 862-874, April.
    9. Christopher D. Walker, 2023. "Parametrization, Prior Independence, and the Semiparametric Bernstein-von Mises Theorem for the Partially Linear Model," Papers 2306.03816, arXiv.org, revised Apr 2025.
    10. Luo, Yu & Graham, Daniel J. & McCoy, Emma J., 2023. "Semiparametric Bayesian doubly robust causal estimation," LSE Research Online Documents on Economics 117944, London School of Economics and Political Science, LSE Library.
    11. James J. Heckman & Hedibert F. Lopes & Rémi Piatek, 2014. "Treatment Effects: A Bayesian Perspective," Econometric Reviews, Taylor & Francis Journals, vol. 33(1-4), pages 36-67, June.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Qihui Chen & Zheng Fang & Ruixuan Liu, 2025. "Debiased Bayesian Inference for High-dimensional Regression Models," Papers 2512.09257, arXiv.org.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Jiang, Liang & Li, Liyao & Miao, Ke & Zhang, Yichong, 2025. "Adjustments with many regressors under covariate-adaptive randomizations," Journal of Econometrics, Elsevier, vol. 249(PB).
    2. Christis Katsouris, 2023. "High Dimensional Time Series Regression Models: Applications to Statistical Learning Methods," Papers 2308.16192, arXiv.org.
    3. Jonathan Fuhr & Philipp Berens & Dominik Papies, 2024. "Estimating Causal Effects with Double Machine Learning -- A Method Evaluation," Papers 2403.14385, arXiv.org, revised Apr 2024.
    4. Roshnie Doon & Sergio Scicchitano, 2025. "The effects of overeducation on wage distribution in Trinidad and Tobago: an unconditional quantile regression analysis," Eurasian Economic Review, Springer;Eurasia Business and Economics Society, vol. 15(1), pages 163-195, March.
    5. Sullivan Hu'e & S'ebastien Laurent & Ulrich Aiounou & Emmanuel Flachaire, 2025. "Estimation in high-dimensional linear regression: Post-Double-Autometrics as an alternative to Post-Double-Lasso," Papers 2511.21257, arXiv.org.
    6. Gozde Sert & Abhishek Chakrabortty & Anirban Bhattacharya, 2025. "Bayesian Semiparametric Causal Inference: Targeted Doubly Robust Estimation of Treatment Effects," Papers 2511.15904, arXiv.org.
    7. Bruno Fava, 2025. "Training and Testing with Multiple Splits: A Central Limit Theorem for Split-Sample Estimators," Papers 2511.04957, arXiv.org, revised Nov 2025.
    8. Achim Ahrens & Victor Chernozhukov & Christian Hansen & Damian Kozbur & Mark Schaffer & Thomas Wiemann, 2025. "An Introduction to Double/Debiased Machine Learning," Papers 2504.08324, arXiv.org, revised Feb 2026.
    9. Dennis Lim & Wenjie Wang & Yichong Zhang, 2024. "A Dimension-Agnostic Bootstrap Anderson-Rubin Test For Instrumental Variable Regressions," Papers 2412.01603, arXiv.org, revised Sep 2025.
    10. Watanabe, Hajime & Maruyama, Takuya, 2024. "A Bayesian sample selection model with a binary outcome for handling residential self-selection in individual car ownership," Journal of choice modelling, Elsevier, vol. 51(C).
    11. Lili Liu & Tiantian Feng & Cheng Zhong & Mingli Cui & Haoran Wang, 2025. "Synergy pathway innovation of carbon emission trading, tradable green certificate and green power trading policies: achieving China’s dual carbon goals," Humanities and Social Sciences Communications, Palgrave Macmillan, vol. 12(1), pages 1-28, December.
    12. Eric Yanchenko & Howard D. Bondell & Brian J. Reich, 2024. "Spatial regression modeling via the R2D2 framework," Environmetrics, John Wiley & Sons, Ltd., vol. 35(2), March.
    13. Helena Barnard & Robin Cowan & Moritz Müller, 2016. "On the value of foreign PhDs in the developing world: Training versus selection effects," Working Papers of BETA 2016-04, Bureau d'Economie Théorique et Appliquée, UDS, Strasbourg.
    14. Lei Suo & Linsen Zhu & Haiying Feng & Wei Li, 2025. "The Impact of Urbanization Level on Urban Ecological Resilience and Its Role Mechanisms: A Case Study of Resource-Based Cities in China," Sustainability, MDPI, vol. 17(19), pages 1-28, September.
    15. Gao, Yue & Zhu, Yun, 2025. "Government debt settlement and corporate supply chain resilience–an empirical study based on china's special supervision campaign," Finance Research Letters, Elsevier, vol. 86(PG).
    16. Shimeles Abebe & Andinet Woldemichael, 2015. "Working Paper 225 - Measuring the Impact of Micro-Health Insurance on Healthcare Utilization: A Bayesian Potential Outcomes Approach," Working Paper Series 2166, African Development Bank.
    17. Yantuan Yu & Ning Zhang, 2025. "Revealing the Power of Market-Based Energy Policy: Evidence from China’s Energy Quota Trading System Using Machine Learning," Working papers 2025rwp-258, Yonsei University, Yonsei Economics Research Institute.
    18. Duhirwe, Patrick Nzivugira & Ngarambe, Jack & Yun, Geun Young, 2024. "Causal effects of policy and occupant behavior on cooling energy," Renewable and Sustainable Energy Reviews, Elsevier, vol. 206(C).
    19. Heejun Shin & Joseph Antonelli, 2023. "Improved inference for doubly robust estimators of heterogeneous treatment effects," Biometrics, The International Biometric Society, vol. 79(4), pages 3140-3152, December.
    20. Tao Sun, 2024. "Bundle Choice Model with Endogenous Regressors: An Application to Soda Tax," Papers 2412.05794, arXiv.org.

    More about this item

    NEP fields

    This paper has been announced in the following NEP Reports:

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:arx:papers:2508.12688. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: arXiv administrators (email available below). General contact details of provider: http://arxiv.org/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.