Identification and validation of a novel machine learning model for predicting severe pelvic endometriosis: A retrospective study.

Journal: Scientific reports
PMID:

Abstract

This study aimed to explore potential risk factors for severe endometriosis and to develop a model to predict the risk of severe endometriosis. A total of 308 patients with endometriosis were analyzed. Least absolute shrinkage and selection operator (LASSO) was performed to identify the potential risk factors for severe endometriosis. Then, we used seven machine learning (ML) algorithms to construct the predictive models. Finally, SHapley Additive exPlanations (SHAP) interpretation was performed to evaluate the contributions of each factor to risk prediction. About 59.2% (183/308) of patients were diagnosed with severe endometriosis. The random forest (RF) model performed best in discriminative ability among the seven ML models, achieving an area under the curve (AUC) of 0.744. After reducing features according to feature importance rank, an explainable final RF model was established with six features. From the SHAP map, we found that the negative sliding sign had the greatest impact on the diagnostic performance of the RF model. This study provided a personalized risk assessment for the development of severe endometriosis, which may enable early identification of high-risk patients, facilitating timely intervention and optimized treatment strategies.

Authors

  • Siqi Cao
    Department of Ultrasound, The First Hospital of China Medical University, 110001, Shenyang, Liaoning Province, China.
  • Xingzhe Li
    Department of Radiation Oncology, University of Texas Southwestern Medical Center, Dallas, Texas.
  • Xin Zheng
    Department of Clinical Laboratory, Longhua Hospital, Shanghai University of Traditional Chinese Medicine, Shanghai, China. Electronic address: dearjanna@126.com.
  • Jiaxin Zhang
    School of Chinese Medicine, Hong Kong Traditional Chinese Medicine Phenome Research Center, State Key Laboratory of Environmental and Biological Analysis, Hong Kong Baptist University, Hong Kong 999077, China.
  • Ziyao Ji
    Department of Ultrasound, The First Hospital of China Medical University, 110001, Shenyang, Liaoning Province, China.
  • Yanjun Liu