arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.17675cs.CV

EM-Net:基于Mamba的3D医学图像分割高效通道与频率学习

EM-Net: Efficient Channel and Frequency Learning with Mamba for 3D Medical Image Segmentation

National-Regional Key Technology Engineering Laboratory for Medical Ultrasound(国家区域医疗超声关键技术工程实验室);School of Biomedical Engineering(生物医学工程学院);Medical School(医学院);Shenzhen University(深圳大学);Medical Ultrasound Image Computing (MUSIC) Lab(医学超声图像计算实验室);Marshall Laboratory of Biomedical Engineering(马歇尔生物医学工程实验室)

Ao Chang, Jiajun Zeng, Ruobing Huang, Dong Ni

AI总结:

本文提出基于Mamba的3D医学图像分割模型EM-Net,通过通道选择与频域学习协调多尺度特征并加速训练,在多器官数据集上以近半参数量和2倍速度实现更高精度。

摘要:

卷积神经网络主要引领了3D医学图像分割,但可能受限于较小的感受野。Transformer模型通过自注意力机制擅长捕捉全局关系,但在高分辨率下面临高计算成本的挑战。最近,作为状态空间模型的Mamba已成为序列建模的有效方法。受其成功启发,我们引入了一种基于Mamba的新型3D医学图像分割模型EM-Net。它不仅通过整合和选择通道高效捕捉区域间的注意力交互,还有效利用频域来协调不同尺度间特征的学习,同时加快训练速度。在两个具有挑战性的多器官数据集上与其他最先进(SOTA)算法进行的综合实验表明,我们的方法展现出更好的分割精度,同时所需的参数量约为SOTA模型的一半,训练速度提高了2倍。

英文摘要:

Convolutional neural networks have primarily led 3D medical image segmentation but may be limited by small receptive fields. Transformer models excel in capturing global relationships through self-attention but are challenged by high computational costs at high resolutions. Recently, Mamba, a state space model, has emerged as an effective approach for sequential modeling. Inspired by its success, we introduce a novel Mamba-based 3D medical image segmentation model called EM-Net. It not only efficiently captures attentive interaction between regions by integrating and selecting channels, but also effectively utilizes frequency domain to harmonize the learning of features across varying scales, while accelerating training speed. Comprehensive experiments on two challenging multi-organ datasets with other state-of-the-art (SOTA) algorithms show that our method exhibits better segmentation accuracy while requiring nearly half the parameter size of SOTA models and 2x faster training speed.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑