CPSNet: Comprehensive Enhancement Representation for Polyp Segmentation Task

Abstract

Accurately segmenting polyp regions in colonoscopy images is crucial for the diagnosis and intervention of colorectal cancer. However, the task of polyp segmentation remains challenging due to the diverse size and shape variations among polyps, their extreme similarity to the background, and frequent rotation of the lens, which further increases the diversity in polyp presentation. To address these challenges effectively, we propose a comprehensive polyp segmentation network (CPSNet). Specifically, we introduce a Comprehensive Spatial Feature Extraction Module (CFEM) that progressively and densely integrates features while forming receptive windows with various shapes. This enables enhanced perception of polyps at manifold sizes and shapes. Additionally, we propose a Fine-grained Region Strengthen Module (FGSM) to supplement uncertain areas around polyps by mitigating background noise interference. In terms of training strategy, we further introduce a Rotation-augmented Constrained Loss (RC Loss), which reinforces consistency constraints on polyp images under multiple rotation angles. Qualitative and quantitative experiments conducted on five public datasets demonstrate both the plug-and-play capability of CFEM as well as the effectiveness and excellence achieved by CPSNet.