VMMedSAM-X: A State-Enhanced Dual-Branch Encoder for Efficient Promptable Medical Image Segmentation
Source abstract
Abstract
Medical image segmentation plays a crucial role in clinical diagnosis and treatment planning. However, existing segmentation frameworks frequently exhibit high computational complexity and often fail to retain fine-grained structural details—especially along intricate anatomical boundaries such as blood vessels and tumor margins. To overcome these limitations, we propose VMMedSAM-X, an efficient and computationally economical medical image segmentation framework that incorporates structured state space modeling into the Medical Segment Anything Model (MedSAM) architecture. The proposed method adopts a state-enhanced encoder that combines extended long short-term memory (xLSTM) with two-dimensional selective scanning (SS2D) and a dual-path cross-attention mechanism to enhance long-range dependency modeling while maintaining linear computational complexity. Experiments conducted on the 1024×1024 ACDC cardiac MRI dataset show that the proposed encoder reduces floating-point operations from 369.44 G to 17.36 G and achieves a 2.4× improvement in inference speed compared with the Vision Transformer (ViT)-based encoder. Additional evaluations on the SegTHOR and MSD-Lung datasets demonstrate consistent improvements in Dice Similarity Coefficient (DSC) and Intersection over Union (IoU) metrics over MedSAM and Vision Mamba U-Net (VM-UNet) baselines. These results indicate that the proposed framework provides an effective and computationally efficient solution for high-resolution medical image segmentation.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Medical Imaging and Analysis · Generative Adversarial Networks and Image Synthesis
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
College of Artificial Intelligence, North China University of Science and Technology, Tangshan 063210, China
References (47)
Show source-provided references
1. Xu, G., Udupa, J.K., Luo, J., Zhao, S., Yu, Y., Raymond, S.B., Peng, H., Ning, L., Rathi, Y., and Liu, W. (2025). Is the medical image segmentation problem solved? A survey of current developments and future directions. arXiv.
2. A survey on deep learning in medical image analysis10.1016/j.media.2017.07.005 ↗
3. Ronneberger, O., Fischer, P., and Brox, T. (2015). U-Net: Convolutional networks for biomedical image segmentation. International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), Springer.10.1007/978-3-319-24574-4_28 ↗
4. Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., and Ronneberger, O. (2016). 3D U-Net: Learning dense volumetric segmentation from sparse annotation. International Conference on Medical Image Computing and Computer-Assisted Intervention, Springer.10.1007/978-3-319-46723-8_49 ↗
5. Advances in medical image segmentation: A comprehensive survey with a focus on lumbar spine applications10.1016/j.compbiomed.2025.111171 ↗
6. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2021, January 5). An image is worth 16 × 16 words: Transformers for image recognition at scale. Proceedings of the International Conference on Learning Representations (ICLR), Virtual.
7. Chen, J., Lu, Y., Yu, Q., Luo, X., Adeli, E., Wang, Y., Lu, L., Yuille, A.L., and Zhou, Y. (2021). TransUNet: Transformers make strong encoders for medical image segmentation. arXiv.
8. Azad, R., Rauf, Z., Khan, A.R., Rathore, S., Khan, S.H., Shah, N.S., Farooq, U., Asif, H., Asif, A., and Zahoora, U. (2024). A recent survey of vision transformers for medical image segmentation. arXiv.
9. Advancements in medical image segmentation: A review of transformer models10.1016/j.compeleceng.2025.110099 ↗
10. Kirillov, A., Mintun, E., Ravi, N., Mao, H., Rolland, C., Gustafson, L., Xiao, T., Whitehead, S., Berg, A.C., and Lo, W.-Y. (2023). Segment Anything. Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), IEEE.10.1109/ICCV51070.2023.00371 ↗
11. Segment anything in medical images10.1038/s41467-024-44824-z ↗
12. Fan, W., Li, A., Xu, M., Sun, W., and Man, F. (2026). Practical application of SAM for breast nodules segmentation. Front. Oncol., 16.10.3389/fonc.2026.1756011 ↗
13. Necessity and impact of specialization of large foundation model for medical segmentation tasks10.1002/mp.17470 ↗
14. Awad, M.A., Mabrouk, M.S., and Elnokrashy, A.F. (2025). Medical image segmentation using transformer encoders and prompt-based learning: A systematic review of adaptation strategies, performance, and challenges. Proceedings of the 2025 Twelfth International Conference on Intelligent Computing and Information Systems (ICICIS), IEEE.10.1109/ICICIS66182.2025.11313186 ↗
15. Gu, A., and Dao, T. (2023). Mamba: Linear-time sequence modeling with selective state spaces. arXiv.
16. Zhang, R., Guo, H., Tian, K., Zhou, J., Yan, M., Zhang, Z., and Zhao, S. (2025). Unified medical image segmentation with state space modeling snake. Proceedings of the 33rd ACM International Conference on Multimedia (ACM MM), Association for Computing Machinery.10.1145/3746027.3755148 ↗
17. ÆMMamba: An efficient medical segmentation model with edge enhancement10.1109/JBHI.2025.3572088 ↗
18. MambaMIM: Pre-training Mamba with state space token interpolation and its application to medical image segmentation10.1016/j.media.2025.103606 ↗
19. xLSTM: Extended long short-term memory
20. Are Vision-xLSTM-embedded U-Nets better at segmenting medical images?10.1016/j.neunet.2025.107925 ↗
21. Deep learning in medical image analysis10.1146/annurev-bioeng-071516-044442 ↗
22. UNet++: Redesigning skip connections to exploit multiscale features in image segmentation10.1109/TMI.2019.2959609 ↗
23. Oktay, O., Schlemper, J., Folgoc, L.L., Lee, M., Heinrich, M., Misawa, K., Mori, K., McDonagh, S., Hammerla, N.Y., and Kainz, B. (2018). Attention U-Net: Learning where to look for the pancreas. arXiv.
24. Deep learning architectures for medical image segmentation: An organized analysis of CNN-based models and uses10.11591/eei.v15i1.10839 ↗
25. Oliveira, J.V.S., Vieira, D.F., Silva, M.P., Fernandes, D.L., Ribeiro, M.H.F., and Oliveira, H.N. (2025). Strategies for deep learning in volumetric medical imaging: A survey. Proceedings of SIBGRAPI, IEEE.10.1109/SIBGRAPI67909.2025.11223395 ↗
26. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021). Swin Transformer: Hierarchical vision transformer using shifted windows. Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), IEEE.10.1109/ICCV48922.2021.00986 ↗
27. Efficient transformers: A survey
28. Systematic review of hybrid vision transformer architectures for radiological image analysis10.1007/s10278-024-01322-4 ↗
29. Generalist models in medical image segmentation: A survey and performance comparison with task-specific approaches10.1016/j.inffus.2025.103709 ↗
30. A narrative review of foundation models for medical image segmentation: Zero-shot performance evaluation on diverse modalities10.21037/qims-2024-2826 ↗
31. Ayllon, E.M., Mantegna, M., Shen, L., Soda, P., Guarrasi, V., and Tortora, M. (2025). Can foundation models really segment tumors? A benchmarking odyssey in lung CT imaging. 2025 IEEE 38th International Symposium on Computer-Based Medical Systems (CBMS), IEEE.10.1109/CBMS65348.2025.00083 ↗
32. Lee, H.H., Gu, Y., Zhao, T., Xu, Y., Yang, J., Usuyama, N., Wong, C., Wei, M., Landman, B.A., and Huo, Y. (2024). Foundation Models for Biomedical Image Segmentation: A Survey. arXiv.
33. Gu, A., Goel, K., and Ré, C. (2022, January 25–29). Efficiently modeling long sequences with structured state spaces. Presented at the International Conference on Learning Representations (ICLR), Virtual Event.
34. Liu, Y., Tian, Y., Zhao, Y., Yu, H., Xie, L., Wang, Y., Ye, Q., Jiao, J., and Liu, Y. (2024). VMamba: Visual state space model. arXiv.
35. Zhu, L., Liao, B., Zhang, Q., Wang, X., Liu, W., and Wang, X. (2024). Vision Mamba: Efficient visual representation learning with bidirectional state space model. arXiv.
36. From channel-spatial attention to state space models: A review of evolving mechanisms in tumour segmentation10.1002/ctd2.70127 ↗
37. Long short-term memory10.1162/neco.1997.9.8.1735 ↗
38. Qayyum, A., Mazher, M., and Niederer, S.A. (2024). Assessing self-supervised xLSTM-UNet architectures for head and neck tumor segmentation. Challenge on Head and Neck Tumor Segmentation for MRI-Guided Applications, Springer.10.1007/978-3-031-83274-1_12 ↗
39. Deep sequential segmentation of organs in volumetric medical scans10.1109/TMI.2018.2881678 ↗
40. One shot model for COVID-19 classification and lesions segmentation in chest CT scans using long short-term memory network with attention mechanism10.1109/MIS.2021.3135474 ↗
41. Attention mechanisms in computer vision: A survey10.1007/s41095-022-0271-y ↗
42. Liu, X., Zhang, C., and Zhang, L. (2024). Vision Mamba: A comprehensive survey and taxonomy. arXiv.
43. Lambert, Z., Petitjean, C., Dubray, B., and Kuan, S. (2020). SegTHOR: Segmentation of thoracic organs at risk in CT images. Proceedings of the 2020 Tenth International Conference on Image Processing Theory, Tools and Applications (IPTA), IEEE.10.1109/IPTA50016.2020.9286453 ↗
44. AMOS: A large-scale abdominal multi-organ benchmark for versatile medical image segmentation
45. Deep learning techniques for automatic MRI cardiac multi-structures segmentation and diagnosis: Is the problem solved?10.1109/TMI.2018.2837502 ↗
46. The medical segmentation decathlon10.1038/s41467-022-30695-9 ↗
47. Heller, N., Isensee, F., Tejpaul, R., Wood, A., Papanikolopoulos, N., and Weight, C. (2024). The 2023 Kidney and Kidney Tumor Segmentation Challenge (KiTS23). Kidney and Kidney Tumor Segmentation: MICCAI 2023 Challenge, KiTS 2023, Springer.10.1007/978-3-031-54806-2 ↗