Cancer Control (Oct 2020)
Predicting Peritoneal Metastasis of Gastric Cancer Patients Based on Machine Learning
Abstract
Objective: The aim is to explore the prediction effect of 5 machine learning algorithms on peritoneal metastasis of gastric cancer. Methods: 1080 patients with postoperative gastric cancer were divided into a training group and test group according to the ratio of 7:3. The model of peritoneal metastasis was established by using 5 machine learning (gbm(Light Gradient Boosting Machine), GradientBoosting, forest, Logistic and DecisionTree). Python pair was used to analyze the machine learning algorithm. Gbm algorithm is used to show the weight proportion of each variable to the result. Result: Correlation analysis showed that tumor size and depth of invasion were positively correlated with the recurrence of patients after gastric cancer surgery. The results of the gbm algorithm showed that the top 5 important factors were albumin, platelet count, depth of infiltration, preoperative hemoglobin and weight, respectively. In training group: Among the 5 algorithm models, the accuracy of GradientBoosting and gbm was the highest (0.909); the AUC values of the 5 algorithms are gbm (0.938), GradientBoosting (0.861), forest (0.796), Logistic(0.741) and DecisionTree(0.712) from high to low. In the test group: among the 5 algorithm models, the accuracy of forest, DecisionTree and gbm was the highest (0.907); AUC values ranged from high to low to gbm (0.745), GradientBoosting (0.725), forest (0.696), Logistic (0.680) and DecisionTree (0.657). Conclusion: Machine learning can predict the peritoneal metastasis in patients with gastric cancer.