CSG-Mamba: A Convolutional Scoring Gating Vision State Space Network for Endoscopic Polyp Segmentation
Organizations: School of Advanced Interdisciplinary Studies, Tiangong University, Tianjin, China
Abstract
Accurate polyp segmentation is critical for computer-aided colonoscopy, yet endoscopic images often contain low-contrast boundaries, mucosal texture interference, specular highlights, and device-dependent appearance shifts. Vision State Space Models (SSMs) provide efficient long-range modeling with linear complexity, but existing Vision Mamba segmentation models typically convert 2D features into 1D scanning sequences, which may weaken local geometric continuity and over-smooth irregular contours. We propose CSG-Mamba, a convolutional scoring gating Vision State Space network for endoscopic polyp segmentation. Built on a VM-UNet-style asymmetric U-shaped encoder-decoder, CSG-Mamba inserts a Convolutional Scoring Gating (CSG) module at the semantically rich bottleneck. CSG generates a local spatial score map through pointwise and large-kernel depthwise convolutions and recalibrates state-space features by multiplicative gating. Experiments with three random seeds show that CSG-Mamba achieves 0.9220 Dice and 15.87 HD95 on Kvasir-SEG, and 0.7418 Dice and 0.6570 mIoU on CVC-ColonDB, outperforming the baselines on most overlap and recall metrics while maintaining competitive boundary accuracy.
Figures & tables
| Model | mIoU | Dice | HD95 | Accuracy | Sensitivity | Params |
|---|---|---|---|---|---|---|
| U-Net | 31.04M | |||||
| ResUNet | 63.85M | |||||
| PraNet | 24.51M | |||||
| TransUNet | 99.04M | |||||
| VM-UNet | 27.43M | |||||
| VM-UNetV2 | 27.63M |
| Model | mIoU | Dice | HD95 | Accuracy | Sensitivity |
|---|---|---|---|---|---|
| U-Net | |||||
| ResUNet | |||||
| PraNet | |||||
| TransUNet | |||||
| VM-UNet | |||||
| VM-UNetV2 |
| Model | mIoU | Dice | HD95 | Accuracy | Sensitivity |
|---|---|---|---|---|---|
| U-Net | |||||
| ResUNet | |||||
| PraNet | |||||
| TransUNet | |||||
| VM-UNet | |||||
| VM-UNetV2 |
| Bottleneck variant | Kvasir HD95 | Kvasir Dice | ClinicDB HD95 | ClinicDB Dice | ColonDB HD95 | ColonDB Dice | Params |
|---|---|---|---|---|---|---|---|
| None | 27.43M | ||||||
| DW + Add | 32.18M | ||||||
| DW + Gate | 32.18M | ||||||
| DW + Add | 32.30M | ||||||
| DW + Gate (CSG) | 32.30M |
| Position | Kvasir HD95 | Kvasir Dice | ClinicDB HD95 | ClinicDB Dice | ColonDB HD95 | ColonDB Dice | Params |
|---|---|---|---|---|---|---|---|
| none | 27.43M | ||||||
| encoder only | 29.11M | ||||||
| decoder only | 33.94M | ||||||
| bottleneck only | 32.30M | ||||||
| all | 40.50M |
Polyp Segmentation
Polyp segmentation uses computer vision to delineate colorectal polyps in colonoscopy images and video for detection and clinical assessment.
See all 23 papers