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

从适应到泛化:面向医学图像分割的自适应视觉提示

From Adaptation to Generalization: Adaptive Visual Prompting for Medical Image Segmentation

Technical University of Munich(慕尼黑技术大学);Korea University(韩国大学);Kyung Hee University(庆熙大学);Seoul National University of Science and Technology(首尔科学技术大学)

Evren Çetinkaya, Sangmin Lee, Jung Uk Kim, Hong Joo Lee, Nassir Navab

AI总结:

本文提出APEX框架,通过学习可微的提示记忆来提取输入特定的提示,以提升医学图像分割在跨域和内域变化中的泛化能力。

摘要:

视觉提示已成为一种强大的方法,用于在不更新模型参数的情况下将预训练模型适应到新领域。然而,现有提示方法通常为每个领域优化一个提示并统一应用于所有输入,限制了其在内在和跨域变异性下的泛化能力,这在医学领域尤为重要。为此,我们提出APEX,一种自适应提示提取框架,从可学习的提示记忆中检索输入特定的提示。该记忆存储了多样且领域判别性的提示表示,并通过从傅里叶谱提取的领域特征进行查询。为了学习鲁棒且判别性的领域特征,我们引入了一种新的低频特征对比学习(LFC)框架,该框架在相同领域内聚类表示,同时将不同领域的表示分开。在两个医学分割任务上的大量实验表明,APEX显著提高了在已见和未见领域之间的泛化能力。此外,它补充了任何现有的主干网络,并且一致地提升了性能,证实了其作为医学领域中即插即用提示解决方案的有效性。代码可在https://github.com/cetinkayaevren/apex/上获得。

英文摘要:

Visual prompting has emerged as a powerful method for adapting pre-trained models to new domains without updating model parameters. However, existing prompting methods typically optimize a single prompt per domain and apply it uniformly to all inputs, limiting their ability to generalize under intra and inter-domain variability, which is especially critical in the medical field. To address this, we propose APEX, an Adaptive Prompt EXtraction framework that retrieves input-specific prompts from a learnable prompt memory. The memory stores diverse, domain-discriminative prompt representations and is queried via domain features extracted from the Fourier spectrum. To learn robust and discriminative domain features, we introduce a novel Low-Frequency Feature Contrastive (LFC) learning framework that clusters representations from the same domain while separating those from different domains. Extensive experiments on two medical segmentation tasks demonstrate that APEX significantly improves generalization across both seen and unseen domains. Furthermore, it complements any existing backbones and consistently enhances performance, confirming its effectiveness as a plug-and-play prompting solution in medical fields. The code is available at https://github.com/cetinkayaevren/apex/

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

↑