让SAM模型在无训练数据下精准分割医学图像,提升12.4%准确率。
Boundary-Aware Test-Time Adaptation for Zero-Shot Medical Image Segmentation
- 测试时通过高斯提示注入和边界感知注意力对齐增强特征表示。
- 在4个医学数据集上平均提升12.4%的DICE分数,超越现有方法。
- 无需任何训练数据,适合医疗场景快速部署与泛化需求。
由于标注数据稀缺和模型计算成本高,传统医学图像分割调优方法面临严峻挑战。当前预训练模型适配方法仍依赖下游任务的特定训练,而零样本分割因基础模型(如SAM)展现出良好泛化能力受到关注。然而,SAM在医学数据上受领域偏移影响,表现受限,亟需高效零样本增强。为此,本文提出BA-TTA-SAM,一种任务无关的测试时自适应框架,通过测试时调整显著提升SAM的零样本分割性能。该框架融合两项机制:(1) 编码器级高斯提示注入,直接将高斯提示嵌入图像编码器,引导初始表征学习;(2) 跨层边界感知注意力对齐,利用ViT主干中层级特征交互,对齐深层语义响应与浅层边界线索。在ISIC、Kvasir、BUSI和REFUGE四个数据集上的实验表明,相比SAM零样本分割,平均提升12.4%的DICE分数。结果表明,本方法在医学图像分割上持续优于最先进模型,显著增强SAM的泛化能力,且无需源域训练数据。公开数据集上的大量实验充分验证了框架优越性。代码已开源。
原文摘要 · Abstract (English)
Due to the scarcity of annotated data and the substantial computational costs of model, conventional tuning methods in medical image segmentation face critical challenges. Current approaches to adapting pretrained models, including full-parameter and parameter-efficient fine-tuning, still rely heavily on task-specific training on downstream tasks. Therefore, zero-shot segmentation has gained increasing attention, especially with foundation models such as SAM demonstrating promising generalization capabilities. However, SAM still faces notable limitations on medical datasets due to domain shifts, making efficient zero-shot enhancement an urgent research goal. To address these challenges, we propose BA-TTA-SAM, a task-agnostic test-time adaptation framework that significantly enhances the zero-shot segmentation performance of SAM via test-time adaptation. This framework integrates two key mechanisms: (1) The encoder-level Gaussian prompt injection embeds Gaussian-based prompts directly into the image encoder, providing explicit guidance for initial representation learning. (2) The cross-layer boundary-aware attention alignment exploits the hierarchical feature interactions within the ViT backbone, aligning deep semantic responses with shallow boundary cues. Experiments on four datasets, including ISIC, Kvasir, BUSI, and REFUGE, show an average improvement of 12.4\% in the DICE score compared with SAM's zero-shot segmentation performance. The results demonstrate that our method consistently outperforms state-of-the-art models in medical image segmentation. Our framework significantly enhances the generalization ability of SAM, without requiring any source-domain training data. Extensive experiments on publicly available medical datasets strongly demonstrate the superiority of our framework. Our code is available at https://github.com/Emilychenlin/BA-TTA-SAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。