EconBase
← All papers

A Unified Theory for Causal Inference: Direct Debiased Machine Learning via Bregman-Riesz Regression

Masahiro Kato

arXiv 30 Oct 2025 · Statistics — Machine Learning

arXiv:2510.26783 · PDF · Extracted main text

Abstract

This note introduces a unified theory for causal inference that integrates Riesz regression, covariate balancing, density-ratio estimation (DRE), targeted maximum likelihood estimation (TMLE), and the matching estimator in average treatment effect (ATE) estimation. In ATE estimation, the balancing weights and the regression functions of the outcome play important roles, where the balancing weights are referred to as the Riesz representer, bias-correction term, and clever covariates, depending on the context. Riesz regression, covariate balancing, DRE, and the matching estimator are methods for estimating the balancing weights, where Riesz regression is essentially equivalent to DRE in the ATE context, the matching estimator is a special case of DRE, and DRE is in a dual relationship with covariate balancing. TMLE is a method for constructing regression function estimators such that the leading bias term becomes zero. Nearest Neighbor Matching is equivalent to Least Squares Density Ratio Estimation and Riesz Regression.

Citation extraction

20
references
40
in-text mentions
20
distinct cited
4
self-citations
3,436
main-text words

appendix boundary found by appendix_command · 100% 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
1Masahiro Kato (2025) Direct debiased machine learning via bregman divergence minimization, 2025b self1.00084100%
2Masahiro Kato (2025) Nearest neighbor matching as least squares density ratio estimation and riesz regression, 2025c self0.92844100%
3Masahiro Kato (2025) Direct bias-correction term estimation for propensity scores and average treatment effect estimation, 2025a self0.84333100%
4David Bruns-Smith, Oliver Dukes, Avi Feller, and Elizabeth L Ogburn (2025) Augmented balancing weights as linear regression0.73732100%
5Jens Hainmueller (2012) Entropy balancing for causal effects: A multivariate reweighting method to produce balanced samples in observational studies0.73732100%
6Qingyuan Zhao (2019) Covariate balancing propensity score by tailored loss functions0.64441100%
7Victor Chernozhukov, Whitney K. Newey, Victor Quintas-Martinez, and… (2024) Automatic debiased machine learning via riesz regression, 20240.51121100%
8Victor Chernozhukov, Whitney K. Newey, and Rahul Singh (2022) Automatic debiased machine learning of causal and structural effects0.40511100%
9Daniel G. Horvitz and Donovan J. Thompson (1952) A generalization of sampling without replacement from a finite universe0.40511100%
10Kosuke Imai and Marc Ratkovic (2013) Estimating treatment effect heterogeneity in randomized program evaluation0.40511100%

Showing the top 10 of 20 scored citations.