arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04308cs.CV

通过元学习增强用于脑肿瘤分割的模态无关表示

Enhancing Modality-Agnostic Representations via Meta-Learning for Brain Tumor Segmentation

Stony Brook University(石溪大学)

Aishik Konwer, Xiaoling Hu, Joseph Bae, Xuan Xu, Chao Chen, Prateek Prasanna

AI总结:

针对医学影像训练与推理中模态缺失问题,提出结合元学习与辅助对抗分支的方法,增强模态无关表示,在脑肿瘤分割缺失模态场景下性能超越现有最优技术。

摘要:

在医学视觉领域,不同的成像模态可提供互补信息。但在实际应用中,推理阶段甚至训练阶段可能无法获取所有模态。以往的方法,例如知识蒸馏或图像合成,通常假设训练阶段所有患者的完整模态数据均可用;由于不同站点数据采集存在差异性,这一假设既不现实也不具备实用性。我们提出一种新方法,即便仅能获取有限的完整模态样本,也能在训练中采用元学习策略来学习增强的模态无关表示。元学习通过在部分模态数据上进行元训练、在有限的完整模态样本上进行元测试,将部分模态表示增强为完整模态表示。此外,我们引入一个辅助对抗学习分支来对该特征增强过程进行共同监督。更具体地说,我们使用一个缺失模态检测器作为判别器,来模拟完整模态设置。我们的分割框架在缺失模态场景下的性能显著优于当前最先进的脑肿瘤分割技术。

英文摘要:

In medical vision, different imaging modalities provide complementary information. However, in practice, not all modalities may be available during inference or even training. Previous approaches, e.g., knowledge distillation or image synthesis, often assume the availability of full modalities for all patients during training; this is unrealistic and impractical due to the variability in data collection across sites. We propose a novel approach to learn enhanced modality-agnostic representations by employing a meta-learning strategy in training, even when only limited full modality samples are available. Meta-learning enhances partial modality representations to full modality representations by meta-training on partial modality data and meta-testing on limited full modality samples. Additionally, we co-supervise this feature enrichment by introducing an auxiliary adversarial learning branch. More specifically, a missing modality detector is used as a discriminator to mimic the full modality setting. Our segmentation framework significantly outperforms state-of-the-art brain tumor segmentation techniques in missing modality scenarios.

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

↑