Machine learning model based on CT radiomics for predicting severity of acute phase traumatic brain injury
10.13929/j.issn.1003-3289.2024.07.008
- VernacularTitle:基于CT影像组学机器学习模型预测急性期创伤性脑损伤严重程度
- Author:
Yuqi YANG
1
,
2
;
Jianing LUO
;
Yongxiang YANG
;
Dongbo ZOU
;
Kun WEI
;
Yongli XIA
;
Min CHEN
;
Yuan MA
Author Information
1. 西南医科大学附属医院神经外科,四川泸州 646000
2. 江油市人民医院重症医学科,四川江油 621700
- Keywords:
brain injuries;
machine learning;
radiomics
- From:
Chinese Journal of Medical Imaging Technology
2024;40(7):992-996
- CountryChina
- Language:Chinese
-
Abstract:
Objective To explore the value of machine learning(ML)models based on non-contrast CT(NCCT)radiomics features for predicting the severity of acute phase traumatic brain injury(TBI).Methods Totally 600 TBI patients were retrospectively collected as observation group,other 65 TBI patients were taken as external validation set,while 50 TBI patients were prospectively enrolled as prospective validation set.Patients in observation group were divided into high-risk subgroup(n=240)and low-risk subgroup(n=360)according to Glasgow outcome scale(GOS)at discharge.The severity of acute phase TBI in observation group was assessed by doctor A and B with the same criteria,then an artificial model was established based on clinical and NCCT data at the time of first diagnosis using logistic regression(LR)method for predicting the severity of acute phase TBI.Patients in observation group were divided into training set(n=420,including 168 in high-risk subgroup and 252 in low-risk subgroup)and test set(n=180,including 72 in high-risk subgroup and 108 in low-risk subgroup)at the ratio of 7∶3.Based on NCCT of training set,radiomics features were extracted and selected,and LR,support vector machine(SVM),random forest(RF)and K-nearest neighbor(KNN)were used to establish 4 ML models.The efficacies of the above models were validated in test set,external validation set(including 34 cases of high-risk and 31 cases of low-risk TBI)and prospective validation set(including 21 cases of high-risk and 29 cases of low-risk TBI),respectively.Results The area under the curve(AUC)of doctor A and B for evaluating the severity of acute phase TBI in observation group was 0.606 and 0.771,respectively,of artificial model was 0.824.Based on NCCT in training set,6 optimal radiomics features were selected to construct LR,SVM,RF and KNN ML models,with AUC of 0.983,0.971,0.970 and 0.984 in test set,respectively,while the AUC of artificial model was 0.708.The AUC of LR,SVM,RF,KNN ML models and artificial model in external validation set was 0.879,0.881,0.984,0.863 and 0.733,while in prospective validation set was 0.984,0.873,0.982,0.897 and 0.704,respectively.Conclusion ML models based on CT radiomics could effectively predict the severity of acute phase TBI.
