vix.ing · top · new · best · stats · spec

Causal isotonic calibration for heterogeneous treatment effects

2023/02/27 by Lars van der Laan, van der Laan, Lars, Ernesto Ulloa‐Pérez +5 · 5 citations
Mathematics · #Advanced Causal Inference Techniques #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Methodology (stat.ME) #Statistical Methods and Bayesian Inference #Statistical Methods and Inference

paper · pdf · doi:10.48550/arxiv.2302.14011

openalex publication_date 2023/02/27 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

We propose causal isotonic calibration, a novel nonparametric method for calibrating predictors of heterogeneous treatment effects. Furthermore, we introduce cross-calibration, a data-efficient variant of calibration that eliminates the need for hold-out calibration sets. Cross-calibration leverages cross-fitted predictors and generates a single calibrated predictor using all available data. Under weak conditions that do not assume monotonicity, we establish that both causal isotonic calibration and cross-calibration achieve fast doubly-robust calibration rates, as long as either the propensity score or outcome regression is estimated accurately in a suitable sense. The proposed causal isotonic calibrator can be wrapped around any black-box learning algorithm, providing robust and distribution-free calibration guarantees while preserving predictive performance.

Cited by

Related