A novel data augmentation based on Gabor filter and convolutional deep learning for improving the classification of COVID-19 chest X-Ray images

Abstract

A dangerous infectious disease of the current century, the COVID-19 has apparently originated in a city in China and turned into a widespread pandemic within a short time. In this paper, a novel method has been presented for improving the screening and classification of COVID-19 patients based on their chest X-Ray (CXR) images. This method eliminates the severe dependence of the deep learning models on large datasets and the deep features extracted from them. In this approach, we have not only resolved the data limitation problem by combining the traditional data augmentation techniques with the generative adversarial networks (GANs), but also have enabled a deeper extraction of features by applying different filter banks such as the Sobel, Laplacian of Gaussian (LoG) and the Gabor filters. To verify the satisfactory performance of the proposed approach, it was applied on several deep transfer models and the results in each step were compared with each other. For training the entire models, we used 4560 CXR images of various patients with the viral, bacterial, fungal, and other diseases; 360 of these images are in the COVID-19 category and the rest belong to the non-COVID-19 diseases. According to the results, the Gabor filter bank achieves the highest growth in the values of the defined evaluation criteria and in just 45 epochs, it is able to elevate the accuracy by up to 32%. We then applied the proposed model on the DenseNet-201 model and compared its performance in terms of the detection accuracy with the performances of 10 existing COVID-19 detection techniques. Our approach was able to achieve an accuracy of 98.5% in the two-class classification procedure; which makes it a state-of-the-art method for detecting the COVID-19.