Comparing different deep learning architectures for classification of chest radiographs

被引:95
作者
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
    Allaouzi, Imane
    Ben Ahmed, Mohamed
    [J]. IEEE ACCESS, 2019, 7 : 64279 - 64288
  • [2] [Anonymous], 2019, ARXIV191106475
  • [3] Bustos A., 2019, ARXIV PREPRINT ARXIV
  • [4] Cohen J.P., 2020, arXiv2006.11988
  • [5] Farooq M., 2020, arXiv preprint arXiv:2003
  • [6] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [7] Fastai: A Layered API for Deep Learning
    Howard, Jeremy
    Gugger, Sylvain
    [J]. INFORMATION, 2020, 11 (02)
  • [8] Densely Connected Convolutional Networks
    Huang, Gao
    Liu, Zhuang
    van der Maaten, Laurens
    Weinberger, Kilian Q.
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 2261 - 2269
  • [9] Iandola F. N., 2016, arXiv
  • [10] Irvin J, 2019, AAAI CONF ARTIF INTE, P590