Automated medical diagnosis of COVID-19 through EfficientNet convolutional neural network

• Published in: Applied soft computing Vol. 96; p. 106691
• Main Authors: Marques, Gonçalo, Agarwal, Deevyankar, de la Torre Díez, Isabel
• Format: Journal Article
• Language: English
• Published: United States Elsevier B.V 01.11.2020
• Subjects: Automated decision support system; Convolutional Neural Network (CNN); COVID-19; Deep learning; Machine learning; COVID-19; Deep learning; Automated decision support system; Machine learning; Convolutional Neural Network (CNN)
• DOI: 10.1016/j.asoc.2020.106691
• ISSN: 1568-4946

COVID-19 infection was reported in December 2019 at Wuhan, China. This virus critically affects several countries such as the USA, Brazil, India and Italy. Numerous research units are working at their higher level of effort to develop novel methods to prevent and control this pandemic scenario. The main objective of this paper is to propose a medical decision support system using the implementation of a convolutional neural network (CNN). This CNN has been developed using EfficientNet architecture. To the best of the authors’ knowledge, there is no similar study that proposes an automated method for COVID-19 diagnosis using EfficientNet. Therefore, the main contribution is to present the results of a CNN developed using EfficientNet and 10-fold stratified cross-validation. This paper presents two main experiments. First, the binary classification results using images from COVID-19 patients and normal patients are shown. Second, the multi-class results using images from COVID-19, pneumonia and normal patients are discussed. The results show average accuracy values for binary and multi-class of 99.62% and 96.70%, respectively. On the one hand, the proposed CNN model using EfficientNet presents an average recall value of 99.63% and 96.69% concerning binary and multi-class, respectively. On the other hand, 99.64% is the average precision value reported by binary classification, and 97.54% is presented in multi-class. Finally, the average F1-score for multi-class is 97.11%, and 99.62% is presented for binary classification. In conclusion, the proposed architecture can provide an automated medical diagnostics system to support healthcare specialists for enhanced decision making during this pandemic scenario. •An automated system to support the diagnosis of COVID-19 patients using EfficientNet.•The proposed method has been validated using 10-fold cross-validation.•An external dataset also has been used for validation.•An accuracy of 99.62% and 96.70% is reported for binary and multi-class classification.