Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study
Objectives This two-center study aimed to establish a model for predicting the risk of lymph node metastasis in gastric cancer patients using machine learning (ML) and logistic regression (LR) algorithms, and to evaluate its predictive performance in clinical practice. Methods Data of a total of 369...
| Main Authors: | , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
SAGE Publishing
2024-01-01
|
| Series: | Technology in Cancer Research & Treatment |
| Online Access: | https://doi.org/10.1177/15330338231222331 |
| _version_ | 1797361423194521600 |
|---|---|
| author | Tong Lu MM Yu Fang MM Haonan Liu MM Chong Chen MM Taotao Li MM Miao Lu MSN Daqing Song MB |
| author_facet | Tong Lu MM Yu Fang MM Haonan Liu MM Chong Chen MM Taotao Li MM Miao Lu MSN Daqing Song MB |
| author_sort | Tong Lu MM |
| collection | DOAJ |
| description | Objectives This two-center study aimed to establish a model for predicting the risk of lymph node metastasis in gastric cancer patients using machine learning (ML) and logistic regression (LR) algorithms, and to evaluate its predictive performance in clinical practice. Methods Data of a total of 369 patients who underwent radical gastrectomy in the Department of General Surgery of Affiliated Hospital of Xuzhou Medical University (Xuzhou, China) from March 2016 to November 2019 were collected and retrospectively analyzed as the training group. In addition, data of 123 patients who underwent radical gastrectomy in the Department of General Surgery of Jining First People's Hospital (Jining, China) were collected and analyzed as the verification group. Besides, 7 ML and logistic models were developed, including decision tree, random forest, support vector machine (SVM), gradient boosting machine (GBM), naive Bayes, neural network, and LR, in order to evaluate the occurrence of lymph node metastasis in patients with gastric cancer. The ML model was established following 10 cross-validation iterations within the training dataset, and subsequently, each model was assessed using the test dataset. The model's performance was evaluated by comparing the area under the receiver operating characteristic curve of each model. Results Compared with the traditional logistic model, among the 7 ML algorithms, except for SVM, the other models exhibited higher accuracy and reliability, and the influences of various risk factors on the model were more intuitive. Conclusion For the prediction of lymph node metastasis in gastric cancer patients, the ML algorithm outperformed traditional LR, and the GBM algorithm exhibited the most robust predictive capability. |
| first_indexed | 2024-03-08T15:53:29Z |
| format | Article |
| id | doaj.art-3d8ed1f49a9c4a7a81e5b904124c7ad8 |
| institution | Directory Open Access Journal |
| issn | 1533-0338 |
| language | English |
| last_indexed | 2024-03-08T15:53:29Z |
| publishDate | 2024-01-01 |
| publisher | SAGE Publishing |
| record_format | Article |
| series | Technology in Cancer Research & Treatment |
| spelling | doaj.art-3d8ed1f49a9c4a7a81e5b904124c7ad82024-01-09T03:03:25ZengSAGE PublishingTechnology in Cancer Research & Treatment1533-03382024-01-012310.1177/15330338231222331Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center StudyTong Lu MM0Yu Fang MM1Haonan Liu MM2Chong Chen MM3Taotao Li MM4Miao Lu MSN5Daqing Song MB6 Department of emergency medicine, Jining No.1 People's Hospital, Jining, China , Xuzhou, China Department of Oncology, The Affiliated Hospital of Xuzhou Medical University, Xuzhou, China Department of Gastroenterology, Xuzhou No.1 People's Hospital, Xuzhou, China Department of emergency medicine, Jining No.1 People's Hospital, Jining, China Wuxi Mental Health Center, Wuxi, China Department of emergency medicine, Jining No.1 People's Hospital, Jining, ChinaObjectives This two-center study aimed to establish a model for predicting the risk of lymph node metastasis in gastric cancer patients using machine learning (ML) and logistic regression (LR) algorithms, and to evaluate its predictive performance in clinical practice. Methods Data of a total of 369 patients who underwent radical gastrectomy in the Department of General Surgery of Affiliated Hospital of Xuzhou Medical University (Xuzhou, China) from March 2016 to November 2019 were collected and retrospectively analyzed as the training group. In addition, data of 123 patients who underwent radical gastrectomy in the Department of General Surgery of Jining First People's Hospital (Jining, China) were collected and analyzed as the verification group. Besides, 7 ML and logistic models were developed, including decision tree, random forest, support vector machine (SVM), gradient boosting machine (GBM), naive Bayes, neural network, and LR, in order to evaluate the occurrence of lymph node metastasis in patients with gastric cancer. The ML model was established following 10 cross-validation iterations within the training dataset, and subsequently, each model was assessed using the test dataset. The model's performance was evaluated by comparing the area under the receiver operating characteristic curve of each model. Results Compared with the traditional logistic model, among the 7 ML algorithms, except for SVM, the other models exhibited higher accuracy and reliability, and the influences of various risk factors on the model were more intuitive. Conclusion For the prediction of lymph node metastasis in gastric cancer patients, the ML algorithm outperformed traditional LR, and the GBM algorithm exhibited the most robust predictive capability.https://doi.org/10.1177/15330338231222331 |
| spellingShingle | Tong Lu MM Yu Fang MM Haonan Liu MM Chong Chen MM Taotao Li MM Miao Lu MSN Daqing Song MB Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study Technology in Cancer Research & Treatment |
| title | Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study |
| title_full | Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study |
| title_fullStr | Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study |
| title_full_unstemmed | Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study |
| title_short | Comparison of Machine Learning and Logic Regression Algorithms for Predicting Lymph Node Metastasis in Patients with Gastric Cancer: A two-Center Study |
| title_sort | comparison of machine learning and logic regression algorithms for predicting lymph node metastasis in patients with gastric cancer a two center study |
| url | https://doi.org/10.1177/15330338231222331 |
| work_keys_str_mv | AT tonglumm comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT yufangmm comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT haonanliumm comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT chongchenmm comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT taotaolimm comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT miaolumsn comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy AT daqingsongmb comparisonofmachinelearningandlogicregressionalgorithmsforpredictinglymphnodemetastasisinpatientswithgastriccanceratwocenterstudy |