Interpretable machine learning model based on CT semantic features and radiomics features to preoperatively predict Ki-67 expression in gastrointestinal stromal tumors.

Journal: Scientific reports
PMID:

Abstract

To develop and validate a machine learning (ML) model which combined computed tomography (CT) semantic and radiomics features to preoperatively predict Ki-67 expression in gastrointestinal stromal tumors (GISTs) patients. We retrospectively collected the clinical, imaging and pathological data of 149 GISTs patients. We randomly assigned the patients in a ratio of 7:3 to a training set (104 cases) and a validation (45 cases) set. We divided the patients into low and high Ki-67 expression group according to postoperative pathology. CT semantic features were analyzed from preoperative enhancement CT images and radiomics features were extracted from venous phase-enhanced images. We used intraclass correlation coefficient, maximal relevance and minimal redundancy and least absolute shrinkage and selection operator method to screen radiomics features and build radiomics label. 6 ML models were used for model construction. Receiver operating characteristic curves were used to evaluate the predictive efficiency of ML models. SHAP analysis was used to explain the contribution of different variables and their risk threshold. AUC of radscores in predicting Ki-67 expression of GIST patients were 0.749 and 0.729 in training and validation set. Among the 6 ML models, SVM exhibited best prediction accuracy. AUC of SVM model in predicting Ki-67 expression of GIST patients were 0.840, 0.767 and 0.832 in training, validation and test set. SHAP analysis showed that radscores and tumor diameter had highly positive contribution to the model. Therefore, the interpretable SVM model can predict Ki-67 expression of GISTs patients individually before surgery, which can provide reliable imaging biomarkers for clinical treatment decisions.

Authors

  • Yating Wang
    Graduate Collaborative Training Base of the First Hospital of Changsha, Hengyang Medical School, University of South China, Hengyang, China.
  • Genji Bai
    Department of Radiology, The Affiliated Huaian No. 1 People's Hospital of Nanjing Medical University, Huaian, Jiangsu, China.
  • Yan Liu
    Department of Clinical Microbiology, Shanghai Tenth People's Hospital, School of Medicine, Tongji University, Shanghai, 200072, People's Republic of China.
  • Min Huang
    Department of Physiology, School of Basic Medicine, Chengdu Medical College, Sichuan, China.
  • Wei Chen
    Department of Urology, Zigong Fourth People's Hospital, Sichuan, China.
  • First Wang
    Department of medical imaging, The Affiliated Huaian No.1 People's Hospital of Nanjing Medical University, No.1, Huang he West Road, Huai'an, 223300, Jiangsu, China.