翻转学习:擦除以分割
计算机视觉与模式识别
2021-08-03 v1 人工智能
机器学习
多智能体系统
摘要
从乳腺超声图像中分割结节具有挑战性,但对诊断至关重要。弱监督分割(WSS)有助于减少耗时且繁琐的手动标注。与现有弱监督方法不同,在本研究中,我们提出一种新颖且通用的称为 Flip Learning 的 WSS 框架,其仅需框标注。具体而言,标签框内的目标将被逐步擦除以翻转分类标签,最终被擦除的区域将被视为分割结果。我们的贡献有三方面。首先,我们提出的方案利用多智能体强化学习框架在超像素级别进行擦除,以利用先验边界知识并加速学习过程。其次,我们设计两种奖励:分类得分奖励与强度分布奖励,分别用于避免欠分割与过分割。第三,我们采用由粗到细的学习策略以减少残差误差并提升分割性能。在大型数据集上的广泛验证表明,我们提出的方法取得了具有竞争力的性能,并展现出缩小全监督与弱监督学习之间差距的巨大潜力。
引用
@article{arxiv.2108.00752,
title = {Flip Learning: Erase to Segment},
author = {Yuhao Huang and Xin Yang and Yuxin Zou and Chaoyu Chen and Jian Wang and Haoran Dou and Nishant Ravikumar and Alejandro F Frangi and Jianqiao Zhou and Dong Ni},
journal= {arXiv preprint arXiv:2108.00752},
year = {2021}
}
备注
Accepted by MICCAI 2021