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

MRI医学图像分割中的域影响:空间域与k空间输入

Domain Influence in MRI Medical Image Segmentation: spatial versus k-space inputs

Erik Gösche, Reza Eghbali, Florian Knoll, Andreas M Rauschecker

AI总结:

本文比较空间域与k空间输入对MRI分割模型的影响,发现k空间可显著提升注意力及MLP模型性能,并使额外位置编码不再有益。

摘要:

应用于图像块的基于Transformer的网络已在许多视觉任务中取得前沿性能。然而,由于缺乏卷积神经网络(CNN)针对局部图像统计特征的内置偏置,它们需要大型数据集和修改才能捕捉图像块之间的关系,尤其是在分割任务中。频域中的图像可能更适合注意力机制,因为局部特征得到全局表示。通过将图像变换到频域,局部特征被全局表示。由于MRI数据采集特性,这些图像尤其适合。本工作研究图像域(空间域或k空间)如何影响深度学习(DL)模型的分割结果,重点关注基于注意力的网络以及基于MLP的其他非卷积模型。我们还考察了当输入图像处于频域时,基于Transformer的网络是否需要额外的位置编码。为进行评估,我们设置了颅骨剥离任务和脑组织分割任务。所使用的基于注意力的模型是PerceiverIO和一个普通Transformer编码器。为与非基于注意力的模型比较,还训练并测试了一个MLP和ResMLP。结果与最先进的医学图像分割模型Swin-Unet进行比较。实验结果表明,使用k空间作为输入域可显著改善分割结果。此外,如果输入处于频域,额外位置编码似乎对基于注意力的网络并无益处。尽管没有任何模型达到Swin-Unet的性能,但复杂度较低的模型通过选择不同的域展现出有前景的提升。

英文摘要:

Transformer-based networks applied to image patches have achieved cutting-edge performance in many vision tasks. However, lacking the built-in bias of convolutional neural networks (CNN) for local image statistics, they require large datasets and modifications to capture relationships between patches, especially in segmentation tasks. Images in the frequency domain might be more suitable for the attention mechanism, as local features are represented globally. By transforming images into the frequency domain, local features are represented globally. Due to MRI data acquisition properties, these images are particularly suitable. This work investigates how the image domain (spatial or k-space) affects segmentation results of deep learning (DL) models, focusing on attention-based networks and other non-convolutional models based on MLPs. We also examine the necessity of additional positional encoding for Transformer-based networks when input images are in the frequency domain. For evaluation, we pose a skull stripping task and a brain tissue segmentation task. The attention-based models used are PerceiverIO and a vanilla Transformer encoder. To compare with non-attention-based models, an MLP and ResMLP are also trained and tested. Results are compared with the Swin-Unet, the state-of-the-art medical image segmentation model. Experimental results show that using k-space for the input domain can significantly improve segmentation results. Also, additional positional encoding does not seem beneficial for attention-based networks if the input is in the frequency domain. Although none of the models matched the Swin-Unet's performance, the less complex models showed promising improvements with a different domain choice.

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

↑