Conformalized Survival Counterfactuals Prediction for General Right-Censored Data
Sijie Ren, Meng Yan, Zhen Zhang, Xu Yinghui, Xinwei Sun
Abstract
This paper aims to develop a lower prediction bound (LPB) for survival time across different treatments in the general right-censored setting. Although previous methods have utilized conformal prediction to construct the LPB, their resulting prediction sets provide only probably approximately correct (PAC)–type miscoverage guarantees rather than exact ones. To address this problem, we propose a new calibration procedure under the potential outcome framework. Under the strong ignorability assumption, we propose a reweighting scheme that can transform the problem into a weighted conformal inference problem, allowing an LPB to be obtained via quantile regression with an exact miscoverage guarantee. Furthermore, our procedure is doubly robust against model misspecification. Empirical evaluations on synthetic and real-world clinical data demonstrate the validity and informativeness of our constructed LPBs, which indicate the potential of our analytical benchmark for comparing and selecting personalized treatments.
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 on4
- Conformalized Survival Distributions: A Generic Post-Process to Increase CalibrationShiang Qi, Yakun Yu, Russell GreinerICML 2024 · 10 citations
- Robust Conformal Prediction Using Privileged InformationShai Feldman, Yaniv RomanoNeurIPS 2024 · 7 citations
- Censored Quantile Regression Neural Networks for Distribution-Free Survival AnalysisTim Pearce, Jong-Hyeon Jeong, Yichen Jia, Jun ZhuNeurIPS 2022
- Conformalized Survival Analysis for General Right-Censored DataHen Davidov, Shai Feldman, Gil Shamai, Ron Kimmel et al.ICLR 2025
Related papers
- Doubly Robust Conformalized Survival Analysis with Right-Censored DataMatteo Sesia, Vladimir SvetnikICML 2025
- Conformal Counterfactual Inference under Hidden ConfoundingZonghao Chen, Ruocheng Guo, Jean-Francois Ton, Yang LiuKDD 2024 · 2 citations
- T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLPJiaye Teng, Zeren Tan, Yang YuanICML 2021 · 18 citations
- Not all distributional shifts are equal: Fine-grained robust conformal inferenceJiahao Ai, Zhimei RenICML 2024 · 15 citations
- Kernel-based Optimally Weighted Conformal Time-Series PredictionJonghyeok Lee, Chen Xu, Yao XieICLR 2025
