SARS-CoV-2 is the causative agent of COVID-19 and leaves characteristic impressions on chest Computed Tomography (CT) images in infected patients and this analysis is performed by radiologists through visual reading of lung images, and failures may occur. In this article, we propose a classification model, called Wavelet Convolutional Neural Network (WCNN) that aims to improve the differentiation of images of patients with COVID-19 from images of patients with other lung infections. The WCNN model was based on a Convolutional Neural Network (CNN) and wavelet transform. The model proposes a new input layer added to the neural network, which was called Wave layer. The hyperparameters values were defined by ablation tests. WCNN was applied to chest CT images to images from two internal and one external repositories. For all repositories, the average results of Accuracy (ACC), Sensitivity (Sen) and Specificity (Sp) were calculated. Subsequently, the average results of the repositories were consolidated, and the final values were ACC = 0.9819, Sen = 0.9783 and Sp = 0.98. The WCNN model uses a new Wave input layer, which standardizes the network input, without using data augmentation, resizing and segmentation techniques, maintaining the integrity of the tomographic image analysis. Thus, applications developed based on WCNN have the potential to assist radiologists with a second opinion in the analysis.1.
Keywords: COVID-19; Chest CT images; Convolutional neural networks; WCNN; Wavelet.
© The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2022, Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.