Riesz Regression As Direct Density Ratio Estimation

arXiv stat.ML / 3/25/2026

💬 OpinionIdeas & Deep AnalysisModels & Research

Key Points

  • The paper explains how Riesz regression in causal inference connects directly to density ratio estimation (DRE) for tasks like average treatment effect estimation.
  • It shows that the Riesz representer can be expressed as a signed density ratio.
  • The authors prove that the Riesz regression objective matches the least-squares importance fitting criterion from prior DRE work.
  • Because of this equivalence, existing DRE theory—such as convergence rates, Bregman-divergence generalizations, and regularization methods—can be carried over to Riesz regression, including for flexible models like neural networks.

Abstract

This study clarifies the relationship between Riesz regression [Chernozhukov et al., 2021] and density ratio estimation (DRE) in causal inference problems, such as average treatment effect estimation. We first show that the Riesz representer can be written as a signed density ratio and then demonstrate that the Riesz regression objective coincides with the least-squares importance fitting criterion [Kanamori et al., 2009]. Although Riesz regression applies to a broad class of representer estimation problems, this equivalence with DRE allows us to transfer existing DRE results, including convergence rate analyses, generalizations based on Bregman divergence minimization, and regularization techniques for flexible models such as neural networks.