LLM2D
SAMRI-2:一种基于记忆的模型,用于膝关节3D MRI中的软骨和半月板分割
SAMRI-2: A Memory-based Model for Cartilage and Meniscus Segmentation in 3D MRIs of the Knee Joint
作者: Danielle L. Ferreira, Bruno A. A. Nunes, Xuzhe Zhang, Laura Carretero Gomez, Maggie Fung, Ravi Soni
发布日期: 2/18/2025
arXiv ID: oai:arXiv.org:2502.10559v1

摘要

arXiv:2502.10559v1 Announce Type: cross 摘要:通过MRI准确评估软骨的形态学特征,如厚度/体积,对于监测膝骨关节炎至关重要。软骨分割仍然是一个具有挑战性的问题,依赖于大量的专家注释数据集,这些数据集受到阅读者间变异性的严重影响。最近视觉基础模型(VFM)的进步,特别是基于记忆的方法,为提高通用性和鲁棒性提供了机会。本研究提出了一种深度学习(DL)方法,利用交互式、基于记忆的视觉基础模型从3D MRI中进行软骨和半月板分割。为了提高空间意识和收敛性,我们在训练过程中引入了混合洗牌策略(HSS),并应用了分割掩模传播技术以增强注释效率。我们在270名患者的3D膝关节MRI上训练了四种AI模型——基于CNN的3D-VNet、两个自动变压器模型(SaMRI2D和SaMRI3D),以及一个基于变压器的可提示记忆型视觉基础模型(SAMRI-2)。我们使用公共和内部数据集进行训练,并在57个外部病例上进行评估,这些病例包括多放射科医生注释和不同的数据采集方式。使用Dice分数(DSC)和交集与并集比(IoU)对模型性能进行了评估,并通过额外的形态学评估进一步量化分割准确性。具有HSS训练的SAMRI-2模型表现优于所有其他模型,平均DSC提高了5分,对于胫骨软骨的最大改进达到12分。此外,它还展示了最低的软骨厚度误差,减少了一至三倍的差异。值得注意的是,SAMRI-2在每次体积仅需三下用户点击的情况下仍能保持高性能,减少了注释工作量同时确保了解剖精度。这种具有空间意识的记忆型视觉基础模型为可靠的AI辅助膝关节MRI分割提供了一种新颖的方法,在肌肉骨骼成像中推进了深度学习技术。