Saved in:
Bibliographic Details
Main Authors: Wang, Yuxin, Frauen, Dennis, Schweisthal, Jonas, Schröder, Maresa, Feuerriegel, Stefan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2510.13397
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909032268169216
author Wang, Yuxin
Frauen, Dennis
Schweisthal, Jonas
Schröder, Maresa
Feuerriegel, Stefan
author_facet Wang, Yuxin
Frauen, Dennis
Schweisthal, Jonas
Schröder, Maresa
Feuerriegel, Stefan
contents Dropout is common in clinical studies, with up to half of patients leaving early due to side effects or other reasons. When dropout is informative (i.e., dependent on survival time), it introduces censoring bias, because of which treatment effect estimates are also biased. In this paper, we propose an assumption-lean framework to assess the robustness of conditional average treatment effect (CATE) estimates in survival analysis when facing censoring bias. Unlike existing works that rely on strong assumptions, such as non-informative censoring, to obtain point estimation, we use partial identification to derive informative bounds on the CATE. Thereby, our framework helps to identify patient subgroups where treatment is effective despite informative censoring. We further propose a novel model-agnostic meta-learner, called SurvB-learner, to estimate the bounds that can be used in combination with arbitrary machine-learning models, and that has favorable theoretical properties such as double-robustness and quasi-oracle efficiency. We finally demonstrate the effectiveness of our meta-learner across various experiments using both simulated and real-world data.
format Preprint
id arxiv_https___arxiv_org_abs_2510_13397
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Assessing the robustness of heterogeneous treatment effects in survival analysis under informative censoring
Wang, Yuxin
Frauen, Dennis
Schweisthal, Jonas
Schröder, Maresa
Feuerriegel, Stefan
Machine Learning
Dropout is common in clinical studies, with up to half of patients leaving early due to side effects or other reasons. When dropout is informative (i.e., dependent on survival time), it introduces censoring bias, because of which treatment effect estimates are also biased. In this paper, we propose an assumption-lean framework to assess the robustness of conditional average treatment effect (CATE) estimates in survival analysis when facing censoring bias. Unlike existing works that rely on strong assumptions, such as non-informative censoring, to obtain point estimation, we use partial identification to derive informative bounds on the CATE. Thereby, our framework helps to identify patient subgroups where treatment is effective despite informative censoring. We further propose a novel model-agnostic meta-learner, called SurvB-learner, to estimate the bounds that can be used in combination with arbitrary machine-learning models, and that has favorable theoretical properties such as double-robustness and quasi-oracle efficiency. We finally demonstrate the effectiveness of our meta-learner across various experiments using both simulated and real-world data.
title Assessing the robustness of heterogeneous treatment effects in survival analysis under informative censoring
topic Machine Learning
url https://arxiv.org/abs/2510.13397