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

RefSAM3D:采用跨模态参考适配SAM以实现3D医学图像分割

RefSAM3D: Adapting SAM with Cross-modal Reference for 3D Medical Image Segmentation

Nanjing Drum Tower Hospital, Nanjing University(南京大学医学院附属鼓楼医院)

Xiang Gao, Kai Lu

AI总结:

针对SAM难以适配3D医学影像的问题,提出RefSAM3D模型,通过3D适配器、跨模态参考提示、分层注意力等改进,在多数据集上实现了优于现有方法的3D医学图像分割性能。

摘要:

Segment Anything Model(SAM,分割一切模型)最初基于2D视觉Transformer(ViT)构建,擅长捕捉2D自然图像中的全局模式,但在CT和MRI等3D医学成像模态中表现不佳。这类模态需要捕捉体空间中的空间信息,以完成器官分割、肿瘤量化等任务。为解决这一挑战,我们提出RefSAM3D,该模型通过引入3D图像适配器和跨模态参考提示生成,将SAM适配至3D医学成像场景。我们的方法修改了视觉编码器以处理3D输入,并改进了掩码解码器以直接生成3D掩码。我们还整合了文本提示,以提升复杂解剖场景下的分割精度与一致性。通过采用分层注意力机制,我们的模型能有效捕捉并整合不同尺度的信息。在多个医学成像数据集上的广泛评估表明,RefSAM3D的性能优于现有最优方法。我们的贡献推动了SAM在医学影像复杂解剖结构精准分割中的应用。

英文摘要:

The Segment Anything Model (SAM), originally built on a 2D Vision Transformer (ViT), excels at capturing global patterns in 2D natural images but struggles with 3D medical imaging modalities like CT and MRI. These modalities require capturing spatial information in volumetric space for tasks such as organ segmentation and tumor quantification. To address this challenge, we introduce RefSAM3D, which adapts SAM for 3D medical imaging by incorporating a 3D image adapter and cross-modal reference prompt generation. Our approach modifies the visual encoder to handle 3D inputs and enhances the mask decoder for direct 3D mask generation. We also integrate textual prompts to improve segmentation accuracy and consistency in complex anatomical scenarios. By employing a hierarchical attention mechanism, our model effectively captures and integrates information across different scales. Extensive evaluations on multiple medical imaging datasets demonstrate the superior performance of RefSAM3D over state-of-the-art methods. Our contributions advance the application of SAM in accurately segmenting complex anatomical structures in medical imaging.

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

↑