WCEDSAM: A Lightweight Multi-Scale Colonoscopy Polyp-Segmentation Network Combining Frequency-Domain Decomposition...

Colorectal cancer screening is challenged by variations in polyp morphology, indistinct polyp boundaries, and the high computational costs associated with current models. To address these issues, a lightweight medical image segmentation model, WCEDSAM, has been developed. WCEDSAM is based on a modified, compact version of MedSAM, which incorporates a Wavelet Transform-based component to extract and separate overlapping features at the pixel level. Additionally, a DSConv-ECA module is positioned before the ViT encoder to capture local features efficiently while reducing parameter count and enhancing inter-channel communication. Experimental results demonstrate that WCEDSAM achieves top performance on five public datasets, including Kvasir-SEG and CVC-ClinicDB, with 15.38 million parameters, achieving mean Dice (mDice) scores of 0.9383 on Kvasir-SEG and 0.9376 on CVC-ClinicDB. Cross-domain evaluations yield mDice scores of 0.9189 on CVC-ColonDB, 0.8961 on CVC-300, and 0.7765 on ETIS datasets, respectively, substantially outperforming other methods such as UNet++ and TransUNet.