arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.01658cs.CVcs.AI

EAUWSeg:消除弱监督医学图像分割中的标注不确定性

EAUWSeg: Eliminating annotation uncertainty in weakly-supervised medical image segmentation

College of Computer Science, Sichuan University(四川大学计算机学院);Institute of High Performance Computing, A*STAR(新加坡科技研究局高性能计算研究所)

Wang Lituan, Zhang Lei, Wang Yan, Wang Zhenbin, Zhang Zhenwei, Zhang Yi

AI总结:

针对弱监督医学图像分割中标注不确定性问题,提出有界多边形标注BPAnno和EAUWSeg学习框架,通过对抗监督、置信度辅助一致性学习实现更优分割性能并显著降低标注工作量。

摘要:

弱监督医学图像分割正受到越来越多的关注,因为它只需要粗略标注而非精确的逐像素标签,从而减轻了专家的工作负担。尽管已经取得了一些进展,但标签高效方法与全监督方法之间仍存在相当大的性能差距,这可以归因于这些弱标签本身的不确定性。为了解决这一问题,我们提出了一种新的弱标注方法及其学习框架 EAUWSeg,以消除标注不确定性。具体而言,我们首先提出有界多边形标注(Bounded Polygon Annotation,BPAnno),只需为一个病灶标注两个多边形。然后,我们提出了一种定制化的学习机制,将有界多边形显式地视为两个独立的标注,通过为模型训练提供对抗性监督信号来学习不变特征。随后,设计了一个置信度辅助一致性学习器,并结合一个分类引导的置信度生成器,通过利用同一类别内像素之间的特征表示一致性以及有界多边形标注中蕴含的类别特定信息,为不确定区域中的像素提供可靠的监督信号。实验结果表明,EAUWSeg 优于现有的弱监督分割方法。此外,与全监督方法相比,所提出的方法不仅性能更优,而且标注工作量也大大减少。这充分证明了我们方法的优越性和有效性。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑