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, its interpretation still depends heavily on radiologist expertise and may lead to 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 novelty of this study lies in the integration of Grey Wolf Optimizer (GWO) for hyperparameter optimization of the YOLOv5n model specifically for pulmonary edema detection in Chest X-ray images. The proposed method employs YOLOv5n as a lightweight object detection architecture because of its computational efficiency and suitability for resource-constrained environments. To improve detection performance and training stability, the hyperparameters of YOLOv5n are optimized using GWO. The model is trained and evaluated using annotated CXR images, and its performance is measured using precision, recall, mAP@0.5, and mAP@0.5–0.95. Experimental results show that the YOLOv5n + GWO model achieved a precision of 0.906, recall of 0.936, mAP@0.5 of 0.963, and mAP@0.5–0.95 of 0.737, indicating improved detection performance compared with the baseline YOLOv5n configuration. The proposed framework demonstrates potential as a decision-support tool for assisting medical personnel in early pulmonary edema screening through efficient and consistent analysis of CXR images.

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,” in Proc. 44th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Glasgow, Scotland, United Kingdom, 2022, pp. 1834–1838, doi: 10.1109/EMBC48229.2022.9871153.

J. Huynh, S. Masoudi, A. Noorbakhsh, A. Mahmoodi, S. Kligerman, A. Yen, K. Jacobs, L. Hahn, K. Hasenstab, M. Pazzani, and A. Hsiao, “Deep Learning Radiographic Assessment of Pulmonary Edema: Optimizing Clinical Performance, Training With Serum Biomarkers,” IEEE Access, vol. 10, pp. 48577–48588, 2022, doi: 10.1109/ACCESS.2022.3172706.

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,” in Proc. International Conference on Inventive Research in Computing Applications (ICIRCA), 2025, pp. 1221–1226, doi: 10.1109/ICIRCA65293.2025.11089624.

A. M. Harale, V. K. Bairagi, E. Boonchieng, and M. R. Bachute, “Nodules Detection in Lungs CT Images Using Improved YOLOV5 and Classification of Types of Nodules by CNN-SVM,” IEEE Access, vol. 12, pp. 140456–140471, 2024, doi: 10.1109/ACCESS.2024.3466292.

M. Suchithra, S. Guleria, and S. Patil, “NeXt-Ray: Next Gen X-Ray Diagnosis Using Deep Learning and GenAI,” in Proc. International Conference on Innovative Trends in Information Technology (ICITIIT), 2025, pp. 1–7, 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,” in Proc. International Conference on Computational, Communication and Information Technology (ICCCIT), 2025, pp. 57–61, doi: 10.1109/ICCCIT62592.2025.10927913.

S. Hao, X. Li, W. Peng, Z. Fan, Z. Ji, and I. Ganchev, “YOLO-CXR: A Novel Detection Network for Locating Multiple Small Lesions in Chest X-Ray Images,” IEEE Access, vol. 12, pp. 156003–156019, 2024, doi: 10.1109/ACCESS.2024.3482102.

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

Q. Guan, Y. Huang, Y. Luo, P. Liu, M. Xu, and Y. Yang, “Discriminative Feature Learning for Thorax Disease Classification in Chest X-ray Images,” IEEE Transactions on Image Processing, vol. 30, pp. 2476–2487, 2021, doi: 10.1109/TIP.2021.3052711.

M. S. Choudhury, R. Rahman, S. T. Bhuiyan, S. Wasi, and S. B. Alam, “Deep Learning for Accurate Classification of Pulmonary Edema Severity in Chest X-Rays: A Comparative Evaluation of CNN Models with Grad-CAM Feature Visualization,” in Proc. 2025 International Conference on Wavelet Analysis and Pattern Recognition (ICWAPR), Bali, Indonesia, 2025, doi: 10.1109/ICWAPR66259.2025.11306863.

B. M. Wiley, B. Zhou, G. Pandompatam, J. Zhou, H. O. Kucuk, and X. Zhang, “Lung Ultrasound Surface Wave Elastography for Assessing Patients With Pulmonary Edema,” IEEE Transactions on Biomedical Engineering, vol. 68, no. 11, pp. 3417–3423, 2021, doi: 10.1109/TBME.2021.3072891.

E. Khalili, B. Priego-Torres, A. León-Jiménez, and D. Sanchez-Morillo, “Automatic Lung Segmentation in Chest X-Ray Images Using SAM With Prompts From YOLO,” IEEE Access, vol. 12, pp. 122805–122819, 2024, doi: 10.1109/ACCESS.2024.3454188.

Y. C. Kuyu and N. Ozekmekci, “Grey Wolf Optimizer to the Hyperparameters Optimization of Convolutional Neural Network with Several Activation Functions,” in Proc. 2022 International Symposium on Multidisciplinary Studies and Innovative Technologies (ISMSIT), Ankara, Turkey, 2022, doi: 10.1109/ISMSIT56059.2022.9932838.

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. 2, pp. 1559–1576, 2021, doi: 10.1007/s12652-020-02228-z.

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

C. Paramita, C. Supriyanto, P. Šolić, C. Wada, and A. A. Dzaky, “Performance Evaluation of YOLOv8 Models for Multi-Class Skin Lesion Detection From Dermoscopic Images,” in Proc. 2025 International Conference on Smart Computing, IoT and Machine Learning (SIML), 2025, doi: 10.1109/SIML65326.2025.11080819.

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, doi: 10.18280/isi.290408.

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: 10.15294/sji.v11i2.2808.

W. Fan, X. Guo, L. Teng, and Y. Wu, “Research on Abnormal Target Detection Method in Chest Radiograph Based on YOLO v5 Algorithm,” in Proc. 2021 IEEE International Conference on Computer Science, Electronic Information Engineering and Intelligent Control Technology (CEI), Fuzhou, China, 2021, doi: 10.1109/CEI52496.2021.9574450.

M. G. Ragab, S. J. Abdulkadir, A. Muneer, A. Alqushaibi, E. H. Sumiea, R. Qureshi, S. M. Al-Selwi, and H. Alhussian, “A Comprehensive Systematic Review of YOLO for Medical Object Detection (2018 to 2023),” IEEE Access, vol. 12, pp. 57815–57836, 2024, doi: 10.1109/ACCESS.2024.3386826.

Y. Yuan, S. Yao, X. Ren, Y. Xiao, Z. Zhou, and Y. Lu, “Chest X-ray Disease Diagnosis Method Based on YOLO Model and Pseudo Color Conversion,” in Proc. 2022 5th International Conference on Data Science and Information Technology (DSIT), Shanghai, China, 2022, doi: 10.1109/DSIT55514.2022.9943871.

J. Leka Shree, E. Joy, and M. S. Sri Ram, “Enhanced Radiological Anomaly Detection Using Optimized YOLO-NAS Model,” in Proc. 2024 International Conference on Advances in Modern Age Technologies for Health and Engineering Science (AMATHE), Shivamogga, India, 2024, doi: 10.1109/AMATHE61652.2024.10582157.

W. Willy and A. Prabowo, “Penerapan Transfer Learning VGG-16 untuk Mendeteksi Penyakit Mata Manusia Berbasis Citra Fundus,” Jurnal Informatika: Jurnal Pengembangan IT, vol. 11, no. 1, pp. 84–93, 2026, doi: 10.30591/jpit.v11i1.9291.

D. Schulz, S. Rasch, M. Heilmaier, R. Abbassi, A. Poszler, J. Ulrich, M. Steinhardt, G. A. Kaissis, R. M. Schmid, R. Braren, and T. Lahmer, “A Deep Learning Model Enables Accurate Prediction and Quantification of Pulmonary Edema From Chest X-Rays,” Critical Care, vol. 27, Art. no. 201, 2023, doi: 10.1186/s13054-023-04426-5.

Y.-C. Cheng, Y.-C. Hung, G.-H. Huang, T.-B. Chen, N.-H. Lu, K.-Y. Liu, and K.-H. Lin, “Deep Learning-Based Object Detection Strategies for Disease Detection and Localization in Chest X-Ray Images,” Diagnostics, vol. 14, no. 23, Art. no. 2636, 2024, doi: 10.3390/diagnostics14232636.

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