摘要
arXiv:2503.20685v1 交叉公告类型:core-formal
摘要:在2D乳腺超声(BUS)和3D自动化乳腺超声(ABUS)中准确分割结节对于临床诊断和治疗规划至关重要。因此,开发自动化的结节分割系统可以提高用户的独立性并加速临床分析。与全监督学习不同,弱监督分割(WSS)可以简化繁琐且复杂的标注过程。然而,当前的WSS方法在实现精确的结节分割方面面临挑战,因为许多方法依赖于不准确的激活图或低效的伪掩码生成算法。在这项研究中,我们提出了一种基于多智能体强化学习的WSS框架,称为翻转学习(Flip Learning),该框架仅依赖于2D/3D框进行精确分割。具体而言,使用多个智能体擦除目标以促进分类标签翻转,擦除区域作为预测分割掩码。本文研究的主要贡献如下:(1) 使用基于超像素/监督体的方法来编码标准化环境,捕捉边界先验并加速学习过程。(2) 引入了三项精心设计的奖励,包括分类分数奖励以及两种强度分布奖励,以精确引导智能体的擦除过程,从而避免出现欠分割和过分割。(3) 实现了一种分阶段的课程学习策略,使智能体能够以渐进挑战的方式与环境交互,从而提高学习效率。在大规模内部BUS和ABUS数据集上广泛验证后,我们的翻转学习方法优于当前最先进的WSS方法和基础模型,并且在性能上可与全监督学习算法相媲美。