LLM2D
VesselSAM:利用SAM进行主动脉血管分割的方法,带LoRA和空洞注意力机制
VesselSAM: Leveraging SAM for Aortic Vessel Segmentation with LoRA and Atrous Attention
作者: Adnan Iltaf, Rayan Merghani Ahmed, Zhenxi Zhang, Bin Li, Shoujun Zhou
发布日期: 3/27/2025
arXiv ID: oai:arXiv.org:2502.18185v3

摘要

arXiv:2502.18185v3 通告类型: replace-cross 摘要:医学图像分割对于临床诊断和治疗计划至关重要,尤其是在处理复杂的解剖结构如血管时。然而,准确分割血管仍然具有挑战性,因为它们的小尺寸、复杂的边缘结构以及容易受到伪影和成像噪声的影响。在这项工作中,我们提出了VesselSAM,这是一种改进的Segment Anything Model (SAM) 版本,特别适用于主动脉血管分割。VesselSAM 结合了 AtrousLoRA 模块,该模块将空洞注意和低秩适配 (LoRA) 集成在一起,以增强分割性能。空洞注意使模型能够捕获多尺度上下文信息,同时保留细微的局部细节和更广泛的全局上下文。此外,LoRA 促进了冻结 SAM 图像编码器的高效微调,减少了可训练参数的数量,从而增强了计算效率。我们使用两个具有挑战性的数据集对 VesselSAM 进行了评估:主动脉血管树 (AVT) 数据集和型 B 主动脉夹层 (TBAD) 数据集。VesselSAM 达到了最先进的性能,多中心数据集的 DSC 得分为 93.50\%, 93.25\%, 93.02\% 和 93.26\%。我们的结果表明,VesselSAM 提供了高分割准确率,在计算开销方面也显著优于现有的大规模模型。这一发展为进一步提高临床环境中的基于 AI 的主动脉血管分割奠定了基础。该代码和模型将在 https://github.com/Adnan-CAS/AtrousLora 释放。