Establishment of a machine learning model for the diagnosis of clinically significant prostate cancer based on transrectal contrast-enhanced ultrasound parameters and clinical data
10.3760/cma.j.cn131148-20220708-00484
- VernacularTitle:基于经直肠超声造影参数及临床资料建立机器学习模型诊断临床显著性前列腺癌
- Author:
Xiu LIU
1
;
Fang LI
;
Yujie FENG
;
Ruixia HONG
;
Ying LI
;
Huai ZHAO
;
Hang ZHOU
;
Jiaqi GONG
Author Information
1. 重庆大学医学院 重庆大学附属肿瘤医院，重庆 400030
- Keywords:
Clinically significant prostate cancer;
Contrast-enhanced ultrasound, transrectal;
Machine learning
- From:
Chinese Journal of Ultrasonography
2023;32(1):20-26
- CountryChina
- Language:Chinese
-
Abstract:
Objective:To establish a machine learning model for the diagnosis of clinically significant prostate cancer based on transrectal contrast-enhanced ultrasound parameters and clinically relevant data.Methods:A retrospective analysis was performed on 151 patients in Chongqing University Cancer Hospital who underwent transrectal contrast-enhanced ultrasonography and transrectal ultrasound-guided needle biopsy from November 2018 to September 2021. The time intensity curve was drawn using VueBox software and 12 parameters such as rise time, peak time, average transit time, peak intensity, and rising slope were quantitatively analyzed. Age, total prostate-specific antigen, free prostate-specific antigen, free prostate-specific antigen ratio, volume, prostate-specific antigen density, and transrectal contrast-enhanced ultrasonography parameters, a total of 18 characteristic parameters, were analyzed and screened through relevant attribute values and information gain attribute values. The screening features were trained and tested by the machine learning single algorithm and integrated algorithm, and then the model was evaluated by the F1 value and the area under the ROC curve(AUC).Results:Using the related attribute value and the information gain attribute value, 12 variables and 5 variables were screened out respectively to establish a machine learning model. The model established by the ensemble algorithm was better than the single algorithm. For the two variable selection methods, the AUC (0.810 vs 0.789) and F1 values (0.748 vs 0.742) of the Bagging ensemble algorithm model, which basic algorithm was decision tree, were the highest, followed by Logistic regression and support vector machine(SVM) in order of AUC and F1 values.Conclusions:Based on transrectal contrast-enhanced ultrasound parameters and clinical data, the Bagging ensemble model based on decision tree has the best performance in diagnosing clinically significant prostate cancer.
