Papers › Outcome-Informed Weighting for Robust ATE Estimation

Outcome-Informed Weighting for Robust ATE Estimation

20 Mar 2025arXiv:2503.15989links table onlyarchive 2025-07-28

Linying Yang, Robin J. Evans

The archive published only this paper's code-link row. Authors, date and abstract are from arXiv's metadata (CC0), read from the Kaggle arXiv metadata snapshot of 2026-09-12 where its title matched the archive's; the title is the archive's.

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.

PaperPDFCode

Code

linyingyang/AMR officialmentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections