Conformalized survival analysis

Type: Preprint

Publication Date: 2023-01-28

Citations: 18

DOI: https://doi.org/10.1093/jrsssb/qkac004

Abstract

Existing survival analysis techniques heavily rely on strong modelling assumptions and are, therefore, prone to model misspecification errors. In this paper, we develop an inferential method based on ideas from conformal prediction, which can wrap around any survival prediction algorithm to produce calibrated, covariate-dependent lower predictive bounds on survival times. In the Type I right-censoring setting, when the censoring times are completely exogenous, the lower predictive bounds have guaranteed coverage in finite samples without any assumptions other than that of operating on independent and identically distributed data points. Under a more general conditionally independent censoring assumption, the bounds satisfy a doubly robust property which states the following: marginal coverage is approximately guaranteed if either the censoring mechanism or the conditional survival function is estimated well. Further, we demonstrate that the lower predictive bounds remain valid and informative for other types of censoring. The validity and efficiency of our procedure are demonstrated on synthetic data and real COVID-19 data from the UK Biobank.

Locations

  • Journal of the Royal Statistical Society Series B (Statistical Methodology) - View - PDF
  • arXiv (Cornell University) - View - PDF
  • PubMed - View

Similar Works

Action Title Year Authors
+ Conformalized Survival Analysis 2021 Emmanuel J. Candès
Lihua Lei
Zhimei Ren
+ PDF Chat Doubly Robust Conformalized Survival Analysis with Right-Censored Data 2024 Matteo Sesia
Vladimir Svetnik
+ PDF Chat Two-sided conformalized survival analysis 2024 Chris Holmes
Ariane Marandon
+ PDF Chat Conformal predictive intervals in survival analysis: a re-sampling approach 2024 Jing Qin
Jin Piao
Jing Ning
Yu Shen
+ Conformalized survival analysis with adaptive cutoffs 2022 Yu Gui
Rohan Hore
Zhimei Ren
Rina Foygel Barber
+ PDF Chat Conformalized survival analysis with adaptive cut-offs 2023 Yu Gui
Rohan Hore
Zhimei Ren
Rina Foygel Barber
+ PDF Chat Doubly Robust and Efficient Calibration of Prediction Sets for Censored Time-to-Event Outcomes 2025 Richard A. Fariña
Arun Kumar Kuchibhotla
Eric J. Tchetgen Tchetgen
+ PDF Chat Conformalized Survival Distributions: A Generic Post-Process to Increase Calibration 2024 Shi-ang Qi
Yakun Yu
Russell Greiner
+ PDF Chat Toward Conditional Distribution Calibration in Survival Prediction 2024 Shi-ang Qi
Yakun Yu
Russell Greiner
+ Conformal prediction with censored data using Kaplan-Meier method 2024 Sun Xiao-lin
Yanhua Wang
+ Conformal Prediction using Conditional Histograms. 2021 Matteo Sesia
Yaniv Romano
+ Conformal Prediction using Conditional Histograms 2021 Matteo Sesia
Yaniv Romano
+ Conformal Prediction using Conditional Histograms 2021 Matteo Sesia
Yaniv Romano
+ Conformal histogram regression 2021 Matteo Sesia
Yaniv Romano
+ PDF Chat Conformal Inference of Individual Treatment Effects Using Conditional Density Estimates 2025 Baozhen Wang
Xingye Qiao
+ T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLP 2021 Jiaye Teng
Zeren Tan
Yang Yuan
+ PDF Chat T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLP 2021 Jiaye Teng
Zeren Tan
Yang Yuan
+ PDF Chat Prediction Sets and Conformal Inference with Censored Outcomes 2025 Weiguang Liu
Áureo de Paula
Elie Tamer
+ A Comprehensive Framework for Evaluating Time to Event Predictions using the Restricted Mean Survival Time 2023 Ariane Cwiling
Vittorio Perduca
Olivier Bouaziz
+ Conformalized Quantile Regression 2019 Yaniv Romano
Evan Patterson
Emmanuel J. Candès