A study on the suitability of different pooling operators for Convolutional Neural Networks in the prediction of COVID-19 through chest x-ray image analysis

Abstract

The 2019 coronavirus disease outbreak, caused by the severe acute respiratory syndrome type-2 virus (SARS-CoV-2), was declared a pandemic in March 2020. Since its emergence to the present day, this disease has brought multiple countries to the brink of health care collapse during several waves of the disease. One of the most common tests performed on patients is chest x-ray imaging. These images show the severity of the patient’s illness and whether it is indeed covid or another type of pneumonia. Automated assessment of this type of imaging could alleviate the time required for physicians to treat and diagnose each patient. To this end, in this paper we propose the use of Convolutional Neural Networks (CNNs) to carry out this process. The aim of this paper is twofold. Firstly, we present a pipeline adapted to this problem, covering all steps from the preprocessing of the datasets to the generation of classification models based on CNNs. Secondly, we have focused our study on the modification of the information fusion processes of this type of architectures, in the pooling layers. We propose a number of aggregation theory functions that are suitable to replace classical processes and have shown their benefits in past applications, and study their performance in the context of the x-ray classification problem. We find that replacing the feature reduction processes of CNNs leads to drastically different behaviours of the final model, which can be beneficial when prioritizing certain metrics such as precision or recall.