Polyp Segmentation Based on Mamba-MLP Mixer and Transformer
摘要
Polyps, or abnormal tissue growths on the mucous membranes of various organs, have the potential to become cancerous, leading to dangerous conditions such as colorectal cancer. In recent decades, deep learning methods, particularly neural networks, have consistently outperformed traditional methods in identifying polyps from colonoscopy and endoscopy images—tasks that were previously time-consuming. Polyp segmentation is a crucial first step that helps doctors analyze and provide timely intervention or appropriate treatment regimens when necessary. This process has been significantly enhanced by the development of cutting-edge neural network models, particularly those based on Transformers and CNNs. However, considering the significant differences in polyp size, form, and appearance of polyps, polyp datasets present a significant challenge, often requiring methods that integrate both Transformer and convolutional approaches for optimized performance. Motivated by this challenge, as a solution, we present an innovative architecture combining Transformer, convolution, and Mamba-based components, referred to as MixMamba-TransNet. We designed the Mamba-MLP Mixer and Convolution Gate blocks for the model’s decoding process, along with incorporating a deep supervision strategy to achieve promising results on the CVC-ClinicDB and Kvasir-SEG databases. Code is available at: https://github.com/ndhieu2oo3/MixMamba-TransNet .