不更新模型参数,用可靠图文预测构建记忆库提升医学图像分割鲁棒性
Memory-Supported Synergistic Adaptation for Training-Free Test-Time Medical Image Segmentation

- 通过动态筛选可靠图文预测构建在线记忆库
- 在多个基准上提升最高达12.2% DSC和11.7% mIoU
- 适合无训练数据、需快速适应的医疗影像场景
测试时自适应(TTA)通过在推理时利用未标注目标数据来缓解分布偏移。尽管基于视觉-语言模型(VLM)的TTA在分类任务中表现良好,但将其扩展到医学图像分割仍具挑战。在此场景下,优化VLM生成预测所带来的增益常被噪声驱动的更新对VLM强预训练特征的破坏所抵消,导致改进有限且不稳定。为此,我们提出无需训练的医学图像分割测试时自适应框架MSSA。MSSA不更新模型参数,而是动态选择可靠的图像-文本预测构建在线记忆库,将其作为文本引导的语义先验,并与跨图像结构对齐相结合以实现稳健适应。具体包括:(i) 噪声感知的记忆构建模块,用于过滤并稳定跨模态预测;(ii) 相关性驱动的原型对齐模块,将目标样本与其结构一致的记忆样本及其可靠预测对齐以提升适应效果。在多个医学分割基准上的大量实验表明,MSSA持续提升基于VLM的分割模型性能,显著优于现有微调类TTA方法,最大提升达12.2% DSC和11.7% mIoU。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) aims to mitigate distribution shifts by adapting models with unlabeled target data at inference time. While TTA with vision-language models (VLMs) has shown promising results in classification, extending it to medical image segmentation remains challenging. In this setting, the adaptation gains from optimizing on VLM-generated predictions are often outweighed by the degradation to the VLM's strong pretrained features caused by noisy, update-driven learning, resulting in limited and unstable improvements. We therefore propose Memory-Supported Synergistic Adaptation (MSSA), a novel training-free TTA framework for medical image segmentation. Without updating model parameters, MSSA dynamically selects reliable image-text predictions to construct an online memory, uses them as text-guided semantic priors, and couples them with cross-image structural alignment for robust adaptation. Specifically, MSSA consists of (i) a noise-aware memory construction module that filters and stabilizes cross-modal predictions, and (ii) a relevance-driven prototype alignment module that aligns the target sample with structurally consistent memory samples and their reliable predictions to improve adaptation. Extensive experiments on multiple medical segmentation benchmarks demonstrate that MSSA consistently improves VLM-based segmentation models and outperforms existing fine-tuning-based TTA methods by a clear margin, with gains of up to 12.2% DSC and 11.7% mIoU. Project page: https://lingrayy.github.io/MSSA/ .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。