Pneumonia ranks as a leading cause of mortality, particularly in children
aged five and under. Detecting this disease typically requires radiologists to
examine chest X-rays and report their findings to physicians, a task
susceptible to human error. The application of Deep Transfer Learning (DTL) for
the identification of pneumonia through chest X-rays is hindered by a shortage
of available images, which has led to less than optimal DTL performance and
issues with overfitting. Overfitting is characterized by a model’s learning
that is too closely fitted to the training data, reducing its effectiveness on
unseen data. The problem of overfitting is especially prevalent in medical
image processing due to the high costs and extensive time required for image
annotation, as well as the challenge of collecting substantial datasets that
also respect patient privacy concerning infectious diseases such as pneumonia.
To mitigate these challenges, this paper introduces the use of conditional
generative adversarial networks (CGAN) to enrich the pneumonia dataset with
2690 synthesized X-ray images of the minority class, aiming to even out the
dataset distribution for improved diagnostic performance. Subsequently, we
applied four modified lightweight deep transfer learning models such as
Xception, MobileNetV2, MobileNet, and EfficientNetB0. These models have been
fine-tuned and evaluated, demonstrating remarkable detection accuracies of
99.26%, 98.23%, 97.06%, and 94.55%, respectively, across fifty epochs. The
experimental results validate that the models we have proposed achieve high
detection accuracy rates, with the best model reaching up to 99.26%
effectiveness, outperforming other models in the diagnosis of pneumonia from
X-ray images.
References
[1]
WHO (2023) Pneumonia. https://www.who.int/health-topics/pneumonia/#tab=tab_1
[2]
Madhavi, A., Abhijna, M.S., Sumanjali, P., Supraja, S., Ghalwan, M. and Chary, M.R. (2023) Automated Diagnosis of Pneumonia Using CNN and Transfer Learning Approaches. E3S Web of Conferences, 430, 01031. https://doi.org/10.1051/e3sconf/202343001031
[3]
Porag, A.M.R., Hasan, M.M. and Ahad, M.T. (2022) A Comparison Study of Deep CNN Architecture in Detecting of Pneumonia. arXiv Preprint arXiv:2212.14744.
[4]
Mukhlif, A.A., Al-Khateeb, B. and Mohammed, M.A. (2022) An Extensive Review of State-of-the-Art Transfer Learning Techniques Used in Medical Imaging: Open Issues and Challenges. Journal of Intelligent Systems, 31, 1085-1111. https://doi.org/10.1515/jisys-2022-0198
[5]
Masud, M. (2022) A Light-Weight Convolutional Neural Network Architecture for Classification of COVID-19 Chest X-Ray Images. Multimedia Systems, 28, 1165-1174. https://doi.org/10.1007/s00530-021-00857-8
[6]
Wen, Y., Chen, L., Deng, Y. and Zhou, C. (2021) Rethinking Pre-Training on Medical Imaging. Journal of Visual Communication and Image Representation, 78, 103145. https://doi.org/10.1016/j.jvcir.2021.103145
[7]
Loey, M., Manogaran, G. and Khalifa, N.E.M. (2020) A Deep Transfer Learning Model with Classical Data Augmentation and CGAN to Detect COVID-19 from Chest Ct radiography Digital Images. Neural Computing and Applications, 1-13. https://doi.org/10.1007/s00521-020-05437-x
[8]
Mehta, T. and Mehendale, N. (2021) Classification of X-Ray Images into COVID-19, Pneumonia, and TB Using CGAN and Fine-Tuned Deep Transfer Learning Models. Research on Biomedical Engineering, 37, 803-813. https://doi.org/10.1007/s42600-021-00174-z
[9]
Iman, M., Arabnia, H.R. and Rasheed, K. (2023) A Review of Deep Transfer Learning and Recent Advancements. Technologies, 11, Article 40. https://doi.org/10.3390/technologies11020040
[10]
Nayak, M. (2018) An Introduction to Conditional GANS (CGANs). Medium.
[11]
Mirza, M. and Osindero, S. (2014) Conditional Generative Adversarial Nets. arXiv Preprint arXiv:1411.1784.
[12]
Elharrouss, O., Akbari, Y., Almaadeed, N. and Al-Maadeed, S. (2022) Backbones-Review: Feature Extraction Networks for Deep Learning and Deep Reinforcement Learning Approaches. arXiv Preprint arXiv:2206.08016.
[13]
Zhu, F. (2021, January 22) MobileNets, EfficientNet and EfficientDet. https://seventt.github.io/2021/01/22/model-efficiency-and-model-accuracy/
[14]
Kathamuthu, N.D., Subramaniam, S., Le, Q.H., Muthusamy, S., Panchal, H., Sundararajan, S.C.M., Alrubaie, A.J. and Zahra, M.M.A. (2023) A Deep Transfer Learning-Based Convolution Neural Network Model for COVID-19 Detection Using Computed Tomography Scan Images for Medical Applications. Advances in Engineering Software, 175, Article 103317. https://doi.org/10.1016/j.advengsoft.2022.103317