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

论文：《SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation》（arXiv 2024）

论文贡献：

  • 设计了ToM模块，用以增强三维特征的顺序建模
  • 设计了门控空间卷积模块（GSC)，用以增强每个ToM之前空间维度上的特征表示

网络结构：

image-20240507091527951

1.Stem

采用深度卷积，内核大小为7×7×7，填充为3×3×3，步幅为2×2×2。

2.TSMamba块

计算过程如下：

image-20240507091715749

注：其中GSC和ToM分别表示所提出的门控空间卷积模块和三向Mamba模块

3.门控空间卷积(GSC)

门控空间卷积(GSC)用于提取mamba层之前的空间关系

image-20240507092130322

计算过程如下：

image-20240507092149524

4.三向mamba（ToM)

从三个方向计算特征依赖关系，将三维输入特征平铺成三个序列，进行相应的特征交互，得到融合后的三维特征

计算过程如下：

image-20240507092415793