Detection of Chest X-ray Abnormalities Using CNN Based on Hyperparameter Optimization

Authors:
Open Access
Publisher website
Google Scholar
Abstract
The chest X-ray (CXR) is a commonly used diagnostic imaging test that requires significant expertise and careful observation due to the complex nature of the pathology and fine texture of lung lesions. Despite the long-term clinical training and professional guidance provided to radiologists, there is still the possibility of errors in diagnosis. Therefore, we have developed a novel approach using a convolutional neural network (CNN) model to detect the abnormalities of CXR images. The model was optimized using algorithms such as Adam and RMSprop. Also, several hyperparameters were optimized, including the pooling layer, convolutional layer, dropout layer, target size, and epochs. Hyperparameter optimization aims to improve the model’s accuracy by testing various combinations of hyperparameter values and optimization algorithms. To evaluate the model’s performance, we used scenario modeling to create 32 models and tested them using a confusion matrix. The results indicated that the best accuracy achieved by the model was 97.94%. This accuracy was based on training and test data using 4538 CXR images. The findings suggest that hyperparameter optimization can improve the CNN model’s accuracy in accurately identifying CXR abnormalities. Therefore, this study has important implications for improving the accuracy and reliability of CXR image interpretation, which could ultimately benefit patients by improving the detection and treatment of lung diseases. Acknowledging dataset constraints, we address future steps for model improvement.
Funding Information
  • Polish Ministry of Education and Science assigned to AGH University of Krakow (PLG/2023/016757)

This publication has 33 references indexed in Scilit: