Back

Late-Ensemble of Convolutional Neural Networks with Test Time Augmentation for Chest XR COVID-19 Detection

Qayyum, A.; Razzak, I.; Mazher, M.; Puig, D.

2022-02-26 health informatics
10.1101/2022.02.25.22271520
Show abstract

COVID-19, a severe acute respiratory syndrome aggressively spread among global populations in just a few months. Since then, it has had four dominant variants (Alpha, Beta, Gamma and Delta) that are far more contagious than original. Accurate and timely diagnosis of COVID-19 is critical for analysis of damage to lungs, treatment, as well as quarantine management [7]. CT, MRI or X-rays image analysis using deep learning provide an efficient and accurate diagnosis of COVID-19 that could help to counter its outbreak. With the aim to provide efficient multi-class COVID-19 detection, recently, COVID-19 Detection challenge using X-ray is organized [12]. In this paper, the late-fusion of features is extracted from pre-trained various convolutional neural networks and fine-tuned these models using the challenge dataset. The DensNet201 with Adam optimizer and EffecientNet-B3 are fine-tuned on the challenge dataset and ensembles the features to get the final prediction. Besides, we also considered the test time augmentation technique after the late-ensembling approach to further improve the performance of our proposed solution. Evaluation on Chest XR COVID-19 showed that our model achieved overall accuracy is 95.67%. We made the code is publicly available1. The proposed approach was ranked 6th in Chest XR COVID-19 detection Challenge [1].

Matching journals

The top 7 journals account for 50% of the predicted probability mass.

1
Informatics in Medicine Unlocked
based on 11 papers
Top 0.1%
12.4%
2
Scientific Reports
based on 701 papers
Top 24%
7.5%
3
Frontiers in Artificial Intelligence
based on 11 papers
Top 0.1%
7.5%
4
PLOS Digital Health
based on 88 papers
Top 2%
7.5%
5
Patterns
based on 15 papers
Top 0.1%
6.3%
6
IEEE Journal of Biomedical and Health Informatics
based on 14 papers
Top 0.1%
5.3%
7
Computers in Biology and Medicine
based on 39 papers
Top 1.0%
5.0%
50% of probability mass above
8
JMIR Medical Informatics
based on 16 papers
Top 1.0%
4.7%
9
BMC Medical Informatics and Decision Making
based on 36 papers
Top 3%
4.4%
10
Journal of Biomedical Informatics
based on 37 papers
Top 2%
2.8%
11
PLOS ONE
based on 1737 papers
Top 80%
2.8%
12
International Journal of Medical Informatics
based on 25 papers
Top 2%
2.8%
13
Journal of Medical Internet Research
based on 81 papers
Top 7%
2.4%
14
IEEE Access
based on 11 papers
Top 0.7%
2.3%
15
Chaos, Solitons & Fractals
based on 17 papers
Top 2%
1.6%
16
npj Digital Medicine
based on 85 papers
Top 11%
1.3%
17
Computer Methods and Programs in Biomedicine
based on 12 papers
Top 0.8%
1.3%
18
Nature Communications
based on 483 papers
Top 35%
1.3%
19
JMIR Public Health and Surveillance
based on 45 papers
Top 8%
1.3%
20
JAMIA Open
based on 35 papers
Top 5%
1.3%
21
Biology Methods and Protocols
based on 19 papers
Top 1%
1.2%
22
Journal of the American Medical Informatics Association
based on 53 papers
Top 6%
1.2%
23
Applied Sciences
based on 10 papers
Top 1%
0.8%
24
Sensors
based on 18 papers
Top 4%
0.7%
25
Frontiers in Psychiatry
based on 56 papers
Top 8%
0.7%
26
Scientific Data
based on 30 papers
Top 4%
0.7%