A Machine Learning Risk Prediction Model for Gastric Cancer with SHapley Additive exPlanations

• Published in: Cancer research and treatment Vol. 57; no. 3; pp. 821 - 829
• Main Authors: Park, Bomi, Kim, Chung Ho, Jun, Jae Kwan, Suh, Mina, Choi, Kui Son, Choi, Il Ju, Oh, Hyun Jin
• Format: Journal Article
• Language: English
• Published: Korea (South) Korean Cancer Association 01.07.2025; 대한암학회
• Subjects: Adult; Aged; Early Detection of Cancer - methods; Female; Helicobacter Infections - complications; Helicobacter Infections - epidemiology; Humans; Machine Learning; Male; Middle Aged; Original; Republic of Korea - epidemiology; Risk Assessment - methods; Risk Factors; ROC Curve; Stomach Neoplasms - diagnosis; Stomach Neoplasms - epidemiology; Stomach Neoplasms - etiology; 의학일반; Republic of Korea; SHapley Additive exPlanations; Stomach neoplasms; Machine learning; Prediction model
• ISSN: 1598-2998; 2005-9256; 2005-9256
• DOI: 10.4143/crt.2024.843

PurposeGastric cancer (GC) prediction models hold potential for enhancing early detection by enabling the identification of high-risk individuals, facilitating personalized risk-based screening, and optimizing the allocation of healthcare resources. Materials and MethodsIn this study, we developed a machine learning-based GC prediction model utilizing data from the Korean National Health Insurance Service, encompassing 10,515,949 adults who had not been diagnosed with GC and underwent GC screening during 2013-2014, with a follow-up period of 5 years. The cohort was divided into training and test datasets at an 8:2 ratio, and class imbalance was mitigated through random oversampling.ResultsAmong various models, logistic regression demonstrated the highest predictive performance, with an area under the receiver operating characteristic curve (AUC) of 0.708, which was consistent with the AUC obtained in external validation (0.669). Importantly, the outcomes were robust to missing data imputation and variable selection. The SHapley Additive exPlanations (SHAP) algorithm enhanced the explainability of the model, identifying advancing age, being male, Helicobacter pylori infection, current smoking, and a family history of GC as key predictors of elevated risk. ConclusionThis predictive model could significantly contribute to the early identification of individuals at elevated risk for GC, thereby enabling the implementation of targeted preventive strategies. Furthermore, the integration of noninvasive and cost-effective predictors enhances the clinical utility of the model, supporting its potential application in routine healthcare settings.