{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/mamba-unet-unet-like-pure-visual-mamba-for","title":"Mamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation","arxiv_id":"2402.05079","date":"2024-02-07","proceeding":null,"authors":["Ziyang Wang","Jian-Qing Zheng","Yichi Zhang","Ge Cui","Lei LI"],"abstract":"In recent advancements in medical image analysis, Convolutional Neural Networks (CNN) and Vision Transformers (ViT) have set significant benchmarks. While the former excels in capturing local features through its convolution operations, the latter achieves remarkable global context understanding by leveraging self-attention mechanisms. However, both architectures exhibit limitations in efficiently modeling long-range dependencies within medical images, which is a critical aspect for precise segmentation. Inspired by the Mamba architecture, known for its proficiency in handling long sequences and global contextual information with enhanced computational efficiency as a State Space Model (SSM), we propose Mamba-UNet, a novel architecture that synergizes the U-Net in medical image segmentation with Mamba's capability. Mamba-UNet adopts a pure Visual Mamba (VMamba)-based encoder-decoder structure, infused with skip connections to preserve spatial information across different scales of the network. This design facilitates a comprehensive feature learning process, capturing intricate details and broader semantic contexts within medical images. We introduce a novel integration mechanism within the VMamba blocks to ensure seamless connectivity and information flow between the encoder and decoder paths, enhancing the segmentation performance. We conducted experiments on publicly available ACDC MRI Cardiac segmentation dataset, and Synapse CT Abdomen segmentation dataset. The results show that Mamba-UNet outperforms several types of UNet in medical image segmentation under the same hyper-parameter setting. The source code and baseline implementations are available.","url_abs":"https://arxiv.org/abs/2402.05079v2","url_pdf":"https://arxiv.org/pdf/2402.05079v2.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"mamba-unet-unet-like-pure-visual-mamba-for","repo_url":"https://github.com/ziyangwang007/mamba-unet","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":null}],"tasks":[{"task_slug":"cardiac-segmentation","task_name":"Cardiac Segmentation"},{"task_slug":"computational-efficiency","task_name":"Computational Efficiency"},{"task_slug":"decoder","task_name":"Decoder"},{"task_slug":"image-segmentation","task_name":"Image Segmentation"},{"task_slug":"mamba","task_name":"Mamba"},{"task_slug":"medical-image-analysis","task_name":"Medical Image Analysis"},{"task_slug":"medical-image-segmentation","task_name":"Medical Image Segmentation"},{"task_slug":"segmentation","task_name":"Segmentation"},{"task_slug":"semantic-segmentation","task_name":"Semantic Segmentation"}],"methods":[{"method_slug":"concatenated-skip-connection","method_name":"Concatenated Skip Connection"},{"method_slug":"convolution","method_name":"Convolution"},{"method_slug":"max-pooling","method_name":"Max Pooling"},{"method_slug":"relu","method_name":"ReLU"},{"method_slug":"set","method_name":"SET"},{"method_slug":"u-net","method_name":"U-Net"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2402.05079","atlas_url":"https://app.syntology.ai/?focus=2402.05079","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2402.05079"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/ziyangwang007/mamba-unet","reach":null}],"summary":{"ran_draft_wrong":2,"ran_honours":3},"by_repo_kind":{"official":{"samples":5,"ran":5,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"13f59065a97aef7e","entry":"kaiming_normal_init_weight","repo":"ziyangwang007/mamba-unet","repo_kind":"official","path":"code/train_Semi_Mamba_UNet.py","file_url":"https://github.com/ziyangwang007/mamba-unet/blob/HEAD/code/train_Semi_Mamba_UNet.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"13f59065a97aef7e"}},{"code_sha256_prefix":"cab70404b0657b24","entry":"patients_to_slices","repo":"ziyangwang007/mamba-unet","repo_kind":"official","path":"code/train_fully_supervised_2D_VIM.py","file_url":"https://github.com/ziyangwang007/mamba-unet/blob/HEAD/code/train_fully_supervised_2D_VIM.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"cab70404b0657b24"}},{"code_sha256_prefix":"f53c4c3922632c64","entry":"patients_to_slices","repo":"ziyangwang007/mamba-unet","repo_kind":"official","path":"code/train_Semi_Mamba_UNet.py","file_url":"https://github.com/ziyangwang007/mamba-unet/blob/HEAD/code/train_Semi_Mamba_UNet.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f53c4c3922632c64"}},{"code_sha256_prefix":"333ad2f6c5aa26e7","entry":"patients_to_slices","repo":"ziyangwang007/mamba-unet","repo_kind":"official","path":"code/train_uncertainty_aware_mean_teacher_2D.py","file_url":"https://github.com/ziyangwang007/mamba-unet/blob/HEAD/code/train_uncertainty_aware_mean_teacher_2D.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"333ad2f6c5aa26e7"}},{"code_sha256_prefix":"b49bacf6c71fb02c","entry":"xavier_normal_init_weight","repo":"ziyangwang007/mamba-unet","repo_kind":"official","path":"code/train_Semi_Mamba_UNet.py","file_url":"https://github.com/ziyangwang007/mamba-unet/blob/HEAD/code/train_Semi_Mamba_UNet.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"b49bacf6c71fb02c"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}