3 个月前

CycleMix:一种基于涂鸦监督的医学图像分割整体策略

CycleMix:一种基于涂鸦监督的医学图像分割整体策略

摘要

构建大规模完全标注的训练数据集成本高昂,尤其是在医学图像分割任务中尤为明显。涂鸦标注(scribble)作为一种较弱形式的标注,在实际应用中更容易获取,但仅依赖有限的涂鸦监督信息来训练分割模型仍面临挑战。为应对这些困难,本文提出一种基于涂鸦学习的医学图像分割新框架——CycleMix,该框架融合了混合增强(mix augmentation)与循环一致性(cycle consistency)机制。在监督信息增强方面,CycleMix采用改进的mixup策略,并引入专门设计的随机遮挡机制,实现对涂鸦标注的增补与删减;在监督信息正则化方面,CycleMix通过引入一致性损失(consistency loss)强化训练目标,有效惩罚不一致的分割结果,显著提升了分割性能。在两个公开数据集ACDC和MSCMRseg上的实验结果表明,所提方法取得了令人振奋的性能表现,其分割精度可与甚至优于完全监督方法相媲美。相关代码及MSCMRseg数据集的专家标注涂鸦已公开发布于:https://github.com/BWGZK/CycleMix。

代码仓库

bwgzk/cyclemix
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semantic-segmentation-on-acdc-scribblesCycleMix
Dice (Average): 84.8%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CycleMix:一种基于涂鸦监督的医学图像分割整体策略 | 论文 | HyperAI超神经