Survival Prediction in Stomach Cancer with Deep Learning: Unveiling Model Decisions with LIME and SHAP

Document Type : Research Articles

Authors

Department of Radiation Oncology, All India Institute of Medical Sciences, Bhubaneswar, India.

Abstract

Objective: Stomach cancer is anticipated to remain a significant global health concern, underscoring the urgent need for sophisticated prognostic models. The aim of the study is to build an intuitive deep learning model for predicting survival probabilities in stomach cancer patients, validating it with external data and merging SHAP and LIME to improve the therapeutic relevance and reliability. Methods: A deep learning survival model was developed with multilayer perceptron, on 1,350 documented stomach cancer cases from the AIIMS, Bhubaneswar Cancer Registry (2018–2022). The model was refined utilizing the Adam optimizer (learning rate = 0.002) with dropout regularization. External validation was performed on an independent cohort of 388 patients from Hi-Tech Medical College and Hospital. Performance was assessed using accuracy, precision, sensitivity, specificity, F1-score, balanced accuracy, Matthews correlation coefficient, concordance index, and AUROC score. LIME and SHAP were utilized to improve interpretability by evaluating both local and global feature contributions. Result: Complex interactions between important prognostic factors such as age, stage, treatment approaches, and socioeconomic level were well explained by LIME and SHAP, thus exposing important elements impacting survival results. Performance measures of the model measured through various metrics showed good generalizability over several datasets. Conclusion: This article focused on interpretable artificial intelligence models in the prognosis for stomach cancer with patient-specific survival projections. Artificial intelligence techniques such as LIME and SHAP improves clinician trust, hence promoting patient specific treatment recommendations.

Keywords

Main Subjects