ZR2ViM: A Recursive Vision Mamba Model for Boundary-Preserving Medical Image Segmentation

Caijian Hua¹Caorong Xiang¹Liuying Li^2*Xia Zhou^2*

• ¹Sichuan University of Science and Engineering, Zigong, China
• ²Zigong First People's Hospital, Zigong, China

Introduction: Medical image segmentation is fundamental to quantitative disease analysis and therapeutic decision-making. However, constrained by limited computational resources, existing deep learning methods often struggle to simultaneously model long-range dependencies and preserve boundary precision, particularly when delineating structures with complex morphology or blurred edges. Method: To overcome these challenges, we propose ZR2ViM, a recursion-enhanced visual state space model designed for medical image segmentation. ZR2ViM augments the Vision Mamba framework with a Zigzag Recursive Reinforced (ZR2) Block that incorporates Stacked State Redistribution (SSR) and a Nested Recursive Connection (NRC). The NRC employs dual inner and outer pathways to iteratively fuse local details with global context while preserving 2D spatial adjacency. Furthermore, a Cross-directional Zigzag WKV (CZ-WKV) module executes multi-step recursive updates along multiple zigzag trajectories, injecting spatial directional information via Quad-Directional Token Shift(Q-Shift) directional priors. Collectively, these mechanisms mitigate serialization-induced banding artifacts and enhance the representation of fine, elongated, and low-contrast structures, all while maintaining near-linear computational complexity. Results: Comprehensive evaluations across four medical imaging domains—spanning dermatoscopic images, breast ultrasound, colorectal polyps, and abdominal multi-organ CT—on five public datasets demonstrate that ZR2ViM consistently outperforms representative convolutional, attention-based, and visual state space architectures in region consistency and boundary localization. Notably, ZR2ViM achieves a 2.15 mm reduction in the HD95 on the Synapse multi-organ CT dataset relative to the CC-ViM baseline, substantiating its superior capability for precise, clinically relevant boundary delineation. Sample et al. Running Title Conclusion: The ZR2ViM framework delivers accurate, boundary-preserving segmentation across diverse imaging modalities and anatomically complex structures, achieving these gains with near-linear computational complexity. These findings demonstrate that ZR2ViM offers a robust and efficient solution for medical image analysis, establishing a promising foundation for advanced clinical and research applications.