Deteksi Edema Paru Pada Citra Chest X-ray Menggunakan YOLOv5n Dengan Optimasi Hyperparameter Berbasis Grey Wolf Optimizer

Nasrudin Affandi Prasetyo; Cinantya Paramita; Amiq Fahmi

doi:10.30591/jpit.v11i2.10375

Deteksi Edema Paru Pada Citra Chest X-ray Menggunakan YOLOv5n Dengan Optimasi Hyperparameter Berbasis Grey Wolf Optimizer

Nasrudin Affandi Prasetyo, Cinantya Paramita, Amiq Fahmi

Abstract

Pulmonary edema is a lung disorder characterized by fluid accumulation in the alveolar and interstitial spaces, which disrupts gas exchange and reduces oxygen levels in the blood. Chest X-ray (CXR) imaging is commonly used for pulmonary edema assessment because it is fast and widely available; however, interpretation of CXR images heavily depends on radiologist expertise and may result in diagnostic variability, particularly in healthcare facilities with limited radiology resources. This study aims to develop an automated pulmonary edema detection system based on deep learning to support more consistent analysis of CXR images. The proposed method utilizes the YOLOv5n model as a lightweight object detection architecture due to its computational efficiency and suitability for resource-constrained environments. To improve detection performance and training stability, hyperparameters of the YOLOv5n model are optimized using the Grey Wolf Optimizer (GWO). The model is trained and evaluated using annotated CXR images, and its performance is assessed using precision, recall, mAP@0.5, and mAP@0.5–0.95 metrics. Experimental results show that the integration of GWO improves detection accuracy and model stability compared to the baseline configuration. The proposed framework demonstrates the potential to support pulmonary edema screening by providing efficient and consistent analysis of CXR images and can function as a decision-support tool for assisting medical personnel in early disease detection.

Keywords

Chest X-ray, Deep Learning, Grey Wolf Optimizer, Pulmonary Edema, YOLOv5n.

Full Text:

References

M. N. Akbar, X. Wang, D. Erdogmus, and S. Dalal, “PENet: Continuous-Valued Pulmonary Edema Severity Prediction on Chest X-ray Using Siamese Convolutional Networks,” Proceedings of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 1834–1838, 2022, doi: 10.1109/EMBC48229.2022.9871153.

Y. Tao, X. Ma, Y. Zhang, K. Huang, Z. Ji, W. Fan, S. Yuan, and Q. Chen, “LAGAN: Lesion-Aware Generative Adversarial Networks for Edema Area Segmentation in SD-OCT Images,” IEEE Journal of Biomedical and Health Informatics, vol. 27, no. 5, pp. 2432–2443, 2023, doi: 10.1109/JBHI.2023.3252665.

M. Fetanat, M. Stevens, P. Jain, C. Hayward, E. Meijering, and N. H. Lovell, “Fully Elman Neural Network: A Novel Deep Recurrent Neural Network Optimized by an Improved Harris Hawks Algorithm for Classification of Pulmonary Arterial Wedge Pressure,” IEEE Transactions on Biomedical Engineering, vol. 69, no. 5, pp. 1733–1744, 2022, doi: 10.1109/TBME.2021.3129459.

R. Rampriya, R. Santoshkumar Reddy, K. Lakshmi Prasad, H. B. Karthik Kumar, and R. Vimal Raja, “Diagnosis and Risk Analysis of Pulmonary Disorder in CXR Images Using Optimized Deep Learning Techniques,” Proceedings of the International Conference on Inventive Research in Computing Applications (ICIRCA), pp. 1221–1226, 2025, doi: 10.1109/ICIRCA65293.2025.11089624.

M. Marathe and M. Bhalekar, “Detection of Lung Cancer Using CT Scans with Deep Learning Approach,” Proceedings of the International Conference on Innovations in Recent Technologies of Computing and Applications (NMITCON), pp. 1–6, 2023, doi: 10.1109/NMITCON58196.2023.10276175.

M. Suchithra, S. Guleria, and S. Patil, “NeXt-Ray: Next Gen X-Ray Diagnosis Using Deep Learning and GenAI,” Proceedings of the International Conference on Innovative Trends in Information Technology (ICITIIT), pp. 1–7, 2025, doi: 10.1109/ICITIIT64777.2025.11041079.

R. T. Lucassen et al., “Deep Learning for Detection and Localization of B-Lines in Lung Ultrasound,” IEEE Journal of Biomedical and Health Informatics, vol. 27, no. 9, pp. 4352–4361, 2023, doi: 10.1109/JBHI.2023.3282596.

V. Tidke et al., “Multi Disease Prediction Related to Pulmonary Area by Leveraging Deep Learning,” Proceedings of the International Conference on Computational, Communication and Information Technology (ICCCIT), pp. 57–61, 2025, doi: 10.1109/ICCCIT62592.2025.10927913.

M. Perpenti et al., “A Novel Empirical Wavelet Transform Approach for Classification of Radiofrequency Lung Ultrasound Signals Applied to Diagnosis of Lung Diseases,” Proceedings of the IEEE Ultrasonics, Ferroelectrics and Frequency Control Joint Symposium (UFFC-JS), pp. 1–4, 2024, doi: 10.1109/UFFC-JS60046.2024.10793884.

W. Lu, Y. Xue, X. Chen, and H. He, “Enhancing Chest X-Ray Classification with Retrieval-Augmented OpenAI GPT Models,” Proceedings of the International Conference on Natural Language Processing (ICNLP), pp. 394–398, 2025, doi: 10.1109/ICNLP65360.2025.11108514.

Z. J. Hussein and E. H. Al-Saadi, “Recent Developments Used for Early Detection of Diabetic Macular Edema by Retinal Images: A Survey,” Proceedings of the Iraqi International Conference on Communication and Information Technologies (IICCIT), pp. 335–341, 2022.

Y. Yang, Y. Bai, and X. Wang, “Three-Dimensional EIT of Pulmonary Edema Based on TV Algorithm,” Proceedings of the International Conference on Frontiers of Electronics, Information and Communication Technologies (ICFEICT), pp. 102–107, 2023.

B. M. Wiley et al., “Lung Ultrasound Surface Wave Elastography for Assessing Patients with Pulmonary Edema,” IEEE Transactions on Biomedical Engineering, vol. 68, no. 11, pp. 3417–3423, 2021.

S. Sethi, Y. K. Meena, and S. P. Mohanty, “Lung Disease Detection Using Improved YOLOv7 Framework with Contextual Transformer and Enhanced Loss Function,” Applied Sciences, vol. 14, no. 14, p. 6111, 2024.

T. F. Abbas and A. H. Shabeeb, “Path Planning and Obstacle Avoidance of a Mobile Robot Based on GWO Algorithm,” Al-Khwarizmi Engineering Journal, vol. 18, no. 2, pp. 13–28, 2022.

M. Safaldin, M. Otair, and L. Abualigah, “Improved Binary Gray Wolf Optimizer and SVM for Intrusion Detection System in Wireless Sensor Networks,” Journal of Ambient Intelligence and Humanized Computing, vol. 12, no. 9, pp. 9349–9366, 2021.

B. Zhao, L. Chang, and Z. Liu, “Fast-YOLO Network Model for X-Ray Image Detection of Pneumonia,” Electronics, vol. 14, no. 5, p. 903, 2025.

C. Paramita et al., “Performance Evaluation of YOLOv8 Models for Multi-Class Skin Lesion Detection from Dermoscopic Images,” Proceedings of the International Conference on Smart Computing, IoT and Machine Learning (SIML), 2025.

E. R. Subhiyakto et al., “Evaluation of Resampling Techniques in CNN-Based Heartbeat Classification,” Ingénierie des Systèmes d’Information, vol. 29, no. 4, pp. 1323–1332, 2024.

C. Paramita et al., “Comparative Analysis of YOLOv5 and YOLOv8 Cigarette Detection in Social Media Content,” Scientific Journal of Informatics, vol. 11, no. 2, pp. 341–352, 2024.

DOI: https://doi.org/10.30591/jpit.v11i2.10375