用区域归一化方法提升噪声标注下的医学图像分割精度。
Region-Normalized DPO for Medical Image Segmentation under Noisy Judges
- 基于区域差异归一化,缓解噪声标注带来的有害更新。
- 在两个医学数据集上均优于标准DPO和强基线模型。
- 适合无额外标注时的医疗图像分割模型优化场景。
尽管密集像素级标注仍是医学图像分割的黄金标准,但其获取成本高且难以扩展。相比之下,现有系统已能生成低成本的自动质量控制信号,如模型一致性、不确定性度量或学习到的掩码质量分数,这些可直接用于模型训练而无需额外标注。然而,这些信号常存在噪声和偏差,使基于偏好微调易受有害更新影响。本文研究了在小规模标注集上训练的监督基础分割器生成提议下,使用噪声判别器进行直接偏好优化(DPO)的方法。发现偏好对的挖掘方式显著影响结果:选择判别器最推荐的提议在判别器可靠时可提升峰值性能,但在弱判别器下会放大错误。为此提出区域归一化DPO(RN-DPO),通过掩码间不一致区域大小对偏好更新进行归一化,降低有害比较的影响,增强优化稳定性。在两个医学数据集及多种设置下,RN-DPO显著提升持续性能并稳定偏好微调过程,优于标准DPO与强基线,且无需额外像素标注。
原文摘要 · Abstract (English)
While dense pixel-wise annotations remain the gold standard for medical image segmentation, they are costly to obtain and limit scalability. In contrast, many deployed systems already produce inexpensive automatic quality-control (QC) signals like model agreement, uncertainty measures, or learned mask-quality scores which can be used for further model training without additional ground-truth annotation. However, these signals can be noisy and biased, making preference-based fine-tuning susceptible to harmful updates. We study Direct Preference Optimization (DPO) for segmentation from such noisy judges using proposals generated by a supervised base segmenter trained on a small labeled set. We find that outcomes depend strongly on how preference pairs are mined: selecting the judge's top-ranked proposal can improve peak performance when the judge is reliable, but can amplify harmful errors under weaker judges. We propose Region-Normalized DPO (RN-DPO), a segmentation-aware objective which normalizes preference updates by the size of the disagreement region between masks, reducing the leverage of harmful comparisons and improving optimization stability. Across two medical datasets and multiple regimes, RN-DPO improves sustained performance and stabilizes preference-based fine-tuning, outperforming standard DPO and strong baselines without requiring additional pixel annotations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。