Conformalized survival analysis

doi:10.1093/jrsssb/qkac004

Conformalized survival analysis

2023-01-28

17

OA

AI

DOI：10.1093/jrsssb/qkac004

原文链接

原文求助

摘要

En 中文

In this paper, we develop an inferential method based on 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. The validity and efficiency of our procedure are demonstrated on synthetic data and real COVID-19 data from the UK Biobank.

Keyword:

censoring

distribution boosting

prediction interval

random forests

survival time

weighted conformal inference

Conformalized survival analysis

摘要

期刊

机构