大家好,今天想给大家介绍一个可以应用在mamba中的卷积模块,叫做门控空间卷积Gated Spatial Convolution (GSC),这个模块出自最新的医学影像顶会MICCAI2024。 论文的名字叫做:SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation.在论文中,Mamba 层通过将 3D 特征展平为 1D 序列来对特征依赖性进行建模。 因此,作者为了提取 Mamba 层之前的空间关系,设计了门控空间卷积(GSC)模块,3D特征被输入到两个卷积块(每个卷积块包含一个InstanceNorm、一个卷积层和一个非线性层),卷积核大小分别为3×3×3和1 ×1×1。然后将这两个特征逐像素相乘,以类似于门控机制的方式控制信息传输。 最后,使用卷积块进一步融合特征,同时使用残差连接来融合输入特征和输出特征。up在这里提供了代码的2d版本,供大家使用,感谢您的观看!