UACANet: Uncertainty Augmented Context Attention for Polyp Segmentation

Abstract

We propose Uncertainty Augmented Context Attention network (UACANet) for polyp segmentation which consider a uncertain area of the saliency map. We construct a modified version of U-Net shape network with additional encoder and decoder and compute a saliency map in each bottom-up stream prediction module and propagate to the next prediction module. In each prediction module, previously predicted saliency map is utilized to compute foreground, background and uncertain area map and we aggregate the feature map with three area maps for each representation. Then we compute the relation between each representation and each pixel in the feature map. We conduct experiments on five popular polyp segmentation benchmarks, Kvasir, CVC-ClinicDB, ETIS, CVC-ColonDB and CVC-300, and achieve state-of-the-art performance. Especially, we achieve 76.6% mean Dice on ETIS dataset which is 13.8% improvement compared to the previous state-of-the-art method. Source code is publicly available at https://github.com/plemeri/UACANet

arXiv comment9 pages, 6 figures, 4 tables. To appear in the Proceedings of the 29th ACM International Conference on Multimedia (ACM MM '21), October 20-24, 2021, Chengdu, China. DOI will be added soon

Cited by in corpus (10)

• Polyp-PVT: Polyp Segmentation with Pyramid Vision Transformers
• FCN-Transformer Feature Fusion for Polyp Segmentation
• Video Polyp Segmentation: A Deep Learning Perspective
• A Survey on Deep Learning for Polyp Segmentation: Techniques, Challenges and Future Trends
• Uncertainty Guided Refinement for Fine-Grained Salient Object Detection
• Rethinking Polyp Segmentation from an Out-of-Distribution Perspective
• Frontiers in Intelligent Colonoscopy
• BiSeg-SAM: Weakly-Supervised Post-Processing Framework for Boosting Binary Segmentation in Segment Anything Models
• LGPS: A Lightweight GAN-Based Approach for Polyp Segmentation in Colonoscopy Images
• PVTAdpNet: Polyp Segmentation using Pyramid vision transformer with a novel Adapter block