Search papers, labs, and topics across Lattice.
The paper introduces SurvHTE-Bench, a comprehensive benchmark for heterogeneous treatment effect (HTE) estimation in survival analysis, addressing the challenges posed by censoring and complex identification assumptions. It includes synthetic, semi-synthetic, and real-world datasets to evaluate survival HTE methods under diverse conditions. The benchmark facilitates a rigorous comparison of methods, revealing their performance under realistic assumption violations and establishing a foundation for fair and reproducible evaluation.
Finally, a standardized benchmark to rigorously compare heterogeneous treatment effect estimation methods in survival analysis, revealing performance nuances across diverse datasets and assumption violations.
Estimating heterogeneous treatment effects (HTEs) from right-censored survival data is critical in high-stakes applications such as precision medicine and individualized policy-making. Yet, the survival analysis setting poses unique challenges for HTE estimation due to censoring, unobserved counterfactuals, and complex identification assumptions. Despite recent advances, from Causal Survival Forests to survival meta-learners and outcome imputation approaches, evaluation practices remain fragmented and inconsistent. We introduce SurvHTE-Bench, the first comprehensive benchmark for HTE estimation with censored outcomes. The benchmark spans (i) a modular suite of synthetic datasets with known ground truth, systematically varying causal assumptions and survival dynamics, (ii) semi-synthetic datasets that pair real-world covariates with simulated treatments and outcomes, and (iii) real-world datasets from a twin study (with known ground truth) and from an HIV clinical trial. Across synthetic, semi-synthetic, and real-world settings, we provide the first rigorous comparison of survival HTE methods under diverse conditions and realistic assumption violations. SurvHTE-Bench establishes a foundation for fair, reproducible, and extensible evaluation of causal survival methods. The data and code of our benchmark are available at: https://github.com/Shahriarnz14/SurvHTE-Bench .
CMU ML