arXiv:2410.06892eess.IVcs.CV2024-10被引 1

通过任务相似性分析,找到最适合的医学图像分割迁移路径。

Selecting the Best Sequential Transfer Path for Medical Image Segmentation with Limited Labeled Data

  • 基于图像与标签相似性计算任务亲和度,选择最佳迁移顺序。
  • 在三个MRI数据集上平均提升2.58%分割准确率,最高达6.00%。
  • 适合标注数据少的医学图像分割场景,尤其对跨任务迁移有效。

医学图像处理常面临标注数据稀缺的问题。尽管迁移学习被广泛采用,但如何选择合适的源任务并有效传递知识仍具挑战。为此,本文提出一种针对医学图像的序列迁移方案,引入任务亲和度度量。结合医学图像分割任务特性,分析任务间的图像与标签相似性,计算任务亲和度分数以评估任务相关性。基于此,选择恰当的源任务,并通过引入中间源任务构建渐进式迁移策略,逐步缩小领域差异,降低迁移成本,从而确定给定目标任务的最佳序列迁移路径。在三组MRI数据集FeTS 2022、iSeg-2019和WMH上的大量实验表明,该方法能有效找到最优源任务序列。相比直接从单一源任务迁移,序列迁移显著提升目标任务性能,平均分割Dice得分提高2.58%,其中FeTS 2022提升达6.00%。代码已开源。

原文摘要 · Abstract (English)

The medical image processing field often encounters the critical issue of scarce annotated data. Transfer learning has emerged as a solution, yet how to select an adequate source task and effectively transfer the knowledge to the target task remains challenging. To address this, we propose a novel sequential transfer scheme with a task affinity metric tailored for medical images. Considering the characteristics of medical image segmentation tasks, we analyze the image and label similarity between tasks and compute the task affinity scores, which assess the relatedness among tasks. Based on this, we select appropriate source tasks and develop an effective sequential transfer strategy by incorporating intermediate source tasks to gradually narrow the domain discrepancy and minimize the transfer cost. Thereby we identify the best sequential transfer path for the given target task. Extensive experiments on three MRI medical datasets, FeTS 2022, iSeg-2019, and WMH, demonstrate the efficacy of our method in finding the best source sequence. Compared with directly transferring from a single source task, the sequential transfer results underline a significant improvement in target task performance, achieving an average of 2.58% gain in terms of segmentation Dice score, notably, 6.00% for FeTS 2022. Code is available at the git repository.

医学图像迁移学习分割序列迁移

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