Light-VM2D-UNet: A Lightweight UNet Enhanced with Mamba2D for Medical Image Segmentation
This paper addresses the fundamental trade-off between global long-range dependency modeling and local detail preservation in lightweight medical image segmentation models. Through systematic analysis of the limitations in spatial dependency capturing exhibited by the parallel vision Mamba (PVM) component in UltraLight VM-UNet procedure, we propose a novel PVM2DRes Block. This module integrates a 2D wavefront scanning strategy into the PVM unit while incorporating local contextual enhancement during parallel state space updates. The proposed approach enables multi-directional and multi-scale global-local feature interactions with maintained linear computational complexity. The resulting Light-VM2D-UNet achieves superior performance with merely 0.12 M parameters and 0.07 GFLOPs, demonstrating consistent improvements in accuracy and mIoU metrics across three benchmark datasets: ISIC-2017, ISIC-2018, and CVC-ClinicDB. Extensive experiments validate that our method not only enhances the modeling capability for long-range feature dependencies and fine-grained local details but also establishes the generalizability of integrating state space models with attention mechanisms. This work provides a more efficient and accurate technical pathway for intelligent diagnostic systems in resource-constrained environments.
Authors
- Zhongyuan Peng
- Huang Tan (ORCID: https://orcid.org/0000-0002-9739-0003)
Institutions
- Hunan Normal University (CN)
- Hunan University of Humanities, Science and Technology (CN)
- Changsha Normal University (CN)
Publication Details
- Published
- 2025-06-20
- DOI
- https://doi.org/10.1109/prmvai65741.2025.11108528
- Primary Topic
- Medical Image Segmentation Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00