arXiv:2508.17136stat.MLcs.LG2025-08被引 5

提出FIDDLE模型,用因子增强深度学习估计处理效应,适合高维强相关数据。

Factor Informed Double Deep Learning For Average Treatment Effect Estimation

  • 融合因子增强神经网络,非参数选择变量并学习低维函数结构
  • 在高维、非线性、强相关数据下仍能一致估计平均处理效应
  • 适用于大数据场景,尤其适合存在模型误设的因果推断任务

针对高维、高度相关且具有稀疏非线性效应的协变量,本文研究平均处理效应(ATE)估计问题。提出一种因子增强型双深度学习方法(FIDDLE),基于FAST-NN构建响应函数与倾向得分估计,在增广逆倾向加权(AIPW)框架下实现稳健估计。FIDDLE无需参数假设,可非参数化地筛选关键变量,并通过神经网络自适应学习低维函数结构。该方法在模型误设条件下仍能一致估计ATE,且在极灵活的倾向得分与结果模型族中达到半参数效率。通过合成与真实数据集的广泛数值实验,验证了理论结论,并表明其在高维情形下优于传统方法。

原文摘要 · Abstract (English)

We investigate the problem of estimating the average treatment effect (ATE) under a very general setup where the covariates can be high-dimensional, highly correlated, and can have sparse nonlinear effects on the propensity and outcome models. We present the use of a Double Deep Learning strategy for estimation, which involves combining recently developed factor-augmented deep learning-based estimators, FAST-NN, for both the response functions and propensity scores to achieve our goal. By using FAST-NN, our method can select variables that contribute to propensity and outcome models in a completely nonparametric and algorithmic manner and adaptively learn low-dimensional function structures through neural networks. Our proposed novel estimator, FIDDLE (Factor Informed Double Deep Learning Estimator), estimates ATE based on the framework of augmented inverse propensity weighting AIPW with the FAST-NN-based response and propensity estimates. FIDDLE consistently estimates ATE even under model misspecification and is flexible to also allow for low-dimensional covariates. Our method achieves semiparametric efficiency under a very flexible family of propensity and outcome models. We present extensive numerical studies on synthetic and real datasets to support our theoretical guarantees and establish the advantages of our methods over other traditional choices, especially when the data dimension is large.

因果推断深度学习高维数据处理效应

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