Deep learning-based improved snapshot ensemble technique for COVID-19 chest X-ray classification

被引:24
作者
Babu, Samson Anosh P. [1 ]
Annavarapu, Chandra Sekhara Rao [1 ]
机构
[1] Indian Inst Technol ISM, Dept Comp Sci & Engn, Dhanbad 826004, Bihar, India
基金
英国科研创新办公室;
关键词
COVID-19; Chest X-ray; Deep learning; Classification; NEURAL-NETWORKS; SEGMENTATION;
D O I
10.1007/s10489-021-02199-4
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
COVID-19 has proven to be a deadly virus, and unfortunately, it triggered a worldwide pandemic. Its detection for further treatment poses a severe threat to researchers, scientists, health professionals, and administrators worldwide. One of the daunting tasks during the pandemic for doctors in radiology is the use of chest X-ray or CT images for COVID-19 diagnosis. Time is required to inspect each report manually. While a CT scan is the better standard, an X-ray is still useful because it is cheaper, faster, and more widely used. To diagnose COVID-19, this paper proposes to use a deep learning-based improved Snapshot Ensemble technique for efficient COVID-19 chest X-ray classification. In addition, the proposed method takes advantage of the transfer learning technique using the ResNet-50 model, which is a pre-trained model. The proposed model uses the publicly accessible COVID-19 chest X-ray dataset consisting of 2905 images, which include COVID-19, viral pneumonia, and normal chest X-ray images. For performance evaluation, the model applied the metrics such as AU-ROC, AU-PR, and Jaccard Index. Furthermore, it also obtained a multi-class micro-average of 97% specificity, 95% f(1)-score, and 95% classification accuracy. The obtained results demonstrate that the performance of the proposed method outperformed those of several existing methods. This method appears to be a suitable and efficient approach for COVID-19 chest X-ray classification.
引用
收藏
页码:3104 / 3120
页数:17
相关论文
共 63 条
[1]   Classification of COVID-19 in chest X-ray images using DeTraC deep convolutional neural network [J].
Abbas, Asmaa ;
Abdelsamea, Mohammed M. ;
Gaber, Mohamed Medhat .
APPLIED INTELLIGENCE, 2021, 51 (02) :854-864
[2]   Artificial intelligence and machine learning to fight COVID-19 [J].
Alimadadi, Ahmad ;
Aryal, Sachin ;
Manandhar, Ishan ;
Munroe, Patricia B. ;
Joe, Bina ;
Cheng, Xi .
PHYSIOLOGICAL GENOMICS, 2020, 52 (04) :200-202
[3]  
[Anonymous], 2015, ACS SYM SER
[4]   Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network [J].
Anthimopoulos, Marios ;
Christodoulidis, Stergios ;
Ebner, Lukas ;
Christe, Andreas ;
Mougiakakou, Stavroula .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1207-1216
[5]   Covid-19: automatic detection from X-ray images utilizing transfer learning with convolutional neural networks [J].
Apostolopoulos, Ioannis D. ;
Mpesiana, Tzani A. .
PHYSICAL AND ENGINEERING SCIENCES IN MEDICINE, 2020, 43 (02) :635-640
[6]   Artificial intelligence in medical imaging: Game over for radiologists? [J].
Caobelli, Federico .
EUROPEAN JOURNAL OF RADIOLOGY, 2020, 126
[7]   Automated invasive ductal carcinoma detection based using deep transfer learning with whole-slide images [J].
Celik, Yusuf ;
Talo, Muhammed ;
Yildirim, Ozal ;
Karabatak, Murat ;
Acharya, U. Rajendra .
PATTERN RECOGNITION LETTERS, 2020, 133 :232-239
[8]   Can AI Help in Screening Viral and COVID-19 Pneumonia? [J].
Chowdhury, Muhammad E. H. ;
Rahman, Tawsifur ;
Khandakar, Amith ;
Mazhar, Rashid ;
Kadir, Muhammad Abdul ;
Bin Mahbub, Zaid ;
Islam, Khandakar Reajul ;
Khan, Muhammad Salman ;
Iqbal, Atif ;
Al Emadi, Nasser ;
Reaz, Mamun Bin Ibne ;
Islam, Mohammad Tariqul .
IEEE ACCESS, 2020, 8 :132665-132676
[9]   Deep learning ensembles for melanoma recognition in dermoscopy images [J].
Codella, N. C. F. ;
Nguyen, Q. -B. ;
Pankanti, S. ;
Gutman, D. A. ;
Helba, B. ;
Halpern, A. C. ;
Smith, J. R. .
IBM JOURNAL OF RESEARCH AND DEVELOPMENT, 2017, 61 (4-5)
[10]   Automatic detection of invasive ductal carcinoma in whole slide images with Convolutional Neural Networks [J].
Cruz-Roa, Angel ;
Basavanhally, Ajay ;
Gonzalez, Fabio ;
Gilmore, Hannah ;
Feldman, Michael ;
Ganesan, Shridar ;
Shih, Natalie ;
Tomaszewski, John ;
Madabhushi, Anant .
MEDICAL IMAGING 2014: DIGITAL PATHOLOGY, 2014, 9041