A Deep Convolutional Neural Network for COVID-19 Detection Using Chest
X-Rays
- URL: http://arxiv.org/abs/2005.01578v4
- Date: Wed, 13 Jan 2021 04:08:10 GMT
- Title: A Deep Convolutional Neural Network for COVID-19 Detection Using Chest
X-Rays
- Authors: Pedro R. A. S. Bassi, Romis Attux
- Abstract summary: We present image classifiers based on Dense Convolutional Networks and transfer learning to classify chest X-ray images according to three labels: COVID-19, pneumonia and normal.
We were able to reach test accuracy of 100% on our test dataset.
- Score: 2.2843885788439797
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Purpose: We present image classifiers based on Dense Convolutional Networks
and transfer learning to classify chest X-ray images according to three labels:
COVID-19, pneumonia and normal.
Methods: We fine-tuned neural networks pretrained on ImageNet and applied a
twice transfer learning approach, using NIH ChestX-ray14 dataset as an
intermediate step. We also suggested a novelty called output neuron keeping,
which changes the twice transfer learning technique. In order to clarify the
modus operandi of the models, we used Layer-wise Relevance Propagation (LRP) to
generate heatmaps.
Results: We were able to reach test accuracy of 100% on our test dataset.
Twice transfer learning and output neuron keeping showed promising results
improving performances, mainly in the beginning of the training process.
Although LRP revealed that words on the X-rays can influence the networks'
predictions, we discovered this had only a very small effect on accuracy.
Conclusion: Although clinical studies and larger datasets are still needed to
further ensure good generalization, the state-of-the-art performances we
achieved show that, with the help of artificial intelligence, chest X-rays can
become a cheap and accurate auxiliary method for COVID-19 diagnosis. Heatmaps
generated by LRP improve the interpretability of the deep neural networks and
indicate an analytical path for future research on diagnosis. Twice transfer
learning with output neuron keeping improved performances.
Related papers
- Performance of GAN-based augmentation for deep learning COVID-19 image
classification [57.1795052451257]
The biggest challenge in the application of deep learning to the medical domain is the availability of training data.
Data augmentation is a typical methodology used in machine learning when confronted with a limited data set.
In this work, a StyleGAN2-ADA model of Generative Adversarial Networks is trained on the limited COVID-19 chest X-ray image set.
arXiv Detail & Related papers (2023-04-18T15:39:58Z) - Randomly Initialized Convolutional Neural Network for the Recognition of
COVID-19 using X-ray Images [0.0]
coronavirus disease (COVID-19) has been declared a worldwide pandemic.
One potential solution to detect COVID-19 is by analyzing the chest X-ray images using Deep Learning (DL) models.
In this study, we propose a novel randomly CNN architecture for the recognition of COVID-19.
The proposed CNN model yields encouraging results with 94% and 99% of accuracy for COVIDx and enhanced COVID-19 dataset, respectively.
arXiv Detail & Related papers (2021-05-17T23:40:37Z) - Many-to-One Distribution Learning and K-Nearest Neighbor Smoothing for
Thoracic Disease Identification [83.6017225363714]
deep learning has become the most powerful computer-aided diagnosis technology for improving disease identification performance.
For chest X-ray imaging, annotating large-scale data requires professional domain knowledge and is time-consuming.
In this paper, we propose many-to-one distribution learning (MODL) and K-nearest neighbor smoothing (KNNS) methods to improve a single model's disease identification performance.
arXiv Detail & Related papers (2021-02-26T02:29:30Z) - Twin Augmented Architectures for Robust Classification of COVID-19 Chest
X-Ray Images [6.127080932156285]
Gold standard for COVID-19 is RT-PCR, testing facilities for which are limited and not always optimally distributed.
We show that popular choices of dataset selection suffer from data homogeneity, leading to misleading results.
We introduce a state-of-the-art technique, termed as Twin Augmentation, for modifying popular pre-trained deep learning models.
arXiv Detail & Related papers (2021-02-16T06:50:17Z) - A Multi-Stage Attentive Transfer Learning Framework for Improving
COVID-19 Diagnosis [49.3704402041314]
We propose a multi-stage attentive transfer learning framework for improving COVID-19 diagnosis.
Our proposed framework consists of three stages to train accurate diagnosis models through learning knowledge from multiple source tasks and data of different domains.
Importantly, we propose a novel self-supervised learning method to learn multi-scale representations for lung CT images.
arXiv Detail & Related papers (2021-01-14T01:39:19Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - COVID-CLNet: COVID-19 Detection with Compressive Deep Learning
Approaches [0.0]
We propose a computer-aided detection (CADe) system that uses the computed tomography (CT) scan images.
This proposed boosted deep learning network (CLNet) is based on the implementation of Deep Learning (DL) networks.
Experiments performed on different compressed methods show promising results for COVID-19 detection.
arXiv Detail & Related papers (2020-12-03T19:56:48Z) - Triage of Potential COVID-19 Patients from Chest X-ray Images using
Hierarchical Convolutional Networks [5.7179132552879395]
The current COVID-19 pandemic has motivated the researchers to use artificial intelligence techniques for a potential alternative to reverse transcription-polymerase chain reaction (RT-PCR)
The chest X-ray (CXR) is one of the alternatives to achieve fast diagnosis but the unavailability of large-scale annotated data makes the clinical implementation of machine learning-based COVID detection difficult.
In this paper, we propose the use of hierarchical convolutional network (HCN) architecture to naturally augment the data along with diversified features.
arXiv Detail & Related papers (2020-11-01T20:01:22Z) - Classification of COVID-19 in CT Scans using Multi-Source Transfer
Learning [91.3755431537592]
We propose the use of Multi-Source Transfer Learning to improve upon traditional Transfer Learning for the classification of COVID-19 from CT scans.
With our multi-source fine-tuning approach, our models outperformed baseline models fine-tuned with ImageNet.
Our best performing model was able to achieve an accuracy of 0.893 and a Recall score of 0.897, outperforming its baseline Recall score by 9.3%.
arXiv Detail & Related papers (2020-09-22T11:53:06Z) - Learning Invariant Feature Representation to Improve Generalization
across Chest X-ray Datasets [55.06983249986729]
We show that a deep learning model performing well when tested on the same dataset as training data starts to perform poorly when it is tested on a dataset from a different source.
By employing an adversarial training strategy, we show that a network can be forced to learn a source-invariant representation.
arXiv Detail & Related papers (2020-08-04T07:41:15Z) - Data Augmentation using Generative Adversarial Networks (GANs) for
GAN-based Detection of Pneumonia and COVID-19 in Chest X-ray Images [0.0]
We propose a new GAN architecture for augmentation of chest X-rays for semi-supervised detection of pneumonia and COVID-19.
We show that the proposed GAN can be used to effectively augment data and improve classification accuracy of disease in chest X-rays for pneumonia and COVID-19.
arXiv Detail & Related papers (2020-06-05T18:30:20Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.