Outcome-Informed Weighting for Robust ATE Estimation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yang, Linying, Evans, Robin J.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910886186188800
author Yang, Linying
Evans, Robin J.
author_facet Yang, Linying
Evans, Robin J.
contents Reliable causal effect estimation from observational data requires adjustment for confounding and sufficient overlap in covariate distributions between treatment groups. However, in high-dimensional settings, lack of overlap often inflates the variance and weakens the robustness of inverse propensity score weighting (IPW) based estimators. Although many approaches that rely on covariate adjustment have been proposed to mitigate these issues, we instead shift the focus to the outcome space. In this paper, we introduce the Augmented Marginal outcome density Ratio (AMR) estimator, an outcome-informed weighting method that naturally filters out irrelevant information, alleviates practical positivity violations and outperforms standard augmented IPW and covariate adjustment-based methods in terms of both efficiency and robustness. Additionally, by eliminating the need for strong a priori assumptions, our post-hoc calibration framework is also effective in settings with high-dimensional covariates. We present experimental results on synthetic data, the NHANES dataset and text applications, demonstrating the robustness of AMR and its superior performance under weak overlap and high-dimensional covariates.
format Preprint
id arxiv_https___arxiv_org_abs_2503_15989
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Outcome-Informed Weighting for Robust ATE Estimation
Yang, Linying
Evans, Robin J.
Methodology
Reliable causal effect estimation from observational data requires adjustment for confounding and sufficient overlap in covariate distributions between treatment groups. However, in high-dimensional settings, lack of overlap often inflates the variance and weakens the robustness of inverse propensity score weighting (IPW) based estimators. Although many approaches that rely on covariate adjustment have been proposed to mitigate these issues, we instead shift the focus to the outcome space. In this paper, we introduce the Augmented Marginal outcome density Ratio (AMR) estimator, an outcome-informed weighting method that naturally filters out irrelevant information, alleviates practical positivity violations and outperforms standard augmented IPW and covariate adjustment-based methods in terms of both efficiency and robustness. Additionally, by eliminating the need for strong a priori assumptions, our post-hoc calibration framework is also effective in settings with high-dimensional covariates. We present experimental results on synthetic data, the NHANES dataset and text applications, demonstrating the robustness of AMR and its superior performance under weak overlap and high-dimensional covariates.
title Outcome-Informed Weighting for Robust ATE Estimation
topic Methodology
url https://arxiv.org/abs/2503.15989