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