用噪声标注训练模型,自动补全准确的医学图像分割标签。
Label Filling via Mixed Supervision for Medical Image Segmentation from Noisy Annotations
- 通过可信像素监督+混合补全策略,从多个标注者噪声数据中学习真实标签。
- 在五个不同模态数据集上均提升分割精度,多发性硬化病灶分割提升7%。
- 适合标注质量不一的医学图像任务,尤其适用于标注成本高的场景。
医学图像分割的成功通常依赖大量高质量标注,但标注者技能和特征差异导致标注存在显著一致性偏差。本文提出一种简单有效的标签补全框架LF-Net,仅使用噪声标注即可预测真实分割标签。核心思想是:利用可信像素(经阈值多数投票筛选)提供确定性监督,同时通过两类混合辅助监督补全其余像素标签——一是从噪声标注内在结构学习的软标签,二是传播各标注者特性的标签。该方法具有两大优势:1)仅用可信像素训练,确保监督信心,引导真实标签学习;2)双重混合监督防止过拟合,保障与真实标签高保真度。在五种不同成像模态的五个数据集上验证,相比现有最佳方法,本方法在所有数据集上均提升分割性能,其中多发性硬化病灶分割的骰子系数(DSC)提升达7%。
原文摘要 · Abstract (English)
The success of medical image segmentation usually requires a large number of high-quality labels. But since the labeling process is usually affected by the raters' varying skill levels and characteristics, the estimated masks provided by different raters usually suffer from high inter-rater variability. In this paper, we propose a simple yet effective Label Filling framework, termed as LF-Net, predicting the groundtruth segmentation label given only noisy annotations during training. The fundamental idea of label filling is to supervise the segmentation model by a subset of pixels with trustworthy labels, meanwhile filling labels of other pixels by mixed supervision. More concretely, we propose a qualified majority voting strategy, i.e., a threshold voting scheme is designed to model agreement among raters and the majority-voted labels of the selected subset of pixels are regarded as supervision. To fill labels of other pixels, two types of mixed auxiliary supervision are proposed: a soft label learned from intrinsic structures of noisy annotations, and raters' characteristics labels which propagate individual rater's characteristics information. LF-Net has two main advantages. 1) Training with trustworthy pixels incorporates training with confident supervision, guiding the direction of groundtruth label learning. 2) Two types of mixed supervision prevent over-fitting issues when the network is supervised by a subset of pixels, and guarantee high fidelity with the true label. Results on five datasets of diverse imaging modalities show that our LF-Net boosts segmentation accuracy in all datasets compared with state-of-the-art methods, with even a 7% improvement in DSC for MS lesion segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。