IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2508.12688.html
   My bibliography  Save this paper

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. 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.
    2. 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).
    3. 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.
    4. 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.
    5. 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.
    6. 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.
    7. 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.
    8. 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.
    9. 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)

    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. 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.
    2. 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).
    3. 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.
    4. 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.
    5. 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.
    6. 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.
    7. 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.
    8. 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.
    9. 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).
    10. 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.
    11. Tao Sun, 2024. "Bundle Choice Model with Endogenous Regressors: An Application to Soda Tax," Papers 2412.05794, arXiv.org.
    12. Yong Bian & Xiqian Wang & Qin Zhang, 2023. "How Does China's Household Portfolio Selection Vary with Financial Inclusion?," Papers 2311.01206, arXiv.org.
    13. Huber, Martin & Meier, Jonas & Wallimann, Hannes, 2022. "Business analytics meets artificial intelligence: Assessing the demand effects of discounts on Swiss train tickets," Transportation Research Part B: Methodological, Elsevier, vol. 163(C), pages 22-39.
    14. Horvath, Akos & Lang, Peter, 2021. "Do loan subsidies boost the real activity of small firms?," Journal of Banking & Finance, Elsevier, vol. 122(C).
    15. Henrika Langen & Martin Huber, 2023. "How causal machine learning can leverage marketing strategies: Assessing and improving the performance of a coupon campaign," PLOS ONE, Public Library of Science, vol. 18(1), pages 1-37, January.
    16. Guiming Liao & Zheli Hu & Zhichen Yang & Qianlin Yin & Jiahui Chen, 2025. "The Perception of Policy Uncertainty and the Labor Income Share of Firms: Empirical Research Based on Double Machine Learning," Managerial and Decision Economics, John Wiley & Sons, Ltd., vol. 46(2), pages 999-1011, March.
    17. Yuchen Lu & Jiakun Zhuang & Jun Chen & Chenlu Yang & Mei Kong, 2025. "The Impact of Farmland Transfer on Urban–Rural Integration: Causal Inference Based on Double Machine Learning," Land, MDPI, vol. 14(1), pages 1-30, January.
    18. Marie Albertine Djuikom, 2018. "Incentives to labour migration and agricultural productivity: The Bayesian perspective," WIDER Working Paper Series wp-2018-45, World Institute for Development Economic Research (UNU-WIDER).
    19. Kangqi Jiang & Xiaofeng Chen & Jiayun Li & Mengling Zhou, 2025. "Technology adoption and extreme stock risk: Evidence from digital tax reform in China," Humanities and Social Sciences Communications, Palgrave Macmillan, vol. 12(1), pages 1-20, December.
    20. Brett R. Gordon & Robert Moakler & Florian Zettelmeyer, 2023. "Predictive Incrementality by Experimentation (PIE) for Ad Measurement," Papers 2304.06828, 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.