A Comprehensive Framework for Evaluating Time to Event Predictions using the Restricted Mean Survival Time (2306.16075v2)
Abstract: The restricted mean survival time (RMST) is a widely used quantity in survival analysis due to its straightforward interpretation. For instance, predicting the time to event based on patient attributes is of great interest when analyzing medical data. In this paper, we propose a novel framework for evaluating RMST estimations. A criterion that estimates the mean squared error of an RMST estimator using Inverse Probability Censoring Weighting (IPCW) is presented. A model-agnostic conformal algorithm adapted to right-censored data is also introduced to compute prediction intervals and to evaluate local variable importance. Finally, a model-agnostic statistical test is developed to assess global variable importance. Our framework is valid for any RMST estimator that is asymptotically convergent and works under model misspecification.
- “Statistical Models Based on Counting Processes”, Springer Series in Statistics New York, NY: Springer US, 1993 DOI: 10.1007/978-1-4612-4348-9
- Per Kragh Andersen, Mette Gerster Hansen and John P. Klein “Regression Analysis of Restricted Mean Survival Time Based on Pseudo-Observations” In Lifetime Data Analysis 10.4, 2004, pp. 335–350 DOI: 10.1007/s10985-004-4771-0
- Paul Blanche, Michael W. Kattan and Thomas A. Gerds “The c-index is not proper for the evaluation of $t$-year predicted risks” In Biostatistics 20.2, 2019, pp. 347–357 DOI: 10.1093/biostatistics/kxy006
- Henrik Boström, Ulf Johansson and Anders Vesterberg “Predicting with Confidence from Survival Data” In Proceedings of the Eighth Symposium on Conformal and Probabilistic Prediction and Applications PMLR, 2019, pp. 123–141
- Leo Breiman “Random Forests” In Machine Learning 45.1, 2001, pp. 5–32 DOI: 10.1023/A:1010933404324
- Emmanuel Candès, Lihua Lei and Zhimei Ren “Conformalized survival analysis” In Journal of the Royal Statistical Society Series B: Statistical Methodology 85.1, 2023, pp. 24–45 DOI: 10.1093/jrsssb/qkac004
- George H. Chen “Deep Kernel Survival Analysis and Subject-Specific Survival Time Prediction Intervals” In Proceedings of the 5th Machine Learning for Healthcare Conference PMLR, 2020, pp. 537–565 URL: https://proceedings.mlr.press/v126/chen20a.html
- David R. Cox “Partial Likelihood” In Biometrika 62.2, 1975, pp. 269–276 DOI: 10.2307/2335362
- Cyrus J. DiCiccio, Thomas J. DiCiccio and Joseph P. Romano “Exact tests via multiple data splitting” In Statistics & Probability Letters 166, 2020, pp. 108865 DOI: 10.1016/j.spl.2020.108865
- Anne Eaton, Terry Therneau and Jennifer Le-Rademacher “Designing clinical trials with (restricted) mean survival time endpoint: Practical considerations” In Clinical Trials 17.3, 2020, pp. 285–294 DOI: 10.1177/1740774520905563
- Thomas R. Fleming and David P. Harrington “Counting Processes and Survival Analysis”, Wiley Series in Probability and Statistics Hoboken, NJ, USA: John Wiley & Sons, Inc., 2005 DOI: 10.1002/9781118150672
- “Estimating a time-dependent concordance index for survival prediction models with covariate dependent censoring” In Statistics in Medicine 32.13, 2013, pp. 2173–2184 DOI: 10.1002/sim.5681
- Thomas A. Gerds and Martin Schumacher “Consistent Estimation of the Expected Brier Score in General Survival Models with Right-Censored Event Times” In Biometrical Journal 48.6, 2006, pp. 1029–1040 DOI: 10.1002/bimj.200610301
- Patrick J. Heagerty and Yingye Zheng “Survival Model Predictive Accuracy and ROC Curves” In Biometrics 61.1, 2005, pp. 92–105 DOI: 10.1111/j.0006-341X.2005.030814.x
- “Random survival forests” In The Annals of Applied Statistics 2.3, 2008 DOI: 10.1214/08-AOAS169
- Theodore Karrison “Restricted Mean Life With Adjustment for Covariates” In Journal of the American Statistical Association 82.400, 1987, pp. 1169–1176 DOI: 10.2307/2289396
- Masahiro Kojima “Variable Selection using Inverse Survival Probability Weighting” arXiv, 2022 URL: http://arxiv.org/abs/2212.02109
- Maxim S. Kovalev, Lev V. Utkin and Ernest M. Kasimov “SurvLIME: A method for explaining machine learning survival models” In Knowledge-Based Systems 203, 2020, pp. 106164 DOI: 10.1016/j.knosys.2020.106164
- “SurvSHAP(t): Time-dependent explanations of machine learning survival models” In Knowledge-Based Systems 262, 2023, pp. 110234 DOI: 10.1016/j.knosys.2022.110234
- Arun Kumar Kuchibhotla “Exchangeability, Conformal Prediction, and Rank Tests” arXiv, 2021 DOI: 10.48550/arXiv.2005.06095
- “Distribution-Free Predictive Inference for Regression” In Journal of the American Statistical Association 113.523, 2018, pp. 1094–1111 DOI: 10.1080/01621459.2017.1307116
- Olivier Lopez “Réduction de dimension en présence de données censurées”, 2007 URL: https://pastel.archives-ouvertes.fr/tel-00195261
- Dariusz Majerek, Wioletta Nowak and Wies Ziba “Conditional strong law of large number” In International Journal of Pure and Applied Mathematics 20, 2005
- “Randomized 2 x 2 trial evaluating hormonal treatment and the duration of chemotherapy in node-positive breast cancer patients. German Breast Cancer Study Group.” In Journal of Clinical Oncology 12.10, 1994, pp. 2086–2093 DOI: 10.1200/JCO.1994.12.10.2086
- Jiaye Teng, Zeren Tan and Yang Yuan “T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLP” In Proceedings of the 38th International Conference on Machine Learning PMLR, 2021, pp. 10203–10213 URL: https://proceedings.mlr.press/v139/teng21a.html
- Lu Tian, Lihui Zhao and L. J. Wei “Predicting the restricted mean event time with the subject’s baseline covariates in survival analysis” In Biostatistics 15.2, 2014, pp. 222–233 DOI: 10.1093/biostatistics/kxt050
- Aad W. Van Der Vaart “Asymptotic statistics”, Cambridge series in statistical and probabilistic mathematics Cambridge, UK: Cambridge University Press, 1998
- Vladimir Vovk, A. Gammerman and Glenn Shafer “Algorithmic learning in a random world” New York: Springer, 2005
- Xin Wang and Douglas E. Schaubel “Modeling restricted mean survival time under general censoring mechanisms” In Lifetime Data Analysis 24.1, 2018, pp. 176–199 URL: http://link.springer.com/10.1007/s10985-017-9391-6
- Lili Zhao “Deep neural networks for predicting restricted mean survival times” In Bioinformatics 36.24, 2021, pp. 5672–5677 DOI: 10.1093/bioinformatics/btaa1082
- David M. Zucker “Restricted Mean Life with Covariates: Modification and Extension of a Useful Survival Analysis Method” In Journal of the American Statistical Association 93.442, 1998, pp. 702–709 DOI: 10.2307/2670120