A parallel UNet integrating KAN and mamba for medical image segmentation
Medical image segmentation is fundamental for delineating lesion and organ boundaries in clinical workflows. While UNet-based models remain widely used, CNN-dominant designs are limited in modeling long-range context, and Transformer-based variants often introduce substantial computational overhead due to quadratic attention. To address this issue, we propose KMP-UNet, a parallel U-shaped framework that combines a Mamba-based state-space branch for linear-complexity contextual modeling and a Kolmogorov-Arnold Network (KAN) branch for nonlinear feature representation. We further introduce a task-oriented fusion block and a skip refinement module to better exploit hierarchical encoder-decoder features. KMP-UNet has a compact model size (about 1.0M parameters in our implementation). We evaluate the proposed method on four public datasets (ISIC2017, ISIC2018, CVC-ClinicDB, and BUSI) using standard segmentation metrics. On ISIC2018, KMP-UNet achieves 0.9038 DSC and 0.9600 accuracy under our protocol. Extensive comparisons and targeted ablations are conducted to analyze the contribution of each component.
Authors
- Wenxu Shi (ORCID: https://orcid.org/0000-0001-6229-7995)
- Liming Xu (ORCID: https://orcid.org/0000-0002-0671-8182)
- Jiayi Liu (ORCID: https://orcid.org/0000-0003-1819-2998)
- Bochuan Zheng
- Jiabao Wu
Institutions
- Army Medical University (CN)
- China West Normal University (CN)
Publication Details
- Journal
- Scientific Reports
- Published
- 2026-03-15
- DOI
- https://doi.org/10.1038/s41598-026-43127-1
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00