mlsanityData to 7 Oct 2026

Assessing the robustness of heterogeneous treatment effects in survival analysis under informative censoring

arXiv 2510.13397 ↗First posted 15 Oct 2025cs.LG stat.MLcited by 1

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 45 of its 98 references that are in the map. First posted on arXiv 15 Oct 2025.

Builds on

45 of 98 references.

References · 45

  1. 2016DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Networkcited by 354
  2. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  3. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  4. 2022Semiparametric doubly robust targeted double machine learning: a reviewcited by 129
  5. 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
  6. 2023Deep Learning for Survival Analysis: A Reviewcited by 75
  7. 2021Sensitivity Analysis of Individual Treatment Effects: A Robust Conformal Inference Approachcited by 62
  8. 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 54
  9. 2021Conformalized Survival Analysiscited by 53
  10. 2018Interval Estimation of Individual-Level Causal Effects Under Unobserved Confoundingcited by 51
  11. 2021Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden Confoundingcited by 50
  12. 2021An Automated Approach to Causal Inference in Discrete Settingscited by 39
33 more
  1. 2015The Influence Function of Semiparametric Estimatorscited by 36
  2. 2017Sensitivity analysis for inverse probability weighting estimators via the percentile bootstrapcited by 33
  3. 2021Sharp Sensitivity Analysis for Inverse Propensity Weighting via Quantile Balancingcited by 32
  4. 2020Estimating heterogeneous treatment effects with right-censored data via causal survival forestscited by 29
  5. 2020A Class of Algorithms for General Instrumental Variable Modelscited by 28
  6. 2021Doubly-Valid/Doubly-Sharp Sensitivity Analysis for Causal Inference with Unmeasured Confoundingcited by 28
  7. 2023B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confoundingcited by 28
  8. 2021SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Datacited by 25
  9. 2022Stochastic Causal Programming for Bounding Treatment Effectscited by 25
  10. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  11. 2022Scalable Sensitivity and Uncertainty Analysis for Causal-Effect Estimates of Continuous-Valued Interventionscited by 20
  12. 2022Conformalized survival analysis with adaptive cutoffscited by 17
  13. 2023A Neural Framework for Generalized Causal Sensitivity Analysiscited by 17
  14. 2017Individualized Treatment Effects with Censored Data via Fully Nonparametric Bayesian Accelerated Failure Time Modelscited by 13
  15. 2020Estimating heterogeneous survival treatment effect in observational data using machine learningcited by 12
  16. 2022Sensitivity analysis under the f-sensitivity models: a distributional robustness perspectivecited by 12
  17. 2024Meta-Learners for Partially-Identified Treatment Effects Across Multiple Environmentscited by 11
  18. 2022Sensitivity Analysis for Marginal Structural Modelscited by 10
  19. 2025Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event Datacited by 10
  20. 2021Inference for treatment-specific survival curves using machine learningcited by 8
  21. 2021Estimating Heterogeneous Treatment Effects for General Responsescited by 7
  22. 2022BITES: Balanced Individual Treatment Effect for Survival datacited by 7
  23. 2021Interpretable Sensitivity Analysis for Balancing Weightscited by 5
  24. 2022Proximal Survival Analysis to Handle Dependent Right Censoringcited by 5
  25. 2024Estimating Heterogeneous Treatment Effects on Survival Outcomes Using Counterfactual Censoring Unbiased Transformationscited by 4
  26. 2024Learning Representations of Instruments for Partial Identification of Treatment Effectscited by 4
  27. 2018One-step Targeted Maximum Likelihood for Time-to-event Outcomescited by 3
  28. 2020Sharpening the Rosenbaum Sensitivity Bounds to Address Concerns About Interactions Between Observed and Unobserved Covariatescited by 3
  29. 2021Addressing Extreme Propensity Scores in Estimating Counterfactual Survival Functions via the Overlap Weightscited by 2
  30. 2025Treatment Effect Estimation in Causal Survival Analysis: Practical Recommendationscited by 2
  31. 2021Partial Identification and Inference in Duration Models with Endogenous Censoringcited by 1
  32. 2024Estimating Interpretable Heterogeneous Treatment Effect with Causal Subgroup Discovery in Survival Outcomescited by 1
  33. 2025Bounds for the regression parameters in dependently censored survival modelscited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.