用分阶段强化学习修复带噪标注,提升医学图像分割精度
Staged Voxel-Level Deep Reinforcement Learning for 3D Medical Image Segmentation with Noisy Annotations
- 将每个体素视为独立智能体,通过动态迭代优化修正错误标注
- 在三个公开数据集上平均提升Dice和IoU超3%,达到当前最优水平
- 适合处理标注质量不一的医学图像分割任务,尤其适用于有噪声数据场景
深度学习在医学图像分割中取得显著进展,但高质量标注数据依赖严重。由于器官形态复杂及标注者差异,噪声标注普遍存在,严重影响模型性能。受医生可基于先验知识修正标注的启发,本文提出端到端的分阶段体素级深度强化学习框架(SVL-DRL),通过动态迭代更新自动缓解错误标签影响,无需人工干预。其核心创新包括:将噪声标注建模为体素相关问题,采用新颖的分阶段强化学习框架确保模型稳健收敛;引入体素级异步优势演员-评论家(vA3C)模块,使每个体素作为独立智能体在训练中动态优化自身状态表示,直接抑制错误标签干扰;设计新型动作空间与复合奖励函数,融合Dice系数与空间连续性度量,显著提升分割精度并保持语义一致性。在三个公开医学图像数据集上的实验表明,该方法在多种设置下均达到当前最优性能,平均Dice和IoU分数提升超过3%。
原文摘要 · Abstract (English)
Deep learning has achieved significant advancements in medical image segmentation. Currently, obtaining accurate segmentation outcomes is critically reliant on large-scale datasets with high-quality annotations. However, noisy annotations are frequently encountered owing to the complex morphological structures of organs in medical images and variations among different annotators, which can substantially limit the efficacy of segmentation models. Motivated by the fact that medical imaging annotator can correct labeling errors during segmentation based on prior knowledge, we propose an end-to-end Staged Voxel-Level Deep Reinforcement Learning (SVL-DRL) framework for robust medical image segmentation under noisy annotations. This framework employs a dynamic iterative update strategy to automatically mitigate the impact of erroneous labels without requiring manual intervention. The key advancements of SVL-DRL over existing works include: i) formulating noisy annotations as a voxel-dependent problem and addressing it through a novel staged reinforcement learning framework which guarantees robust model convergence; ii) incorporating a voxel-level asynchronous advantage actor-critic (vA3C) module that conceptualizes each voxel as an autonomous agent, which allows each agent to dynamically refine its own state representation during training, thereby directly mitigating the influence of erroneous labels; iii) designing a novel action space for the agents, along with a composite reward function that strategically combines the Dice value and a spatial continuity metric to significantly boost segmentation accuracy while maintain semantic integrity. Experiments on three public medical image datasets demonstrates State-of-The-Art (SoTA) performance under various experimental settings, with an average improvement of over 3\% in both Dice and IoU scores.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。