SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xing, Zhaohu, Ye, Tian, Yang, Yijun, Liu, Guang, Zhu, Lei
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929499844640768
author Xing, Zhaohu
Ye, Tian
Yang, Yijun
Liu, Guang
Zhu, Lei
author_facet Xing, Zhaohu
Ye, Tian
Yang, Yijun
Liu, Guang
Zhu, Lei
contents The Transformer architecture has shown a remarkable ability in modeling global relationships. However, it poses a significant computational challenge when processing high-dimensional medical images. This hinders its development and widespread adoption in this task. Mamba, as a State Space Model (SSM), recently emerged as a notable manner for long-range dependencies in sequential modeling, excelling in natural language processing filed with its remarkable memory efficiency and computational speed. Inspired by its success, we introduce SegMamba, a novel 3D medical image \textbf{Seg}mentation \textbf{Mamba} model, designed to effectively capture long-range dependencies within whole volume features at every scale. Our SegMamba, in contrast to Transformer-based methods, excels in whole volume feature modeling from a state space model standpoint, maintaining superior processing speed, even with volume features at a resolution of {$64\times 64\times 64$}. Comprehensive experiments on the BraTS2023 dataset demonstrate the effectiveness and efficiency of our SegMamba. The code for SegMamba is available at: https://github.com/ge-xing/SegMamba
format Preprint
id arxiv_https___arxiv_org_abs_2401_13560
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation
Xing, Zhaohu
Ye, Tian
Yang, Yijun
Liu, Guang
Zhu, Lei
Computer Vision and Pattern Recognition
The Transformer architecture has shown a remarkable ability in modeling global relationships. However, it poses a significant computational challenge when processing high-dimensional medical images. This hinders its development and widespread adoption in this task. Mamba, as a State Space Model (SSM), recently emerged as a notable manner for long-range dependencies in sequential modeling, excelling in natural language processing filed with its remarkable memory efficiency and computational speed. Inspired by its success, we introduce SegMamba, a novel 3D medical image \textbf{Seg}mentation \textbf{Mamba} model, designed to effectively capture long-range dependencies within whole volume features at every scale. Our SegMamba, in contrast to Transformer-based methods, excels in whole volume feature modeling from a state space model standpoint, maintaining superior processing speed, even with volume features at a resolution of {$64\times 64\times 64$}. Comprehensive experiments on the BraTS2023 dataset demonstrate the effectiveness and efficiency of our SegMamba. The code for SegMamba is available at: https://github.com/ge-xing/SegMamba
title SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2401.13560