UNAS-Net: A deep convolutional neural network for predicting Covid-19 severity

Abstract

We present a study on Covid-19 detection using deep learning algorithms that help predict and detect Covid-19. Chest X-ray images were used as the input dataset to prepare and train the proposed model. In this context, deep learning architecture (DLA) and optimisation strategies have been proposed and explored to support the automated detection of Covid-19. A model based on a convolutional neural network was proposed to extract features of images for the feature-learning phase. Data augmentation and fine-tuning with deep-feature-based methods were applied to improve the model. Image enhancement and saliency maps were used to enhance visualisation and estimate the disease severity level based on two parameters; degree of opacity and geographic extent. Contrast-limited adaptive histogram equalization and Otsu thresholding were employed with several parameters to investigate the effects on the visualisation results. An experimental investigation was performed between the proposed method and other pretrained DLAs. The proposed work obtained excellent classification accuracy and sensitivity of 97.36% and 95.24% respectively. In addition, the input parameters for image enhancement significantly affected the results. The overall performance metrics were perfect for DenseNet and adequately high for the proposed work which is comparable to other models. Data augmentation and fine-tuning successfully handed the networks to enhance the overall performance, especially in our case with limited datasets.

Keywords

• Deep learning
• Covid-19
• Image enhancement