Aguirre Olea, Pablo Leopoldo

2024-01-01, Solar, Mauricio, Aguirre Olea, Pablo Leopoldo

The idea of using X–rays and Computed Tomography (CT) images as diagnostic method has been explored in several studies. Most of these studies work with slices of CT image in 2D, requiring less computational capacity and less time to process them than 3D. The processing of volumetric data (the complete CT images in 3D) adds an extra dimension of information. However, the magnitude of the data is considerably larger than working with slices in 2D, so extra computational processing is required. In this study a model capable of performing a classification of a 3D input that represents the volume of the CT scan is proposed. The model is able to classify the 3D input between COVID–19 and Non–COVID–19, but reducing the use of resources when performing the classification. The proposed model is the ResNet–50 model with a new dimension of information added, which is a simple autoencoder. This autoencoder is trained on the same dataset, and a vector representation of each exam is generated and used together with the exams to feed the ResNet–50. To validate the proposal, the same proposed model is compared with and without the autoencoder module that provides more information to the proposed model. The proposed model obtains better metrics than the same model without the autoencoder, confirming that extracting relevant features from the dataset helps improve the performance of the model.