Staffordshire University logo
STORE - Staffordshire Online Repository

Chest X-ray pneumothorax segmentation using U-Net with EfficientNet and ResNet architectures

Abedalla, Ayat, Abdullah, Malak, Al-Ayyoub, Mahmoud and BENKHELIFA, Elhadj (2021) Chest X-ray pneumothorax segmentation using U-Net with EfficientNet and ResNet architectures. PeerJ Computer Science, 7. e607. ISSN 2376-5992

[img]
Preview
Text
Chest_X-ray_pneumothorax_segmentation_using_U-Net_.pdf - Publisher's typeset copy
Available under License Creative Commons Attribution 4.0 International (CC BY 4.0) .

Download (13MB) | Preview

Abstract or description

Medical imaging refers to visualizing techniques for providing valuable information about the human body’s internal structures for clinical applications, diagnosis, treatment, and scientific research. One of the essential methods for analyzing and processing medical images is segmentation, which helps doctors diagnose accurately by providing detailed information on the body’s required part. However, segmenting medical images faces several challenges, such as requiring trained medical experts and being timeconsuming and error-prone. Thus, it appears necessary for an automatic medical image segmentation system. Deep learning algorithms have recently shown outstanding performance for segmentation tasks, especially semantic segmentation networks that provide pixel-level image understanding. By introducing the first Fully Convolutional Network (FCN) for semantic image segmentation, several segmentation networks have been proposed on its basis. One of the state-of-the-art convolutional networks in the medical image field is U-Net. This paper presents a novel end-to-end semantic segmentation model, named Ens4B-UNet, for medical images as Ensembles 4 U-Net architectures with pre-trained Backbone networks. Ens4B-UNet utilizes U-Net’s success with several significant improvements by adapting powerful and robust Convolutional Neural Networks (CNNs) as backbones for U-Nets encoders and using the nearest-neighbor up-sampling in the decoders. Ens4B-UNet is designed based on the weighted average ensemble of four encoder-decoder segmentation models. The backbone networks of all ensembled models are pre-trained on the ImageNet dataset to exploit the benefit of transfer learning. For improving our models, we apply several techniques for training and predicting, including Stochastic Weight Averaging (SWA), data augmentation, Test-Time Augmentation (TTA), and different types of optimal thresholds. We evaluate and test our models on the 2019 Pneumothorax Challenge dataset, which contains 12,047 training images with 12,954 masks and 3,205 test images. Our proposed segmentation network achieves a 0.8608 mean Dice Similarity Coefficient (DSC) on the test set, which is among the top 1-percent systems in the Kaggle competition.

Item Type: Article
Faculty: School of Digital, Technologies and Arts > Computer Science, AI and Robotics
Depositing User: Elhadj BENKHELIFA
Date Deposited: 07 Nov 2022 15:31
Last Modified: 24 Feb 2023 14:04
URI: https://eprints.staffs.ac.uk/id/eprint/7504

Actions (login required)

View Item View Item

DisabledGo Staffordshire University is a recognised   Investor in People. Sustain Staffs
Legal | Freedom of Information | Site Map | Job Vacancies
Staffordshire University, College Road, Stoke-on-Trent, Staffordshire ST4 2DE t: +44 (0)1782 294000