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

部分监督多器官医学图像分割中的有标签到无标签分布对齐

Labeled-to-Unlabeled Distribution Alignment for Partially-Supervised Multi-Organ Medical Image Segmentation

The Hong Kong University of Science and Technology(香港科技大学);Huazhong University of Science and Technology(华中科技大学)

Xixi Jiang, Dong Zhang, Xiang Li, Kangyi Liu, Kwang-Ting Cheng, Xin Yang

AI总结:

本文针对部分监督多器官分割中有标签与无标签像素分布不匹配的问题,提出LTUDA框架,通过跨集合数据增强和基于原型的分布对齐提升判别能力,在多个数据集上超越现有部分监督甚至全监督方法。

摘要:

部分监督多器官医学图像分割旨在利用多个部分标注的数据集开发一个统一的语义分割模型,其中每个数据集仅为单一类别的器官提供标签。然而,有标签前景器官的有限可用性以及缺乏用于区分无标签前景器官与背景的监督,构成了一个重大挑战,这导致有标签像素与无标签像素之间存在分布不匹配。尽管现有的伪标签方法可以用于同时从有标签和无标签像素中学习,但它们在此任务中容易性能下降,因为它们依赖于有标签像素和无标签像素具有相同分布的假设。在本文中,为了解决分布不匹配问题,我们提出了一种有标签到无标签分布对齐(LTUDA)框架,该框架对齐特征分布并增强判别能力。具体而言,我们引入了一种跨集合数据增强策略,该策略在有标签和无标签器官之间进行区域级混合,以减少分布差异并丰富训练集。此外,我们提出了一种基于原型的分布对齐方法,该方法隐式地减少类内变化,并增大无标签前景与背景之间的分离度。这可以通过鼓励两个原型分类器和一个线性分类器的输出之间的一致性来实现。在AbdomenCT-1K数据集以及四个基准数据集(包括LiTS、MSD-Spleen、KiTS和NIH82)的联合数据集上的大量实验结果表明,我们的方法以相当大的优势优于最先进的部分监督方法,甚至超过了全监督方法。源代码公开于https://github.com/xjiangmed/LTUDA。

英文摘要:

Partially-supervised multi-organ medical image segmentation aims to develop a unified semantic segmentation model by utilizing multiple partially-labeled datasets, with each dataset providing labels for a single class of organs. However, the limited availability of labeled foreground organs and the absence of supervision to distinguish unlabeled foreground organs from the background pose a significant challenge, which leads to a distribution mismatch between labeled and unlabeled pixels. Although existing pseudo-labeling methods can be employed to learn from both labeled and unlabeled pixels, they are prone to performance degradation in this task, as they rely on the assumption that labeled and unlabeled pixels have the same distribution. In this paper, to address the problem of distribution mismatch, we propose a labeled-to-unlabeled distribution alignment (LTUDA) framework that aligns feature distributions and enhances discriminative capability. Specifically, we introduce a cross-set data augmentation strategy, which performs region-level mixing between labeled and unlabeled organs to reduce distribution discrepancy and enrich the training set. Besides, we propose a prototype-based distribution alignment method that implicitly reduces intra-class variation and increases the separation between the unlabeled foreground and background. This can be achieved by encouraging consistency between the outputs of two prototype classifiers and a linear classifier. Extensive experimental results on the AbdomenCT-1K dataset and a union of four benchmark datasets (including LiTS, MSD-Spleen, KiTS, and NIH82) demonstrate that our method outperforms the state-of-the-art partially-supervised methods by a considerable margin, and even surpasses the fully-supervised methods. The source code is publicly available at https://github.com/xjiangmed/LTUDA.

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

↑