CovXNet
CovXNet classifies chest X-ray images to detect COVID-19 and differentiate viral, bacterial, and normal cases using a multi-dilation convolutional neural network.
Key Features:
- Multi-dilation depthwise convolution: Architecture built around depthwise convolution with varying dilation rates to extract multi-scale features from chest X-ray images.
- Transfer learning and fine-tuning: Initial training on a large dataset of normal and traditional (viral/bacterial) pneumonia chest X-rays followed by fine-tuning on a smaller COVID-19-specific chest X-ray set.
- Multi-resolution network variants: Multiple forms of the network are designed and optimized for different image resolutions.
- Stacking ensemble: A stacking algorithm combines outputs from diverse architectures to enhance prediction accuracy.
- Gradient-based discriminative localization: Integration of a gradient-based discriminative localization technique to identify and distinguish abnormal regions associated with different pneumonia types.
- Performance: Reported accuracies of 97.4% for COVID-19 vs normal, 96.9% for COVID-19 vs viral pneumonia, 94.7% for COVID-19 vs bacterial pneumonia, and 90.2% for multiclass differentiation among COVID-19, normal, viral, and bacterial pneumonias.
Scientific Applications:
- Automated classification: Detection and differentiation of COVID-19, viral pneumonia, bacterial pneumonia, and normal cases from chest X-ray images.
- Abnormality localization: Identification and localization of abnormal regions in chest X-rays corresponding to different types of pneumonia.
- Model evaluation on chest X-rays: Assessment of deep learning classification performance using two distinct chest X-ray datasets.
Methodology:
Depthwise convolution with varying dilation rates; initial training on a large normal and traditional pneumonia chest X-ray dataset followed by transfer learning via fine-tuning on a smaller COVID-19 chest X-ray set; design of multiple network variants for different resolutions; stacking of model outputs; integration of gradient-based discriminative localization; evaluation on two distinct datasets.
Topics
Collections
Details
- License:
- Apache-2.0
- Tool Type:
- command-line tool
- Programming Languages:
- Python
- Added:
- 1/18/2021
- Last Updated:
- 2/18/2021
Operations
Publications
Mahmud T, Rahman MA, Fattah SA. CovXNet: A multi-dilation convolutional neural network for automatic COVID-19 and other pneumonia detection from chest X-ray images with transferable multi-receptive feature optimization. Computers in Biology and Medicine. 2020;122:103869. doi:10.1016/j.compbiomed.2020.103869. PMID:32658740. PMCID:PMC7305745.