arXiv:2501.04871stat.MLcs.LG2025-01被引 10

提出新梯度提升法直接估计因果推断中的关键表示量

RieszBoost: Gradient Boosting for Riesz Regression

  • 用梯度提升直接学习瑞斯表示量,无需解析表达式
  • 在多种因果效应估计中表现优于或媲美传统方法
  • 适合表格数据,对正性假设不敏感,结果更稳健

回答因果问题常需估计条件期望的线性泛函,如平均处理效应或纵向修改治疗策略的效果。根据瑞斯表示定理,这些泛函可表示为结果条件期望与瑞斯表示量期望乘积。传统方法通过推导其解析表达式、估计各分量并代入实现,但推导困难且易受正性违反影响,导致方差大、置信区间宽。本文提出一种新型梯度提升算法,无需解析形式即可直接估计瑞斯表示量。该方法适用于表格数据,具有灵活性、非参数性和计算高效性。模拟研究表明,该算法在多种泛函估计中表现与或优于间接估计方法,提供了一种用户友好且稳健的因果量估计方案。

原文摘要 · Abstract (English)

Answering causal questions often involves estimating linear functionals of conditional expectations, such as the average treatment effect or the effect of a longitudinal modified treatment policy. By the Riesz representation theorem, these functionals can be expressed as the expected product of the conditional expectation of the outcome and the Riesz representer, a key component in doubly robust estimation methods. Traditionally, the Riesz representer is estimated indirectly by deriving its explicit analytical form, estimating its components, and substituting these estimates into the known form (e.g., the inverse propensity score). However, deriving or estimating the analytical form can be challenging, and substitution methods are often sensitive to practical positivity violations, leading to higher variance and wider confidence intervals. In this paper, we propose a novel gradient boosting algorithm to directly estimate the Riesz representer without requiring its explicit analytical form. This method is particularly suited for tabular data, offering a flexible, nonparametric, and computationally efficient alternative to existing methods for Riesz regression. Through simulation studies, we demonstrate that our algorithm performs on par with or better than indirect estimation techniques across a range of functionals, providing a user-friendly and robust solution for estimating causal quantities.

因果推断梯度提升瑞斯回归

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。