Submit your papersSubmit Now
For Enquiries: [email protected]
IIARD LogoIIARD

Hybrid EfficientNet-U-Net Architecture for Automated Pneumothorax Localization and Diagnosis from Chest Radiographs

Miracle Ugomma Anunobi, Vincent Sunday Umana, Sunday Chukwuebuka Uduogu, Merit Chinonso Opara, Richard Iherorochi Nneji, Confidence Chigozirim Olumba, Precious Mojolaoluwa Ojo, Grace Ugochi Nneji

Abstract

Pneumothorax is a critical thoracic condition characterized by air accumulation in the pleural cavity, which can be life-threatening if not promptly diagnosed and treated. However, conventional chest radiograph diagnosis heavily relies on radiologists' experience, leading to diagnostic variability and potential delays in treatment. This research proposes an innovative deep learning model integrating EfficientNet and U-Net architectures to accurately detect pneumothorax from chest X-ray images. The proposed hybrid EfficientNet-UNet model leverages EfficientNet's robust feature extraction capabilities combined with U-Net’s effectiveness in segmentation tasks, significantly enhancing detection precision. Comprehensive training and validation were conducted using publicly available datasets of chest radiographs. Experimental results demonstrated promising performance, with the model achieving a high ROC_AUC score of 0.86, improved mean Intersection over Union of 0.20, and mean Dice coefficient of 0.30. Additionally, sensitivity reached approximately 0.40, while specificity remained stable around 0.89, indicating the model’s robust discriminatory ability. These outcomes highlight the proposed model’s effectiveness in automating pneumothorax diagnosis, offering a powerful tool to assist clinical decision-making, reduce diagnostic errors, and improve patient outcomes.

Keywords

Deep LearningEfficientNetU-NetPneumothoraxChest X-ray

References

Chaturvedi, A., Lee, S., Klionsky, N., & Chaturvedi, A. (2016). Demystifying the persistent pneumothorax: Role of imaging. Insights into Imaging, 7(3). https://doi.org/10.1007/s13244-016-0486-5 Koch, B. W., Howell, D. M., & Kahwaji, C. I. (2024). EMS pneumothorax. In StatPearls. StatPearls Publishing. Zhong, X., Goldacre, R., Morris, E. J. A., & Hallifax, R. J. (2024). Trends in incidence of pneumothorax in England before, during and after the COVID-19 pandemic (2017–2023): A population-based observational study. The Lancet Regional Health – Europe, 44. https://doi.org/10.1016/j.lanepe.2024.100994 Medscape. (2022). Pneumothorax imaging: Practice essentials, radiography, computed tomography. Ahn, J. S., et al. (2022). Association of artificial intelligence–aided chest radiograph interpretation with reader performance and efficiency. JAMA Network Open, 5(8), e2229289. https://doi.org/10.1001/jamanetworkopen.2022.29289 Rudolph J. et al. (2021). ‘Interpretation of Thoracic Radiography Shows Large Discrepancies Depending on the Qualification of the Physician—Quantitative Evaluation of Interobserver Agreement in a Representative Emergency Department Scenario’, Diagnostics , 11 (10), p. 1868. https//doi.org/10.3390/diagnostics11101868. Gefter, W. B., Post, B. A., & Hatabu, H. (2023). Commonly missed findings on chest radiographs. Chest, 163(3), 650–661. https://doi.org/10.1016/j.chest.2022.10.039 Agrawal, A. G., & Choudhary, P. (2023). ALCNN: Attention-based lightweight convolutional neural network for pneumothorax detection in chest X-rays. Biomedical Signal Processing and Control, 79, 104126. https://doi.org/10.1016/j.bspc.2022.104126 Taylor, A. G., Mielke, C., & Mongan, J. (2018). Automated detection of moderate and large pneumothorax on frontal chest X-rays using deep convolutional neural networks. PLOS Medicine, 15(11), e1002697. https://doi.org/10.1371/journal.pmed.1002697 Spiegel, P. K. (1995). The first clinical X-ray made in America—100 years. American Journal of Roentgenology, 164(1), 241–243. https://doi.org/10.2214/ajr.164.1.7998549 Urooj, F., Akbar, S., Hassan, S. A., Firdous, S., & Bashir, M. J. (2022). Computer-aided diagnosis of pneumothorax through X-ray images using deep learning—A review. In Prognostic Models in Healthcare (pp. 403–432). Springer Nature. https://doi.org/10.1007/978-981-19- 2057-8_15 Alzubaidi, L., et al. (2021). Review of deep learning: Concepts, CNN architectures, challenges, applications, future directions. Journal of Big Data, 8(1), 53. https://doi.org/10.1186/s40537-021-00444-8 Khan, A., Sohail, A., Zahoora, U., & Qureshi, A. S. (2020). A survey of the recent architectures of deep convolutional neural networks. Artificial Intelligence Review, 53(8), 5455–5516. https://doi.org/10.1007/s10462-020-09825-6 Lin, M., Chen, Q., & Yan, S. (2014). Network in network. arXiv:1312.4400. https://doi.org/10.48550/arXiv.1312.4400 Liu, Y., Pu, H., & Sun, D.-W. (2021). Efficient extraction of deep image features using CNN. Trends in Food Science & Technology, 113, 193–204. https://doi.org/10.1016/j.tifs.2021.04.042 Zou, X., Wang, Z., Li, Q., & Sheng, W. (2019). Integration of residual network and CNN. Neurocomputing, 367, 39–45. https://doi.org/10.1016/j.neucom.2019.08.023 Choudhary, R., Kaur, N., & Goel, P. (2024). Diagnosis of pneumothorax using CNN. ICIPCN 2024, 412–417. https://doi.org/10.1109/ICIPCN63822.2024.00073 CSDN. (2025). In-depth analysis of PyTorch CNN examples. Tpoint Tech. (2025). CNN introduction. Mostafa, M., & Wu, F.-X. (2021). Diagnosis using convolutional autoencoder. Academic Press. Abedalla, A., et al. (2021). Chest X-ray pneumothorax segmentation using U-Net. PeerJ Computer Science, 7, e607. https://doi.org/10.7717/peerj-cs.607 Iqbal, T., et al. (2021). Automatic diagnosis of pneumothorax from chest radiographs. IEEE Access, 9, 145817–145839. https://doi.org/10.1109/ACCESS.2021.3122998 Malhotra, P., et al. (2022). Deep learning-based CAD for pneumothorax detection. Sensors, 22(6), 2278. https://doi.org/10.3390/s22062278 Wang, Y., Sun, L., & Jin, Q. (2021). Enhanced diagnosis of pneumothorax. IEEE/ACM TCBB, 18(3), 951–962. https://doi.org/10.1109/TCBB.2019.2911947 Monday, H.N., Nneji, G.U., Hossin, M.A., Mark, K.D., Umana, E.S., Mgbejime, G.T. and Li, J. (2025). Enhancing ECG Classification in Cardiac Diagnostics: A Novel Approach Using Adaptive Focal Cross-Entropy Loss Function. IEEE Journal of Biomedical and Health Informatics. Li, X., Wang, Y., Monday, H.N. and Nneji, G.U. (2025). A novel residual learning of multi-scale feature extraction model for the classification of rice grain varieties. Computers and Electronics in Agriculture, 237, p.110491. Nneji, G. U., et al. (2025). FFS-IML: Fusion-based statistical feature selection. International Journal of Machine Learning and Cybernetics, 1–34. Monday, H. N., et al. (2022a). Wavelet convolutional capsule network. Complex & Intelligent Systems, 8(1), 1–15. https://doi.org/10.1007/s40747-022-00733-6 Monday, H. N., et al. (2022b). WMR-DepthwiseNet. Diagnostics, 12(3), 765. https://doi.org/10.3390/diagnostics12030765 Monday, H. N., et al. (2022c). COVID-19 diagnosis using Siamese neural network. Diagnostics, 12(3), 741. https://doi.org/10.3390/diagnostics12030741 Nneji, G. U., et al. (2022a). Fine-tuned Siamese network. Diagnostics, 12(3). https://doi.org/10.3390/diagnostics12030717 Nneji, G. U., et al. (2022b). COVID-19 identification from low-quality CT. Healthcare, 10(2), 403. https://doi.org/10.3390/healthcare10020403 Nneji, G. U., et al. (2022c). Identification of diabetic retinopathy. Diagnostics, 12(2), 540. https://doi.org/10.3390/diagnostics12020540 Nneji, G. U., et al. (2022d). Multi-channel image processing for pneumonia. Diagnostics, 12(2), 325. https://doi.org/10.3390/diagnostics12020325