LightGBM is an Effective Predictive Model for Postoperative Complications in Gastric Cancer: A Study Integrating Radiomics with Ensemble Learning

Wenli Wang,Rongrong Sheng,Shumei Liao,Zifeng Wu,Linjun Wang,Cunming Liu,Chun Yang,Riyue Jiang
DOI: https://doi.org/10.1007/s10278-024-01172-0
2024-06-28
Abstract:Postoperative complications of radical gastrectomy seriously affect postoperative recovery and require accurate risk prediction. Therefore, this study aimed to develop a prediction model specifically tailored to guide perioperative clinical decision-making for postoperative complications in patients with gastric cancer. A retrospective analysis was conducted on patients who underwent radical gastrectomy at the First Affiliated Hospital of Nanjing Medical University between April 2022 and June 2023. A total of 166 patients were enrolled. Patient demographic characteristics, laboratory examination results, and surgical pathological features were recorded. Preoperative abdominal CT scans were used to segment the visceral fat region of the patients through 3Dslicer, a 3D Convolutional Neural Network (3D-CNN) to extract image features and the LASSO regression model was employed for feature selection. Moreover, an ensemble learning strategy was adopted to train the features and predict postoperative complications of gastric cancer. The prediction performance of the LGBM (Light Gradient Boosting Machine), XGB (XGBoost), RF (Random Forest), and GBDT (Gradient Boosting Decision Tree) models was evaluated through fivefold cross-validation. This study successfully constructed a model for predicting early complications following radical gastrectomy based on the optimal algorithm, LGBM. The LGBM model yielded an AUC value of 0.9232 and an accuracy of 87.28% (95% CI, 75.61-98.95%), surpassing the performance of other models. Through ensemble learning and integration of perioperative clinical data and visceral fat radiomics, a predictive LGBM model was established. This model has the potential to facilitate individualized clinical decision-making and the early recovery of patients with gastric cancer post-surgery.
What problem does this paper attempt to address?