摘要
arXiv:2504.07117v1 Announce Type: cross
摘要:在白内障手术中,准确的手术器械分割对于技能评估和工作流程优化等任务至关重要。然而,有限的标注数据使得开发完全自动的模型变得困难。类似于SAM2的提示方法虽然具有灵活性,但仍然对点提示的位置非常敏感,常常导致不一致的分割结果。为了解决这个问题,我们引入了RP-SAM2,它结合了一个新颖的移动块和复合损失函数,以稳定点提示。我们的方法减少了标注人员对精确点定位的依赖,同时保持了稳健的分割能力。在Cataract1k数据集上的实验表明,RP-SAM2提高了分割准确性,mDSC提高了2%,mHD95减少了21.36%,并且在随机单点提示结果中降低了偏差,相较于SAM2。此外,在CaDIS数据集上,RP-SAM2生成的伪掩码用于微调SAM2的掩码解码器,其表现优于SAM2生成的伪掩码。这些结果突显了RP-SAM2在数据受限的医学环境中半自动器械分割问题中具有实用、稳定和可靠的解决方案。代码可在 https://github.com/BioMedIA-MBZUAI/RP-SAM2 获取。