A Relative Error-Based Evaluation Framework of Heterogeneous Treatment Effect Estimators
Jiayi Guo, Haoxuan Li, Ye Tian, Peng Wu
Abstract
While significant progress has been made in heterogeneous treatment effect (HTE) estimation, the evaluation of HTE estimators remains underdeveloped. In this article, we propose a robust evaluation framework based on relative error, which quantifies performance differences between two HTE estimators. We first derive the key theoretical conditions on the nuisance parameters that are necessary to achieve a robust estimator of relative error. Building on these conditions, we introduce novel loss functions and design a neural network architecture to estimate nuisance parameters and obtain robust estimation of relative error, thereby achieving reliable evaluation of HTE estimators. We provide the large sample properties of the proposed relative error estimator. Furthermore, beyond evaluation, we propose a new learning algorithm for HTE that leverages both the previously HTE estimators and the nuisance parameters learned through our neural network architecture. Extensive experiments demonstrate that our evaluation framework supports reliable comparisons across HTE estimators, and the proposed learning algorithm for HTE exhibits desirable performance.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on6
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- Estimating the Effects of Continuous-valued Interventions using Generative Adversarial NetworksIoana Bica, James Jordon, Mihaela van der SchaarNeurIPS 2020 · 137 citations
- Optimal Transport for Treatment Effect EstimationHao Wang, Jiajun Fan, Zhichao Chen, Haoxuan Li et al.NeurIPS 2023 · 71 citations
- In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarICML 2023 · 36 citations
- Counterfactual Cross-Validation: Stable Model Selection Procedure for Causal Inference ModelsYuta Saito, Shota YasuiICML 2020 · 34 citations
Related papers
- Estimating Heterogeneous Treatment Effects: Mutual Information Bounds and Learning AlgorithmsXingzhuo Guo, Yuchen Zhang, Jianmin Wang, Mingsheng LongICML 2023 · 11 citations
- Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event DataDennis Frauen, Maresa Schröder, Konstantin Hess, Stefan FeuerriegelNeurIPS 2025 · 13 citations
- Direct Doubly Robust Estimation of Conditional Quantile ContrastsJosh Givens, Song Liu, Henry Reeve, Katarzyna RelugaICLR 2026
- Improving Data-driven Heterogeneous Treatment Effect Estimation Under Structure UncertaintyChristopher Tran, Elena ZhelevaKDD 2022 · 4 citations
- Minimax Optimal Nonparametric Estimation of Heterogeneous Treatment EffectsZijun Gao, Yanjun HanNeurIPS 2020 · 17 citations
