Computer aided diagnosis of thyroid nodules based on the devised small-datasets multi-view ensemble learning

被引:50
作者
Chen, Yifei [1 ]
Li, Dandan [1 ]
Zhang, Xin [1 ]
Jin, Jing [1 ]
Shen, Yi [2 ]
机构
[1] Harbin Inst Technol, Dept Control Sci & Engn, Harbin, Peoples R China
[2] Harbin Inst Technol, Shenzhen Engn Lab Med Intelligent Wireless, Shenzhen, Peoples R China
基金
国家重点研发计划;
关键词
Thyroid nodule ultrasound images; Small dataset; Computer aided diagnosis; Deep learning; CLASSIFICATION;
D O I
10.1016/j.media.2020.101819
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
With the development of deep learning, its application in diagnosis of benign and malignant thyroid nodules has been widely concerned. However, it is difficult to obtain medical images, resulting in insufficient number of data, which contradicts the large amount of data required for acquiring effective deep learning diagnostic models. A multi-view ensemble learning based on voting mechanism is proposed herein to boost the performance of the models trained by small-dataset thyroid nodule ultrasound images. The method integrates three kinds of diagnosis results which are obtained from 3-view dataset which is composed of thyroid nodule ultrasound images, medical features extracted based on U-Net output and useful features selected by mRMR from the statistical features and texture features. To obtain preliminary diagnosis results, the images are utilized for training GoogleNet. For improving the results, supplementary methods were proposed based on the medical features and the selected features. To analyze the contribution of these features and acquire two groups of diagnosis results, the designed Xgboost classifier is utilized for obtaining two groups of features respectively. Subsequently, the boosting final results are obtained through majority voting mechanism. Furthermore, the proposed method is utilized to diagnose sequence images (the images extracted by frame from videos) to solve the poor results caused by slight differences. Finally, better final results are obtained for both of the normal dataset and the sequence dataset (consisting of sequence images). Compared with the accuracies obtained by only training deep learning models with small datasets, the diagnostic accuracies of the above two datasets are improved to 92.11% and 92.54% respectively by utilizing the proposed method. (C) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:13
相关论文
共 29 条
[1]  
Brattain L.J., 2018, ABDOM RADIOL, DOI [10.1007/s0 0261-018- 1517-0., DOI 10.1007/S00261-018-1517-0.]
[2]  
Bushberg J.T., 2003, MED PHYS, DOI [10.1118/1.1585033, DOI 10.1118/1.1585033.]
[3]  
Chang R.F., 2005, BREAST CANC RES TREA, DOI [10.1007/s10549-004- 2043- z., DOI 10.1007/S10549-004-2043-Z.]
[4]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[5]   Thyroid Nodule Classification in Ultrasound Images by Fine-Tuning Deep Convolutional Neural Network [J].
Chi, Jianning ;
Walia, Ekta ;
Babyn, Paul ;
Wang, Jimmy ;
Groot, Gary ;
Eramian, Mark .
JOURNAL OF DIGITAL IMAGING, 2017, 30 (04) :477-486
[6]  
Cortese L., 2018, OPT INFOBASE C, DOI [10.1364/OTS.2018. OTh4D.5., DOI 10.1364/OTS.2018.OTH4D.5.]
[7]  
Du X., 2017, P 2016 31 YOUTH AC A, DOI [10.1109/YAC.2016.7804882., DOI 10.1109/YAC.2016.7804882.]
[8]  
Frannita E.L., 2018, P 2018 2 INT C BIOM, DOI [10.1109/IBIOMED.2018.8534944., DOI 10.1109/IBIOMED.2018.8534944.]
[9]  
Galloway M.M., 1975, Comput. Graph. Image Process, V4, P172, DOI DOI 10.1016/S0146-664X(75)80008-6
[10]  
Gurcan M.N., 2009, IEEE REV BIOMED ENG, DOI [10. 1109/RBME.2009.2034865, DOI 10.1109/RBME.2009.2034865.]