Machine learning survival models for relapse prediction in a early stage lung cancer patient

Timilsina, Mohan, Buosi, Samuele, Janik, Adrianna, Minervini, Pasquale, Costabello, Luca, Torrente, Maria, Provencio, Mariano, Calvo, Virginia, Camps, Carlos, Ortega, Ana L., Massutí, Bartomeu, Garcia Campelo, M.Rosario, del Barco, Edel, Bosch-Barrera, Joaquim, Novacek, Vit (2023). Machine learning survival models for relapse prediction in a early stage lung cancer patient. Paper presented at the International Joint Conference on Neural Networks (IJCNN), Gold Coast, Australia, 18-23 June. https;/dx.doi.org/10.1109/IJCNN54540.2023.10191078

Abstract

Lung cancer is one of the leading health complica tions causing high mortality worldwide. The relapsing behavior of medically treated early-stage lung cancer makes this disease even more complicated. Thus predicting such relapse using a data-centric approach provides a complementary perspective for clinicians to understand the disease. In this preliminary work, we explored off-the-shelf survival models to predict the relapse of early-stage lung cancer patients. We analyzed the survival models on a cohort of 1348 early-stage non-small cell lung cancer (NSCLC) patients in different timestamps. Using the prediction explanation model SHAP (SHapley Additive exPlanations), we further explained the best-performing survival model’s predic tions. Our explainable predictive model is a potential tool for oncologists that address an unmet clinical need for post-treatment patient stratification based on the relapse hazard.

Publisher

IEEE

Rights

Attribution 4.0 International

cb

Collections

Data Science Institute (Conference Papers)