Iteratively Pruned Deep Learning Ensembles for COVID-19 Detection in Chest X-rays

作者: Sameer K. Antani , Les R. Folio , Sivaramakrishnan Rajaraman , Jen Siegelman , Philip O. Alderson

DOI:

关键词:

摘要: We demonstrate use of iteratively pruned deep learning model ensembles for detecting pulmonary manifestation COVID-19 with chest X-rays. This disease is caused by the novel Severe Acute Respiratory Syndrome Coronavirus 2 (SARS-CoV-2) virus, also known as (2019-nCoV). A custom convolutional neural network and a selection ImageNet pretrained models are trained evaluated at patient-level on publicly available CXR collections to learn modality-specific feature representations. The learned knowledge transferred fine-tuned improve performance generalization in related task classifying CXRs normal, showing bacterial pneumonia, or COVID-19-viral abnormalities. best performing reduce complexity memory efficiency. predictions best-performing combined through different ensemble strategies classification performance. Empirical evaluations that weighted average significantly improves resulting an accuracy 99.01% area under curve 0.9972 findings CXRs. transfer, iterative pruning, resulted improved predictions. expect this can be quickly adopted screening using radiographs.

参考文章(41)
MaclinRichard, OpitzDavid, Popular ensemble methods Journal of Artificial Intelligence Research. ,(1999) , 10.5555/3013545.3013549
Xiaosong Wang, Yifan Peng, Le Lu, Zhiyong Lu, Mohammadhadi Bagheri, Ronald M. Summers, ChestX-Ray8: Hospital-Scale Chest X-Ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases computer vision and pattern recognition. pp. 3462- 3471 ,(2017) , 10.1109/CVPR.2017.369
Ramprasaath R Selvaraju, Michael Cogswell, Abhishek Das, Ramakrishna Vedantam, Devi Parikh, Dhruv Batra, None, Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization International Journal of Computer Vision. ,vol. 128, pp. 336- 359 ,(2020) , 10.1007/S11263-019-01228-7
Linda Wang, Zhong Qiu Lin, Alexander Wong, COVID-Net: a tailored deep convolutional neural network design for detection of COVID-19 cases from chest X-ray images. Scientific Reports. ,vol. 10, pp. 19549- 19549 ,(2020) , 10.1038/S41598-020-76550-Z
Thomas G. Dietterich, Ensemble Methods in Machine Learning Multiple Classifier Systems. pp. 1- 15 ,(2000) , 10.1007/3-540-45014-9_1
Karen Simonyan, Andrew Zisserman, Very Deep Convolutional Networks for Large-Scale Image Recognition computer vision and pattern recognition. ,(2014)
Philipp Fischer, Thomas Brox, None, U-Net: Convolutional Networks for Biomedical Image Segmentation medical image computing and computer assisted intervention. pp. 234- 241 ,(2015) , 10.1007/978-3-319-24574-4_28
Sema Candemir, Sameer Antani, Stefan Jaeger, Renee Browning, George R. Thoma, Lung boundary detection in pediatric chest x-rays Proceedings of SPIE. ,vol. 9418, ,(2015) , 10.1117/12.2081060
Ilya Sutskever, Geoffrey Hinton, Alex Krizhevsky, Ruslan Salakhutdinov, Nitish Srivastava, Dropout: a simple way to prevent neural networks from overfitting Journal of Machine Learning Research. ,vol. 15, pp. 1929- 1958 ,(2014)
M Sai Praneeth, Xudong Peng, Alice Li, Shahrzad Hosseini Vajargah, Going deeper with convolutions computer vision and pattern recognition. pp. 1- 9 ,(2015) , 10.1109/CVPR.2015.7298594