Development and validation of AI models using LR and LightGBM for predicting distant metastasis in breast cancer: a dual-center study

• Published in: Frontiers in oncology Vol. 14; p. 1409273
• Main Authors: Zhang, Wen-Hai, Tan, Yang, Huang, Zhen, Tan, Qi-Xing, Zhang, Yue-Mei, Chen, Bin-Jie, Wei, Chang-Yuan
• Format: Journal Article
• Language: English
• Published: Switzerland Frontiers Media S.A 2024
• Subjects: breast cancer; distant metastasis; LightGBM algorithm; AI; LightGBM algorithm; breast cancer; LR; distant metastasis
• ISSN: 2234-943X; 2234-943X
• DOI: 10.3389/fonc.2024.1409273

This study aims to develop an artificial intelligence model utilizing clinical blood markers, ultrasound data, and breast biopsy pathological information to predict the distant metastasis in breast cancer patients. Data from two medical centers were utilized, Clinical blood markers, ultrasound data, and breast biopsy pathological information were separately extracted and selected. Feature dimensionality reduction was performed using Spearman correlation and LASSO regression. Predictive models were constructed using LR and LightGBM machine learning algorithms and validated on internal and external validation sets. Feature correlation analysis was conducted for both models. The LR model achieved AUC values of 0.892, 0.816, and 0.817 for the training, internal validation, and external validation cohorts, respectively. The LightGBM model achieved AUC values of 0.971, 0.861, and 0.890 for the same cohorts, respectively. Clinical decision curve analysis showed a superior net benefit of the LightGBM model over the LR model in predicting distant metastasis in breast cancer. Key features identified included creatine kinase isoenzyme (CK-MB) and alpha-hydroxybutyrate dehydrogenase. This study developed an artificial intelligence model using clinical blood markers, ultrasound data, and pathological information to identify distant metastasis in breast cancer patients. The LightGBM model demonstrated superior predictive accuracy and clinical applicability, suggesting it as a promising tool for early diagnosis of distant metastasis in breast cancer.