Enhancing Pneumonia Disease Classification using Genetic Algorithm-Tuned DCGANs and VGG-16 Integration

Kania Ardhani  Putri; Wikky Fawwaz Al Maki

doi:10.35882/jeeemi.v6i1.349

Kania Ardhani Putri School of Computing, Telkom University, Bandung, Indonesia https://orcid.org/0009-0006-8673-2430
Wikky Fawwaz Al Maki School of Computing, Telkom University, Bandung, Indonesia https://orcid.org/0000-0002-1696-431X

DOI: https://doi.org/10.35882/jeeemi.v6i1.349

Keywords: Pneumonia, Deep Learning, VGG16, Genetic Algorithm, Deep Convolutional Generative Adversarial Network

Abstract

Diagnostic complications arise from pneumonia, characterized by lung inflammation caused by alveolar fluid accumulation, particularly in regions with limited radiologists. To tackle this issue, a new method utilizes the VGG16 methodology for categorization, bolstered by genetic algorithms. In addition, Deep Convolutional Generative Adversarial Networks (DCGANs) improve the dataset by adding fake X-rays of pneumonia. Genetic algorithms are used to optimize hyperparameters in classification tasks. In contrast, DCGANs are employed to increase data augmentation techniques, boosting models' accuracy in identifying and categorizing pneumonia cases. The study partitioned a dataset into training, testing, and validation sets for pneumonia X-ray pictures. The training of GANs entails utilizing both generators and discriminators to produce increasingly realistic pictures gradually. The genetic algorithm enhances the hyperparameter tuning process, resulting in a substantial increase in accuracy. Initially, VGG16 achieved a success rate of 89.50% and a fitness score of 87.50%. Post-optimization and DCGAN augmentation, accuracy climbed to 95.50%, and F1-Score improved to 94.75%. This study combines genetic algorithms and DCGANs to create a model that can produce genuine pneumonia X-ray pictures and enhance categorization accuracy.

Downloads

Download data is not yet available.

References

İstanbul AREL Üniversitesi, IEEE Engineering in Medicine and Biology Society, Institute of Electrical and Electronics Engineers. Turkey Section, Institute of Electrical and Electronics Engineers, Ayan, Enes, and Halil Murat Ünver. “Diagnosis of pneumonia from chest X-ray images using deep learning.” 2019 Scientific Meeting on Electrical-Electronics & Biomedical Engineering and Computer Science (EBBT). Ieee, 2019.

Amity University and Institute of Electrical and Electronics Engineers, Sharma, Harsh, et al. “Feature extraction and classification of chest x-ray images using CNN to detect pneumonia.” 2020 10th international conference on cloud computing, data science & engineering (Confluence). IEEE, 2020.

D. R. Chandranegara, Z. Sari, M. B. Dewantoro, H. Wibowo, and W. Suharso, “Implementation of Generative Adversarial Network (GAN) Method for Pneumonia Dataset Augmentation,” Kinetik: Game Technology, Information System, Computer Network, Computing, Electronics, and Control, May 2023, doi: 10.22219/kinetik.v8i2.1675.

R. Jain, P. Nagrath, G. Kataria, V. Sirish Kaushik, and D. Jude Hemanth, “Pneumonia detection in chest X-ray images using convolutional neural networks and transfer learning,” Measurement (Lond), vol. 165, Dec. 2020, doi: 10.1016/j.measurement.2020.108046.

J. P. Metlay et al., “Diagnosis and treatment of adults with community-acquired pneumonia,” Am J Respir Crit Care Med, vol. 200, no. 7, pp. E45–E67, Oct. 2019, doi: 10.1164/rccm.201908-1581ST.

J. A. Ramirez et al., “Treatment of Community-Acquired Pneumonia in Immunocompromised Adults: A Consensus Statement Regarding Initial Strategies,” Chest, vol. 158, no. 5. Elsevier Inc., pp. 1896–1911, Nov. 01, 2020. doi: 10.1016/j.chest.2020.05.598.

D. Srivastav, A. Bajpai, and P. Srivastava, “Improved classification for pneumonia detection using transfer learning with GAN-based synthetic image augmentation,” in Proceedings of the Confluence 2021: 11th International Conference on Cloud Computing, Data Science and Engineering, Institute of Electrical and Electronics Engineers Inc., Jan. 2021, pp. 433–437. doi: 10.1109/Confluence51648.2021.9377062.

S. R. Islam, S. P. Maity, A. K. Ray, and M. Mandal, “Deep learning on compressed sensing measurements in pneumonia detection,” Int J Imaging Syst Technol, vol. 32, no. 1, pp. 41–54, Jan. 2022, doi: 10.1002/ima.22651.

G. J. Chowdary, G. Suganya, M. Premalatha, and S. Ganapathy, “Impact Of Machine Learning Models In Pneumonia Diagnosis With Features Extracted From Chest X-Rays Using VGG16,” 2021.

Z. P. Jiang, Y. Y. Liu, Z. E. Shao, and K. W. Huang, “An improved VGG16 model for pneumonia image classification,” Applied Sciences (Switzerland), vol. 11, no. 23, Dec. 2021, doi: 10.3390/app112311185.

M. Nishio, S. Noguchi, H. Matsuo, and T. Murakami, “Automatic classification between COVID-19 pneumonia, non-COVID-19 pneumonia, and the healthy on chest X-ray image: combination of data augmentation methods,” Sci Rep, vol. 10, no. 1, Dec. 2020, doi: 10.1038/s41598-020-74539-2.

C. He, S. Huang, R. Cheng, K. C. Tan, and Y. Jin, “Evolutionary Multiobjective Optimization Driven by Generative Adversarial Networks (GANs),” Oct. 2019, [Online]. Available: http://arxiv.org/abs/1910.04966

B. Yelmen et al., “Creating artificial human genomes using generative neural networks,” PLoS Genet, vol. 17, no. 2, Feb. 2021, doi: 10.1371/JOURNAL.PGEN.1009303.

M. A. A. Albadr, S. Tiun, M. Ayob, F. T. Al-Dhief, K. Omar, and F. A. Hamzah, “Optimised genetic algorithm-extreme learning machine approach for automatic COVID-19 detection,” PLoS One, vol. 15, no. 12 December, Dec. 2020, doi: 10.1371/journal.pone.0242899.

H. E. Bate, “Genetic Algorithm-based Optimization of Generative Adversarial Networks and its Applications.”

B. Gülmez, “A novel deep neural network model based Xception and genetic algorithm for detection of COVID-19 from X-ray images,” Ann Oper Res, vol. 328, no. 1, pp. 617–641, Sep. 2023, doi: 10.1007/s10479-022-05151-y.

J. Gu, Y. Shen, and B. Zhou, “Image Processing Using Multi-Code GAN Prior,” 2019.

C. Dewi, R. C. Chen, Y. T. Liu, and S. K. Tai, “Synthetic Data generation using DCGAN for improved traffic sign recognition,” Neural Comput Appl, vol. 34, no. 24, pp. 21465–21480, Dec. 2022, doi: 10.1007/s00521-021-05982-z.

M. K. D. Z. K. Kermany D.S.; Goldbaum M.; Zhang K.; Goldbaum, “Large Dataset of Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images,” 2018.

L. J. Isaksson et al., “Effects of MRI image normalization techniques in prostate cancer radiomics,” Physica Medica, vol. 71, pp. 7–13, Mar. 2020, doi: 10.1016/j.ejmp.2020.02.007.

J. Xu, X. Sun, Z. Zhang, G. Zhao, and J. Lin, “Understanding and Improving Layer Normalization.” [Online]. Available: https://github.com/pytorch/fairseq

A. Kulkarni, D. Chong, and F. A. Batarseh, “Foundations of data imbalance and solutions for a data democracy,” Data Democracy: At the Nexus of Artificial Intelligence, Software Development, and Knowledge Engineering, pp. 83–106, Jan. 2020, doi: 10.1016/B978-0-12-818366-3.00005-8.

M. Liu and J. Yang, “Image classification of brain tumor based on channel attention mechanism,” in Journal of Physics: Conference Series, IOP Publishing Ltd, Oct. 2021. doi: 10.1088/1742-6596/2035/1/012029.

K. Eckle and J. Schmidt-Hieber, “A comparison of deep networks with ReLU activation function and linear spline-type methods,” Neural Networks, vol. 110, pp. 232–242, Feb. 2019, doi: 10.1016/J.NEUNET.2018.11.005.

Q. Wu, Y. Chen, and J. Meng, “Dcgan-based data augmentation for tomato leaf disease identification,” IEEE Access, vol. 8, pp. 98716–98728, 2020, doi: 10.1109/ACCESS.2020.2997001.

I. D. Raji, H. Bello-Salau, I. J. Umoh, A. J. Onumanyi, M. A. Adegboye, and A. T. Salawudeen, “Simple Deterministic Selection-Based Genetic Algorithm for Hyperparameter Tuning of Machine Learning Models,” Applied Sciences (Switzerland), vol. 12, no. 3, Feb. 2022, doi: 10.3390/app12031186.

G. T. Reddy, M. P. K. Reddy, K. Lakshmanna, D. S. Rajput, R. Kaluri, and G. Srivastava, “Hybrid genetic algorithm and a fuzzy logic classifier for heart disease diagnosis,” Evol Intell, vol. 13, no. 2, pp. 185–196, Jun. 2020, doi: 10.1007/s12065-019-00327-1.

Institute of Electrical and Electronics Engineers and Manav Rachna International Institute of Research and Studies, Proceedings of the International Conference on Machine Learning, Big Data, Cloud and Parallel Computing : Trends, perspectives, and prospects : COMITCON-2019 : 14th-16th February 2019.

P. Naveen and B. Diwan, “Pre-trained VGG-16 with CNN architecture to classify X-Rays images into normal or pneumonia,” in 2021 International Conference on Emerging Smart Computing and Informatics, ESCI 2021, Institute of Electrical and Electronics Engineers Inc., Mar. 2021, pp. 102–105. doi: 10.1109/ESCI50559.2021.9396997.

D. Albashish, R. Al-Sayyed, A. Abdullah, M. H. Ryalat, and N. Ahmad Almansour, “Deep CNN Model based on VGG16 for Breast Cancer Classification,” in 2021 International Conference on Information Technology, ICIT 2021 - Proceedings, Institute of Electrical and Electronics Engineers Inc., Jul. 2021, pp. 805–810. doi: 10.1109/ICIT52682.2021.9491631.

A. Nasiri, A. Taheri-Garavand, and Y. D. Zhang, “Image-based deep learning automated sorting of date fruit,” Postharvest Biol Technol, vol. 153, pp. 133–141, Jul. 2019, doi: 10.1016/j.postharvbio.2019.04.003.

P. Ganakwar, “Convolutional Neural Network-VGG16 for Road Extraction from Remotely Sensed Images,” Int J Res Appl Sci Eng Technol, vol. 8, no. 8, pp. 916–922, Aug. 2020, doi: 10.22214/ijraset.2020.30796.

K. Behara, E. Bhero, and J. T. Agee, “Skin Lesion Synthesis and Classification Using an Improved DCGAN Classifier,” Diagnostics, vol. 13, no. 16, Aug. 2023, doi: 10.3390/diagnostics13162635.

M. Padala, D. Das, and S. Gujar, “Effect of Input Noise Dimension in GANs,” Apr. 2020, [Online]. Available: http://arxiv.org/abs/2004.06882

G. IEEE Engineering in Medicine and Biology Society. Annual International Conference (41st : 2019 : Berlin, IEEE Engineering in Medicine and Biology Society, and Institute of Electrical and Electronics Engineers, 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC) : Biomedical Engineering Ranging from Wellness to Intensive Care : 41st EMB Conference 2019 : July 23-27, Berlin.