arXiv:2411.11105cs.CVcs.AI2024-11被引 1

构建共享标签空间,用单一模型解决多个医学图像多标签分割任务。

Label Sharing Incremental Learning Framework for Independent Multi-Label Segmentation Tasks

  • 通过映射不同数据集标签到统一公共标签空间,实现多任务共享。
  • 单模型在多个医疗图像数据集上均达良好分割性能,且训练参数更少。
  • 天然支持增量学习,新增数据集无需重构模型结构。

在需为多个具有独立标签集的数据集构建分割模型的场景中,传统方法需为每个数据集单独训练模型或使用共享编码器+多头结构。本文提出一种新型标签共享框架:构建一个共享的公共标签空间,并将各数据集的独立标签系统性映射至该空间,从而将多个异构标签集转换为一个统一标签的大数据集。由此,所有分割任务可由单一模型统一处理,避免了针对特定任务的网络结构调整,实现参数与数据效率提升。此外,该框架天然适用于增量学习,新数据集的分割能力可便捷地逐步添加。我们在多个涉及多标签分割的医学图像数据集上验证了该方法的有效性,实验表明其性能优于现有方法,且具备出色的增量学习能力。

原文摘要 · Abstract (English)

In a setting where segmentation models have to be built for multiple datasets, each with its own corresponding label set, a straightforward way is to learn one model for every dataset and its labels. Alternatively, multi-task architectures with shared encoders and multiple segmentation heads or shared weights with compound labels can also be made use of. This work proposes a novel label sharing framework where a shared common label space is constructed and each of the individual label sets are systematically mapped to the common labels. This transforms multiple datasets with disparate label sets into a single large dataset with shared labels, and therefore all the segmentation tasks can be addressed by learning a single model. This eliminates the need for task specific adaptations in network architectures and also results in parameter and data efficient models. Furthermore, label sharing framework is naturally amenable for incremental learning where segmentations for new datasets can be easily learnt. We experimentally validate our method on various medical image segmentation datasets, each involving multi-label segmentation. Furthermore, we demonstrate the efficacy of the proposed method in terms of performance and incremental learning ability vis-a-vis alternative methods.

多标签分割增量学习医学图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。