# Machine Learning for Lymph Node Metastasis Prediction in Early Gastric Cancer: A Comparative Analysis

## TL;DR

This study compares machine learning models to predict lymph node metastasis in early gastric cancer patients, aiming to improve treatment decisions.

## Contribution

A comparative analysis of seven machine learning models for predicting lymph node metastasis in early gastric cancer.

## Key findings

- Random Forest, Extreme Gradient Boosting, and Neural Network models showed strong performance with AUC values above 0.779.
- Logistic Models and Random Forest performed better in T1a and T1b subgroups compared to other models.
- SHAP analysis identified key variables influencing lymph node metastasis prediction in different subgroups.

## Abstract

Lymph node metastasis (LNM) plays a crucial role in informing treatment decisions and prognosis for early gastric cancer (EGC). This study aimed to offer a practical approach to predict LNM in EGC by using machine learning algorithms.

This study collected data from 1085 patients with EGC who underwent radical gastrectomy with D1+ or D2 lymph node resection. Seven machine-learning algorithms were compared, and hyperparameters were fine-tuned to identify the model with the best accuracy, Brier class and Area Under the Curve (AUC). The efficacy of the selected model was evaluated.

Following comparison, the Random Forest (RF), Extreme Gradient Boosting (Boost), and Neural Network (NNT) models exhibited exemplary performance on the training dataset, with AUC values of 0.796, 0.788, and 0.779, respectively, on the validation set. We conducted parallel analyses within the T1a and T1b subgroups, where Logistics Models (LM) and RF yielded AUCs of 0.710 and 0.636 in the T1a validation set, and LM, RF, and Boost achieved AUCs of 0.666, 0.658, and 0.558, respectively in the T1b validation set. Variable importance analysis utilizing SHAP revealed distinct values for lymph node metastasis (LNM) in EGC patients, as well as in those stratified into T1a and T1b groups.

The machine learning model holds the potential to guide more effective treatment strategies for early gastric cancer (EGC), specifically in addressing lymph node metastasis (LNM). The identified risk factors contribute valuable insights for personalized decision-making in the management of EGC patients.