Polyp segmentation with interference filtering and dynamic uncertainty mining
Yunhua Zhang,Gang Yang,Congjin Gong,Jianhao Zhang,Shuo Wang,Yutao Wang
DOI: https://doi.org/10.1088/1361-6560/ad2b94
IF: 3.5
2024-02-23
Physics in Medicine and Biology
Abstract:Accurate polyp segmentation from colonoscopy images plays a crucial role in the early diagnosis and treatment of colorectal cancer. However, existing polyp segmentation methods are inevitably affected by various image noises, such as reflections, motion blur, and feces, which significantly affect the performance and generalization of the model. In addition, coupled with ambiguous boundaries between polyps and surrounding tissue, i.e., small inter-class differences, accurate polyp segmentation remains a challenging problem. To address these issues, we propose a novel two-stage polyp segmentation method that leverages a preprocessing sub-network (Pre-Net) and a Dynamic Uncertainty Mining Network (DUMNet) to improve the accuracy of polyp segmentation. Pre-Net identifies and filters out interference regions before feeding the colonoscopy images to the polyp segmentation network DUMNet. Considering the confusing polyp boundaries, DUMNet employs the Uncertainty Mining Module (UMM) to dynamically focus on foreground, background, and uncertain regions based on different pixel confidences. UMM helps to mine and enhance more detailed context, leading to coarse-to-fine polyp segmentation and precise localization of polyp regions. We conduct experiments on five popular polyp segmentation benchmarks: ETIS, CVC-ClinicDB, CVC-ColonDB, EndoScene, and Kvasir. Our method achieves state-of-the-art performance. Furthermore, the proposed Pre-Net has strong portability and can improve the accuracy of existing polyp segmentation models. Our code will be released.
engineering, biomedical,radiology, nuclear medicine & medical imaging