About the journal

• Home page
• Contact information
• Aims and scope
• Indexing information
• Editorial policies
• ComSIS consortium
• Journal boards
• Managing board

For authors

• Information for contributors
• Paper submission
• Article submission through OJS
• Copyright transfer form
• Download section

For readers

• Forthcoming articles
• Current issue
• Archive

For reviewers

• View and review submissions

News

• Journal's Facebook page
• Calls for special issues
• New issue notification

Hyper-parameter Optimization of Convolutional Neural Networks for Classifying COVID-19 X-ray Images

Grega Vrbančič¹, Špela Pečnik¹ and Vili Podgorelec¹

1. University of Maribor, Faculty of Electrical Engineering and Computer Science
   Koroška cesta 46, SI-2000 Maribor, Slovenia
   grega.vrbancic@um.si

Abstract

For more than a year the COVID-19 epidemic is threatening people all over the world. Numerous researchers are looking for all possible insights into the new corona virus SARS-CoV-2. One of the possibilities is an in-depth analysis of X-ray images from COVID-19 patients, commonly conducted by a radiologist, which are due to high demand facing with overload. With the latest achievements in the field of deep learning, the approaches using transfer learning proved to be successful when tackling such problem. However, when utilizing deep learning methods, we are commonly facing the problem of hyper-parameter settings. In this research, we adapted and generalized transfer learning based classification method for detecting COVID-19 from X-ray images and employed different optimization algorithms for solving the task of hyper-parameter settings. Utilizing different optimization algorithms our method was evaluated on a dataset of 1446 X-ray images, with the overall accuracy of 84.44%, outperforming both conventional CNN method as well as the compared baseline transfer learning method. Besides quantitative analysis, we also conducted a qualitative in-depth analysis using the local interpretable modelagnostic explanations method and gain some in-depth view of COVID-19 characteristics and the predictive model perception.

Key words

COVID-19, classification, CNN, transfer learning, optimization

Digital Object Identifier (DOI)

https://doi.org/10.2298/CSIS210209056V

Publication information

Volume 19, Issue 1 (January 2022)
Year of Publication: 2022
ISSN: 2406-1018 (Online)
Publisher: ComSIS Consortium

Full text

Available in PDF
Portable Document Format

How to cite

Vrbančič, G., Pečnik, Š., Podgorelec, V.: Hyper-parameter Optimization of Convolutional Neural Networks for Classifying COVID-19 X-ray Images. Computer Science and Information Systems, Vol. 19, No. 1, 327-352. (2022), https://doi.org/10.2298/CSIS210209056V