用双多边形标注消除医学图像分割中的标注不确定性
EAUWSeg: Eliminating annotation uncertainty in weakly-supervised medical image segmentation
- 用两个多边形标注病灶,降低标注模糊性
- 通过对抗监督学习提取不变特征,提升模型鲁棒性
- 结合分类信息与像素一致性,增强不确定区域的分割精度
弱监督医学图像分割因只需粗略标注而非精确像素级标签,显著减轻专家工作负担,正受到广泛关注。尽管已有进展,但其性能仍明显落后于全监督方法,主要源于弱标签本身的不确定性。为此,本文提出一种新型弱标注方法及学习框架EAUWSeg,以消除标注不确定性。首先,引入有界多边形标注(BPAnno),仅需对病灶标注两个多边形即可;其次,设计针对性的学习机制,将有界多边形视为两个独立标注,通过对抗监督信号训练模型,学习不变特征;随后,构建置信度辅助一致性学习器,结合分类引导的置信度生成器,利用同类像素间特征一致性及有界多边形所包含的类别信息,为不确定区域提供可靠监督信号。实验表明,EAUWSeg优于现有弱监督分割方法;相比全监督模型,不仅性能更优,且标注成本大幅降低,充分验证了该方法的有效性与优越性。
原文摘要 · Abstract (English)
Weakly-supervised medical image segmentation is gaining traction as it requires only rough annotations rather than accurate pixel-to-pixel labels, thereby reducing the workload for specialists. Although some progress has been made, there is still a considerable performance gap between the label-efficient methods and fully-supervised one, which can be attributed to the uncertainty nature of these weak labels. To address this issue, we propose a novel weak annotation method coupled with its learning framework EAUWSeg to eliminate the annotation uncertainty. Specifically, we first propose the Bounded Polygon Annotation (BPAnno) by simply labeling two polygons for a lesion. Then, the tailored learning mechanism that explicitly treat bounded polygons as two separated annotations is proposed to learn invariant feature by providing adversarial supervision signal for model training. Subsequently, a confidence-auxiliary consistency learner incorporates with a classification-guided confidence generator is designed to provide reliable supervision signal for pixels in uncertain region by leveraging the feature presentation consistency across pixels within the same category as well as class-specific information encapsulated in bounded polygons annotation. Experimental results demonstrate that EAUWSeg outperforms existing weakly-supervised segmentation methods. Furthermore, compared to fully-supervised counterparts, the proposed method not only delivers superior performance but also costs much less annotation workload. This underscores the superiority and effectiveness of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。