Application of machine learning algorithms to predict lymph node metastasis in gastric neuroendocrine neoplasms

Background: Neuroendocrine neoplasms (NENs) are tumors that originate from secretory cells of the diffuse endocrine system and typically produce bioactive amines or peptide hormones. This paper describes the development and validation of a predictive model of the risk of lymph node metastasis among gastric NEN patients based on machine learning platform.

Methods: In this investigation, data from 1256 patients were used, of whom 119 patients from the First Affiliated Hospital of Soochow University in China and 1137 cases from the surveillance epidemiology and end results (SEER) database were combined. Six machine learning algorithms, including the logistic regression model (LR), random forest (RF), decision tree (DT), Naive Bayes (NB), support vector machine (SVM), and k-nearest neighbor algorithm (KNN), were used to build the predictive model. The performance of the models was evaluated using the area under the receiver operating characteristic curve (AUC), accuracy, sensitivity, and specificity.

Results: Among the 1256 patients with gastric NENs, 276 patients (21.97 %) developed lymph node metastasis. T stage, tumor size, degree of differentiation, and sex were predictive factors of lymph node metastasis. The RF model achieved the best predictive performance among the six machine learning models, with an AUC, accuracy, sensitivity, and specificity of 0.81, 0.78, 0.76, and 0.82, respectively.

Conclusion: The RF model provided the best prediction and can help physicians determine the lymph node metastasis risk of gastric NEN patients to formulate individualized medical strategies.