Comparing different deep learning architectures for classification of chest radiographs

被引:102
作者
Bressem, Keno K. [1 ]
Adams, Lisa C. [2 ]
Erxleben, Christoph [1 ]
Hamm, Bernd [1 ,2 ]
Niehues, Stefan M. [1 ]
Vahldiek, Janis L. [1 ]
机构
[1] Charite Univ Med Berlin, Campus Benjamin Franklin, Hindenburgdamm 30, D-12203 Berlin, Germany
[2] Charite Univ Med Berlin, Campus Mitte, Charitepl 1, D-10117 Berlin, Germany
关键词
D O I
10.1038/s41598-020-70479-z
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Chest radiographs are among the most frequently acquired images in radiology and are often the subject of computer vision research. However, most of the models used to classify chest radiographs are derived from openly available deep neural networks, trained on large image datasets. These datasets differ from chest radiographs in that they are mostly color images and have substantially more labels. Therefore, very deep convolutional neural networks (CNN) designed for ImageNet and often representing more complex relationships, might not be required for the comparably simpler task of classifying medical image data. Sixteen different architectures of CNN were compared regarding the classification performance on two openly available datasets, the CheXpert and COVID-19 Image Data Collection. Areas under the receiver operating characteristics curves (AUROC) between 0.83 and 0.89 could be achieved on the CheXpert dataset. On the COVID-19 Image Data Collection, all models showed an excellent ability to detect COVID-19 and non-COVID pneumonia with AUROC values between 0.983 and 0.998. It could be observed, that more shallow networks may achieve results comparable to their deeper and more complex counterparts with shorter training times, enabling classification performances on medical image data close to the state-of-the-art methods even when using limited hardware.
引用
收藏
页数:16
相关论文
共 25 条
[1]   A Novel Approach for Multi-Label Chest X-Ray Classification of Common Thorax Diseases [J].
Allaouzi, Imane ;
Ben Ahmed, Mohamed .
IEEE ACCESS, 2019, 7 :64279-64288
[2]  
[Anonymous], 2019, ARXIV191106475
[3]  
[Anonymous], 2019, ARXIV PREPRINT ARXIV
[4]  
Cohen J.P., 2020, COVID-19 image data col- lection
[5]  
Farooq M., 2020, COVID RESNET DEEP LE, P14395
[6]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[7]   Fastai: A Layered API for Deep Learning [J].
Howard, Jeremy ;
Gugger, Sylvain .
INFORMATION, 2020, 11 (02)
[8]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[9]  
Iandola F.N., 2016, SQUEEZENET ALEXNET L
[10]  
Irvin J, 2019, AAAI CONF ARTIF INTE, P590