Skip to main content

ORIGINAL RESEARCH article

Front. Med.
Sec. Pulmonary Medicine
Volume 11 - 2024 | doi: 10.3389/fmed.2024.1498403

Optimization of Convolutional Neural Network and Visual Geometry Group-16 using Genetic Algorithms for Pneumonia detection

Provisionally accepted
  • 1 University of Hail, Ha'il, Saudi Arabia
  • 2 University of Sfax, Sfax, Tunisia

The final, formatted version of the article will be published soon.

    Pneumonia is still a major global health issue, so effective diagnostic methods are needed. This research proposes a new methodology for improving Convolutional Neural Networks (CNNs) and the Visual Geometry Group-16 (VGG16) model by incorporating Genetic algorithms (GAs) to detect pneumonia. The work uses a dataset of 5856 frontal chest radiography images critical in training and testing machine learning algorithms. The issue relates to challenges of medical image classification, the complexity of which can be significantly addressed by properly optimizing CNN. Moreover, our proposed methodology used GAs to determine the hyperparameters for CNNs and VGG16 and fine-tune the architecture to improve the existing performance measures. The evaluation of the optimized models showed some good performances with purely Convolutional Neural Network archetyping, averaging 97% in terms of training accuracy and 94% based on the testing process. At the same time, it has a low error rate of 0.072. Although adding this layer affected the training and testing time, it created a new impression on the test accuracy and training accuracy of the VGG16 model, with 90.90% training accuracy, 90.90%, and a loss of 0.11. Future work will involve contributing more examples so that a richer database of radiographic images is attained, optimizing the GA parameters even more, and pursuing the use of ensemble applications so that the diagnosis capability is heightened. Apart from emphasizing the contribution of GAs in improving the CNN architecture, this study also seeks to contribute to the early detection of pneumonia to minimize the complications faced by patients, especially children.

    Keywords: Pneumonia, deep learning, Convolutional Neural Network, Genetic Algorithm, Visual Geometry Group-16

    Received: 18 Sep 2024; Accepted: 11 Nov 2024.

    Copyright: © 2024 Chihaoui, Dhibi and Ferchichi. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence: Mejda Chihaoui, University of Hail, Ha'il, Saudi Arabia

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.