UNet with Self-Adaptive Mamba-Like Attention and Causal-Resonance Learning for Medical Image Segmentation
On the map: State-space sequence models (S4, Mamba) → Mamba for medical image analysis
Abstract (arXiv)
Medical image segmentation plays an important role in various clinical applications; however, existing deep learning models face trade-offs between efficiency and accuracy. Convolutional Neural Networks (CNNs) capture local details well but miss the global context, whereas transformers handle the global context but at a high computational cost. Recently, State Space Sequence Models (SSMs) have shown potential for capturing long-range dependencies with linear complexity; however, their direct use in medical image segmentation remains limited due to incompatibility with image structures and autoregressive assumptions. To overcome these challenges, we propose SAMA-UNet, a novel U-shaped architecture that introduces two key innovations. First, the Self-Adaptive Mamba-like Aggregated Attention (SAMA) block adaptively integrates local and global features through dynamic attention weighting, enabling an efficient representation of complex anatomical patterns. Second, the causal resonance multi-scale module (CR-MSM) improves encoder-decoder interactions by adjusting feature resolution and causal dependencies across scales, enhancing the semantic alignment between low- and high-level features. Extensive experiments on MRI, CT, and endoscopy datasets demonstrate that SAMA-UNet consistently outperforms CNN, Transformer, and Mamba-based methods. It achieves 85.38% DSC and 87.82% NSD on BTCV, 92.16% and 96.54% on ACDC, 67.14% and 68.70% on EndoVis17, and 84.06% and 88.47% on ATLAS23, establishing new benchmarks across modalities. These results confirm the effectiveness of SAMA-UNet in combining efficiency and accuracy, making it a promising solution for real-world clinical segmentation tasks. The source code is available on GitHub.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Mamba for medical image analysis, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 28 of its 51 references that are in the map. First posted on arXiv 21 May 2025.
Builds on
28 of 51 references.
Closest ancestors · 7
Same direction, newest first
- 2024MLLA-UNet: Mamba-like Linear Attention in an Efficient U-Shape Model for Medical Image Segmentation31 Oct
- 2024MSVM-UNet: Multi-Scale Vision Mamba UNet for Medical Image Segmentation25 Aug
- 2024VM-UNET-V2 Rethinking Vision Mamba UNet for Medical Image Segmentation14 Mar
- 2024LKM-UNet: Large Kernel Vision Mamba UNet for Medical Image Segmentation12 Mar
- 2024Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining5 Feb
- 2024VM-UNet: Vision Mamba UNet for Medical Image Segmentation4 Feb
- 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation9 Jan
Other references · 21
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2020Deformable DETR: Deformable Transformers for End-to-End Object Detectioncited by 3,422
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
- 2024VMamba: Visual State Space Modelcited by 969
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2016Language Modeling with Gated Convolutional Networkscited by 702
- 2022Vision Transformer with Deformable Attentioncited by 295
9 more
- 2023LongNet: Scaling Transformers to 1,000,000,000 Tokenscited by 204
- 2022Metrics reloaded: Recommendations for image analysis validationcited by 195
- 2022Medical Image Segmentation Review: The success of U-Netcited by 191
- 2023FLatten Transformer: Vision Transformer using Focused Linear Attentioncited by 171
- 20192017 Robotic Instrument Segmentation Challengecited by 168
- 2024MambaOut: Do We Really Need Mamba for Vision?cited by 143
- 2024Demystify Mamba in Vision: A Linear Attention Perspectivecited by 102
- 2025A Survey on Mamba Architecture for Vision Applicationscited by 8
- 2024FER-YOLO-Mamba: Facial Expression Detection and Classification Based on Selective State Spacecited by 7
Built on by
Citing papers per year: 2025: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024Mamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation
- 2025ACM-UNet: Adaptive Integration of CNNs and Mamba for Efficient Medical Image Segmentation
- 2025Prompt-Guided Dual-Path UNet with Mamba for Medical Image Segmentation
- 2025MM-UNet: Meta Mamba UNet for Medical Image Segmentation
- 2024Rotate to Scan: UNet-like Mamba with Triplet SSM Module for Medical Image Segmentation
- 2024Semi-Mamba-UNet: Pixel-Level Contrastive and Pixel-Level Cross-Supervised Visual Mamba-based UNet for Semi-Supervised Medical Image Segmentation
- 2024LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation
- 2026A Hybrid Mamba-SAM Architecture for Efficient 3D Medical Image Segmentation