Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation

Abstract

Colonoscopy is a procedure to detect colorectal polyps which are the primary cause for developing colorectal cancer. However, polyp segmentation is a challenging task due to the diverse shape, size, color, and texture of polyps, shuttle difference between polyp and its background, as well as low contrast of the colonoscopic images. To address these challenges, we propose a feature enhancement network for accurate polyp segmentation in colonoscopy images. Specifically, the proposed network enhances the semantic information using the novel Semantic Feature Enhance Module (SFEM). Furthermore, instead of directly adding encoder features to the respective decoder layer, we introduce an Adaptive Global Context Module (AGCM), which focuses only on the encoder's significant and hard fine-grained features. The integration of these two modules improves the quality of features layer by layer, which in turn enhances the final feature representation. The proposed approach is evaluated on five colonoscopy datasets and demonstrates superior performance compared to other state-of-the-art models.

References in corpus (6)

• Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentations
• Automated polyp detection in colon capsule endoscopy
• UNet++: A Nested U-Net Architecture for Medical Image Segmentation
• Uncertainty and Interpretability in Convolutional Neural Networks for Semantic Segmentation of Colorectal Polyps
• A Comparative Study on Polyp Classification using Convolutional Neural Networks
• Deep Feature Augmentation for Occluded Image Classification

Cited by in corpus (2)

• A Survey on Deep Learning for Polyp Segmentation: Techniques, Challenges and Future Trends
• Colonoscopy Polyp Detection and Classification: Dataset Creation and Comparative Evaluations