提升医学图像分割中SAM3的测试时自适应能力
Concept Alignment Contrast and Long-Short Prompt Memory for Test-Time Adaptation of SAM3 in Medical Image Segmentation

- 用概念对齐对比度选择最优增强视图作为监督信号
- 长短记忆提示模块实现快速稳定的一次性适应
- 通过密集伪标签优化提示嵌入,适合无标注医学分割
如分割任意模型3(SAM3)等概念分割模型在自然图像上表现优异,但在医学图像中因成像原理和风格差异导致性能下降。测试时自适应(TTA)可通过无需标注的在线更新提升测试性能。然而,现有视觉-语言TTA方法主要依赖图像级不确定性最小化,无法反映医学分割中的区域语义正确性,且缺乏持续单遍适应的稳定性机制,导致在缺乏可靠密集监督时性能受限。为此,本文提出针对医学图像的SAM3测试时自适应框架CM-TTA。首先,针对同一测试样本的多个增强视图,引入新颖的概念对齐对比(CAC)度量,利用文本-视觉语义一致性来鲁棒评估预测质量,选择最优增强视图为监督信号。其次,设计长短提示记忆(LSPM)模块:短记忆基于CAC分数动态融合近期提示以实现敏捷局部适应,长记忆保持稳定全局提示生成增强伪标签。最后,提出密集监督提示更新(DSPU)策略,以增强伪标签为密集监督优化提示嵌入。在前列腺和皮肤病变分割任务上的大量实验表明,所提CM-TTA框架显著优于现有SAM3 TTA方法。代码已公开于https://github.com/SherlockZYB/CM-TTA。
原文摘要 · Abstract (English)
Concept segmentation models like Segment Anything Model 3 (SAM3) show strong generalization on natural images, yet their performance degrades in medical imaging due to the domain gap caused by different imaging principles and styles. Test-Time Adaptation (TTA) is essential for improving the testing performance by updating the model on the fly without annotations. However, existing vision-language TTA methods are mainly driven by image-level uncertainty minimization, which does not necessarily reflect region-level semantic correctness in medical segmentation. Moreover, they often lack mechanisms to maintain stability in continual one-pass adaptation, leading to limited performance when reliable dense supervision is missing for segmentation. To address these issues, we propose Concept Alignment Contrast and LongShort Prompt Memory for Test-Time Adaptation (CM-TTA) of SAM3 for medical images. First, for a test sample with multiple augmentations, we introduce a novel Concept Alignment Contrast (CAC) metric, which leverages textual-visual semantic consistency to robustly evaluate prediction quality to select the best augmented view as the supervision. Second, to balance rapid and stable adaptation, we design a Long-Short Prompt Memory (LSPM) module. The short memory dynamically fuses recent prompts based on CAC scores for agile local adaptation, while the long memory maintains a stable global prompt to generate enhanced pseudo-labels. Finally, a Densely Supervised Prompt Update (DSPU) strategy is proposed to optimize the prompt embeddings with enhanced pseudo labels as dense supervision. Extensive experiments on prostate and skin lesion segmentation demonstrate that our CM-TTA framework significantly outperforms existing methods for TTA of SAM3. The code is available at https://github.com/SherlockZYB/CM-TTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。