arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.12814cs.CV

从少到多:基于掩码上下文建模和连续伪标签的涂鸦医学图像分割

From Few to More: Scribble-based Medical Image Segmentation via Masked Context Modeling and Continuous Pseudo Labels

National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, School of Computer Science and Engineering, Northwestern Polytechnical University(集成空天地海大数据应用技术国家工程实验室,计算机科学与工程学院,西北工业大学)

Zhisong Wang, Yiwen Ye, Ziyang Chen, Minglei Shu, Yanning Zhang, Yong Xia

AI总结:

本文提出弱监督医学图像分割模型MaCo,通过掩码上下文建模和连续伪标签,有效处理稀疏标注挑战,在三个公开数据集上均优于现有方法并创下新纪录。

摘要:

基于涂鸦的弱监督分割方法在医学图像分割中展现出令人鼓舞的结果,显著降低了标注成本。然而,现有方法通常依赖辅助任务来强制语义一致性,并使用硬伪标签进行监督,忽略了使用稀疏标注训练的模型所面临的独特挑战。这些模型必须从有限数据中预测逐像素的分割图,因此有效处理不同水平的标注丰富度至关重要。本文提出MaCo,一种基于“从少到多”原则设计的用于医学图像分割的弱监督模型。MaCo利用掩码上下文建模(MCM)和连续伪标签(CPL)。MCM采用基于注意力的掩码策略来扰动输入图像,确保模型的预测与原始图像的预测对齐。CPL通过对距离图应用指数衰减函数,将涂鸦标注转换为连续的逐像素标签,生成表示每个像素属于特定类别可能性的置信度图,而非依赖硬伪标签。我们在三个公开数据集上评估MaCo,并将其与其他弱监督方法进行比较。结果表明,MaCo在所有数据集上均优于竞争方法,在弱监督医学图像分割中创下新纪录。

英文摘要:

Scribble-based weakly supervised segmentation methods have shown promising results in medical image segmentation, significantly reducing annotation costs. However, existing approaches often rely on auxiliary tasks to enforce semantic consistency and use hard pseudo labels for supervision, overlooking the unique challenges faced by models trained with sparse annotations. These models must predict pixel-wise segmentation maps from limited data, making it crucial to handle varying levels of annotation richness effectively. In this paper, we propose MaCo, a weakly supervised model designed for medical image segmentation, based on the principle of "from few to more." MaCo leverages Masked Context Modeling (MCM) and Continuous Pseudo Labels (CPL). MCM employs an attention-based masking strategy to perturb the input image, ensuring that the model's predictions align with those of the original image. CPL converts scribble annotations into continuous pixel-wise labels by applying an exponential decay function to distance maps, producing confidence maps that represent the likelihood of each pixel belonging to a specific category, rather than relying on hard pseudo labels. We evaluate MaCo on three public datasets, comparing it with other weakly supervised methods. Our results show that MaCo outperforms competing methods across all datasets, establishing a new record in weakly supervised medical image segmentation.

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

↑