English

Unified Medical Image Segmentation with State Space Modeling Snake

Computer Vision and Pattern Recognition 2026-03-10 v2 Artificial Intelligence

Abstract

Unified Medical Image Segmentation (UMIS) is critical for comprehensive anatomical assessment but faces challenges due to multi-scale structural heterogeneity. Conventional pixel-based approaches, lacking object-level anatomical insight and inter-organ relational modeling, struggle with morphological complexity and feature conflicts, limiting their efficacy in UMIS. We propose Mamba Snake, a novel deep snake framework enhanced by state space modeling for UMIS. Mamba Snake frames multi-contour evolution as a hierarchical state space atlas, effectively modeling macroscopic inter-organ topological relationships and microscopic contour refinements. We introduce a snake-specific vision state space module, the Mamba Evolution Block (MEB), which leverages effective spatiotemporal information aggregation for adaptive refinement of complex morphologies. Energy map shape priors further ensure robust long-range contour evolution in heterogeneous data. Additionally, a dual-classification synergy mechanism is incorporated to concurrently optimize detection and segmentation, mitigating under-segmentation of microstructures in UMIS. Extensive evaluations across five clinical datasets reveal Mamba Snake's superior performance, with an average Dice improvement of 3\% over state-of-the-art methods.

Keywords

Cite

@article{arxiv.2507.12760,
  title  = {Unified Medical Image Segmentation with State Space Modeling Snake},
  author = {Ruicheng Zhang and Haowei Guo and Kanghui Tian and Jun Zhou and Mingliang Yan and Zeyu Zhang and Shen Zhao},
  journal= {arXiv preprint arXiv:2507.12760},
  year   = {2026}
}

Comments

This paper has been accepted by ACM MM 2025

R2 v1 2026-07-01T04:05:24.529Z