Construction of risk prediction model of sentinel lymph node metastasis in breast cancer patients based on machine learning algorithm.
Authors
Affiliations (4)
Affiliations (4)
- Department of Ultrasound, The First Affiliated Hospital of Chongqing University of Chinese Medicine, Chongqing Hospital of Traditional Chinese Medicine, Chongqing, 400021, China.
- Department of Radiology, The First Affiliated Hospital of Chongqing University of Chinese Medicine, Chongqing Hospital of Traditional Chinese Medicine, Chongqing, 400021, China.
- Department of Mammary Gland, The First Affiliated Hospital of Chongqing University of Chinese Medicine, Chongqing Hospital of Traditional Chinese Medicine, Chongqing, 400021, China.
- Department of Traditional Chinese Medicine, ChongQing JiangJin District Hospital of Chinese Medicine (Jiangjin Hospital, Chongqing University of Chinese Medicin), Chongqing, 402260, China. [email protected].
Abstract
The aim of this study was to develop and validate a machine learning (ML) based prediction model for sentinel lymph node metastasis in breast cancer to identify patients with a high risk of sentinel lymph node metastasis. In this machine learning study, we retrospectively collected 225 female breast cancer patients who underwent sentinel lymph node biopsy (SLNB). Feature screening was performed using the logistic regression analysis. Subsequently, five ML algorithms, namely LOGIT, LASSO, XGBOOST, RANDOM FOREST model and GBM model were employed to train and develop an ML model. In addition, model interpretation was performed by the Shapley Additive Explanations (SHAP) analysis to clarify the importance of each feature of the model and its decision basis. Combined univariate and multivariate logistic regression analysis, identified Multifocal, LVI, Maximum Diameter, Shape US, Maximum Cortical Thickness as significant predictors. We than successfully leveraged machine learning algorithms, particularly the RANDOM FOREST model, to develop a predictive model for sentinel lymph node metastasis in breast cancer. Finally, the SHAP method identified Maximum Diameter and Maximum Cortical Thickness as the primary decision factors influencing the ML model's predictions. With the integration of pathological and imaging characteristics, ML algorithm can accurately predict sentinel lymph node metastasis in breast cancer patients. The RANDOM FOREST model showed ideal performance. With the incorporation of these models in the clinic, can helpful for clinicians to identify patients at risk of sentinel lymph node metastasis of breast cancer and make more reasonable treatment decisions.