SurvHTE-Bench: A Benchmark for Heterogeneous Treatment Effect Estimation in Survival Analysis
Shahriar Noroozizadeh, Xiaobin Shen, Jeremy C. Weiss, George H. Chen
摘要
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 .
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper4
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 被引用 114 次
- SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event DataAlicia Curth, Changhee Lee, Mihaela van der SchaarNeurIPS 2021 · 被引用 41 次
- An Effective Meaningful Way to Evaluate Survival ModelsShiang Qi, Neeraj Kumar, Mahtab Farrokh, Weijie Sun 等ICML 2023 · 被引用 28 次
- Heterogeneous Treatment Effect in Time-to-Event Outcomes: Harnessing Censored Data with Recursively Imputed TreesTomer Meir, Uri Shalit, Malka GorfineICML 2025
相关 Paper
- Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event DataDennis Frauen, Maresa Schröder, Konstantin Hess, Stefan FeuerriegelNeurIPS 2025 · 被引用 13 次
- Stable Estimation of Heterogeneous Treatment EffectsAnpeng Wu, Kun Kuang, Ruoxuan Xiong, Bo Li 等ICML 2023 · 被引用 31 次
- Comparison of meta-learners for estimating multi-valued treatment heterogeneous effectsNaoufal Acharki, Ramiro Lugo, Antoine Bertoncello, Josselin GarnierICML 2023 · 被引用 18 次
- Causal Effect Estimation under Informative Censoring: A Multiple-Imputation Approach with Shadow CalibrationYingrong Wang, Baohong Li, Dian Jin, Yi He 等KDD 2026
- Estimating Heterogeneous Treatment Effects: Mutual Information Bounds and Learning AlgorithmsXingzhuo Guo, Yuchen Zhang, Jianmin Wang, Mingsheng LongICML 2023 · 被引用 11 次
