Doubly Robust Conformalized Survival Analysis with Right-Censored Data
Matteo Sesia, Vladimir Svetnik
Abstract
We present a conformal inference method for constructing lower prediction bounds for survival times from right-censored data, extending recent approaches designed for more restrictive type-I censoring scenarios. The proposed method imputes unobserved censoring times using a machine learning model, and then analyzes the imputed data using a survival model calibrated via weighted conformal inference. This approach is theoretically supported by an asymptotic double robustness property. Empirical studies on simulated and real data demonstrate that our method leads to relatively informative predictive inferences and is especially robust in challenging settings where the survival model may be inaccurate.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on3
- Conformal Prediction with Missing ValuesMargaux Zaffran, Aymeric Dieuleveut, Julie Josse, Yaniv RomanoICML 2023 · 31 citations
- Derandomized novelty detection with FDR control via conformal e-valuesMeshi Bashari, Amir Epstein, Yaniv Romano, Matteo SesiaNeurIPS 2023 · 28 citations
- Conformalized Survival Distributions: A Generic Post-Process to Increase CalibrationShiang Qi, Yakun Yu, Russell GreinerICML 2024 · 10 citations
Related papers
- Conformalized Survival Analysis for General Right-Censored DataHen Davidov, Shai Feldman, Gil Shamai, Ron Kimmel et al.ICLR 2025
- Conformalized Survival Counterfactuals Prediction for General Right-Censored DataSijie Ren, Meng Yan, Zhen Zhang, Xu Yinghui et al.ICLR 2026
- T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLPJiaye Teng, Zeren Tan, Yang YuanICML 2021 · 18 citations
- Causality-Based Conformal Imputation Correction with Non-Random Missing LabelsChunyuan Zheng, Xiang Li, Hang Pan, Eric Wang et al.KDD 2026
- Toward Conditional Distribution Calibration in Survival PredictionShiang Qi, Yakun Yu, Russell GreinerNeurIPS 2024 · 5 citations
