StackNet-DenVIS: a multi-layer perceptron stacked ensembling approach for COVID-19 detection using X-ray images

被引:21
作者
Autee, Pratik [1 ]
Bagwe, Sagar [1 ]
Shah, Vimal [1 ,2 ]
Srivastava, Kriti [1 ]
机构
[1] Dwarkadas J Sanghvi Coll Engn, Dept Comp Engn, Mumbai, Maharashtra, India
[2] A-602 Venkatesh Pooja,Balaji Complex,150 Feet Rd, Thana 401101, Maharashtra, India
关键词
Covid-19; Stacked generalization; Transfer learning; Deep neural networks; Generative adversarial networks; Image segmentation; LUNG SEGMENTATION;
D O I
10.1007/s13246-020-00952-6
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
The highly contagious nature of Coronavirus disease 2019 (Covid-19) resulted in a global pandemic. Due to the relatively slow and taxing nature of conventional testing for Covid-19, a faster method needs to be in place. The current researches have suggested that visible irregularities found in the chest X-ray of Covid-19 positive patients are indicative of the presence of the disease. Hence, Deep Learning and Image Classification techniques can be employed to learn from these irregularities, and classify accordingly with high accuracy. This research presents an approach to create a classifier model named StackNet-DenVIS which is designed to act as a screening process before conducting the existing swab tests. Using a novel approach, which incorporates Transfer Learning and Stacked Generalization, the model aims to lower the False Negative rate of classification compensating for the 30% False Negative rate of the swab tests. A dataset gathered from multiple reliable sources consisting of 9953 Chest X-rays (868 Covid and 9085 Non-Covid) was used. Also, this research demonstrates handling data imbalance using various techniques involving Generative Adversarial Networks and sampling techniques. The accuracy, sensitivity, and specificity obtained on our proposed model were 95.07%, 99.40% and 94.61% respectively. To the best of our knowledge, the combination of accuracy and false negative rate obtained by this paper outperforms the current implementations. We must also highlight that our proposed architecture also considers other types of viral pneumonia. Given the unprecedented sensitivity of our model we are optimistic it contributes to a better Covid-19 detection.
引用
收藏
页码:1399 / 1414
页数:16
相关论文
共 39 条
[31]  
Szegedy SI, 2016, ARXIV160207261
[32]   Unveiling COVID-19 from CHEST X-Ray with Deep Learning: A Hurdles Race with Small Data [J].
Tartaglione, Enzo ;
Barbano, Carlo Alberto ;
Berzovini, Claudio ;
Calandri, Marco ;
Grangetto, Marco .
INTERNATIONAL JOURNAL OF ENVIRONMENTAL RESEARCH AND PUBLIC HEALTH, 2020, 17 (18) :1-17
[33]   2 MODIFICATIONS OF CNN [J].
TOMEK, I .
IEEE TRANSACTIONS ON SYSTEMS MAN AND CYBERNETICS, 1976, 6 (11) :769-772
[34]   CovidGAN: Data Augmentation Using Auxiliary Classifier GAN for Improved Covid-19 Detection [J].
Waheed, Abdul ;
Goyal, Muskan ;
Gupta, Deepak ;
Khanna, Ashish ;
Al-Turjman, Fadi ;
Pinheiro, Placido Rogerio .
IEEE ACCESS, 2020, 8 :91916-91923
[35]   Spatiotemporal Pyramid Network for Video Action Recognition [J].
Wang, Yunbo ;
Long, Mingsheng ;
Wang, Jianmin ;
Yu, Philip S. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2097-2106
[36]   STACKED GENERALIZATION [J].
WOLPERT, DH .
NEURAL NETWORKS, 1992, 5 (02) :241-259
[37]  
Wong A, 2020, SCI REP-UK, DOI DOI 10.1038/S41598-020-76550-Z
[38]   Computer aided detection of SARS based on radiographs data mining [J].
Xie Xuanyang ;
Gong Yuchang ;
Wan Shouhong ;
Li Xi .
2005 27TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY, VOLS 1-7, 2005, :7459-7462
[39]   Coronavirus Disease 2019 (COVID-19) A Perspective from China [J].
Zu, Zi Yue ;
Jiang, Meng Di ;
Xu, Peng Peng ;
Chen, Wen ;
Ni, Qian Qian ;
Lu, Guang Ming ;
Zhang, Long Jiang .
RADIOLOGY, 2020, 296 (02) :E15-E25