Comparison of deep learning models based on Chest X-ray image classification

Yiqing Zhang; Yukun Xu; Zhengyang Kong; Zheqi Hu

doi:10.54254/2755-2721/64/20241352

Research Article

Open access

Published on 15 May 2024

Download pdf

Zhang,Y.;Xu,Y.;Kong,Z.;Hu,Z. (2024). Comparison of deep learning models based on Chest X-ray image classification. Applied and Computational Engineering,64,55-62.

Export citation

Comparison of deep learning models based on Chest X-ray image classification

Yiqing Zhang *^,1, Yukun Xu ², Zhengyang Kong ³, Zheqi Hu ⁴

¹ SDU-ANU Joint Science College, Shandong University
² SDU-ANU Joint Science College, Shandong University
³ School of Electronic Engineering, Xi'an University of Posts and Telecommunications
⁴ University of Electronic Science and technology of China

* Author to whom correspondence should be addressed.

https://doi.org/10.54254/2755-2721/64/20241352

Abstract

Pneumonia is a common respiratory disease characterized by inflammation in the lungs, emphasizing the importance of accurate diagnosis and timely treatment. Despite some progress in medical image segmentation, overfitting and low efficiency have been observed in practical applications. This paper aims to leverage image data augmentation methods to mitigate overfitting and achieve lightweight and highly accurate automatic detection of lung infections in X-ray images. We trained three models, namely VGG16, MobileNetV2, and InceptionV3, using both augmented and unaugmented image datasets. Comparative results demonstrate that the augmented VGG16 model (VGG16-Augmentation) achieves an average accuracy of 96.8%. While the accuracy of MobileNetV2-Augmentation is slightly lower than that of VGG16-Augmentation, it still achieves an average prediction accuracy of 94.2% and the number of model parameters is only 1/9 of VGG16-augmentation. This is particularly beneficial for rapid screening of pneumonia patients and more efficient real-time detection scenarios. Through this study, we showcase the potential application of image data augmentation methods in pneumonia detection and provide performance comparisons among different models. These findings offer valuable insights for the rapid diagnosis and screening of pneumonia patients and provide useful guidance for future research and the implementation of efficient real-time monitoring of lung conditions in practical healthcare settings.

Keywords

Chest X-Ray Images, Data Augmentation, VGG16, MobileNetV2, InceptionV3

View pdf

References

[1]. A. Krizhevsky, I. Sutskever, and G. E. Hinton (2012). Imagenet classification with deep convolutional neural networks. Communications of the ACM, 60, 84–90. https://doi.org/10.1145/3065386

[2]. Lawpanom, R.; Songpan, W.; Kaewyotha, J. (2024). Advancing Facial Expression Recognition in Online Learning Education Using a Homogeneous Ensemble Convolutional Neural Network Approach. Applied Sciences, 14(3). https://doi.org/10.3390/app14031156

[3]. Gulzar, Y. (2023). Fruit Image Classification Model Based on MobileNetV2 with Deep Transfer Learning Technique. Sustainability, 15(3). https://doi.org/10.3390/su15031906

[4]. Qian. (2023). Performance comparison among VGG16, InceptionV3, and resnet on galaxy morphology classification. Journal of Physics, 2580. https://dx.doi.org/10.1088/1742-6596/2580/1/012009

[5]. Rahman, T.; Chowdhury, M.E.H.; Khandakar, A.; Islam, K.R.; Islam, K.F.; Mahbub, Z.B.; Kadir, M.A.; Kashem, S. (2020). Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection Using Chest X-ray. Applied Sciences, 10(9). https://doi.org/10.3390/app10093233

[6]. Kim S, Rim B, Choi S, Lee A, Min S, Hong M. (2022). Deep Learning in Multi-Class Lung Diseases' Classification on Chest X-ray Images. Diagnostics (Basel), 12(4). https://doi.org/10.3390/diagnostics12040915

[7]. E. Ayan and H. M. Ünver (2019). Diagnosis of Pneumonia from Chest X-Ray Images Using Deep Learning. Scientific Meeting on Electrical-Electronics & Biomedical Engineering and Computer Science (EBBT), 1-5. https://ieeexplore.ieee.org/document/8741582

[8]. Apostolopoulos, I.D., Mpesiana, T.A. (2020). Covid-19: automatic detection from X-ray images utilizing transfer learning with convolutional neural networks. Phys Eng Sci Med, 43, 635–640. https://link.springer.com/article/10.1007/s13246-020-00865-4

[9]. Kermany, Daniel S., Kang Zhang and Michael H. Goldbaum(2018). Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images for Classification. https://api.semanticscholar.org/CorpusID:126183849

[10]. C. Szegedy, V. Vanhoucke, S. Ioffe, J. Shlens and Z. Wojna (2016). Rethinking the Inception Architecture for Computer Vision. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2818-2826. https://ieeexplore.ieee.org/document/7780677

[11]. Simonyan K, Zisserman A (2014). Very Deep Convolutional Networks for Large-Scale Image Recognition. CoRR, abs/1409.1556. https://doi.org/10.48550/arXiv.1409.1556

[12]. M. Sandler, A. Howard, M. Zhu, A. Zhmoginov and L. -C. Chen (2018). MobileNetV2: Inverted Residuals and Linear Bottlenecks. 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 4510-4520. https://ieeexplore.ieee.org/abstract/document/8578572

Cite this article

Zhang,Y.;Xu,Y.;Kong,Z.;Hu,Z. (2024). Comparison of deep learning models based on Chest X-ray image classification. Applied and Computational Engineering,64,55-62.

Data availability

The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.

Disclaimer/Publisher's Note

The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

About volume

Volume title: Proceedings of the 6th International Conference on Computing and Data Science

Conference website: https://www.confcds.org/

ISBN：978-1-83558-425-5(Print) / 978-1-83558-426-2(Online)

Conference date: 12 September 2024

Editor：Alan Wang, Roman Bauer

Series: Applied and Computational Engineering

Volume number: Vol.64

ISSN：2755-2721(Print) / 2755-273X(Online)

© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license. Authors who publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See Open access policy for details).