arXiv:2504.02008q-bio.QMcs.AI2025-04ICCV被引 8

不更新参数,通过优化图像嵌入提升医学分割模型性能。

Test-time Adaptation for Foundation Medical Segmentation Model without Parametric Updates

  • 用嵌入优化替代参数更新,避免遗忘与计算开销
  • 在三数据集上平均提升3%的Dice分数
  • 适合需高效适配的医疗影像场景

基础医学分割模型如MedSAM在器官和病灶分割上表现优异,但在结构复杂或外观特殊的病灶上仍表现不佳,且受边界框提示干扰。现有测试时自适应(TTA)方法因部分或全参数更新,受限于信号不足或灾难性遗忘,且计算成本高。本文理论分析表明,在MedSAM架构下,直接优化图像嵌入可达到与参数更新相当的效果。为此,提出基于分布近似潜在条件随机场损失与熵最小化损失的框架,实现无参数更新下的高效适配。实验显示,该方法在三个数据集上平均提升约3% Dice分数,同时计算复杂度降低7倍以上。

原文摘要 · Abstract (English)

Foundation medical segmentation models, with MedSAM being the most popular, have achieved promising performance across organs and lesions. However, MedSAM still suffers from compromised performance on specific lesions with intricate structures and appearance, as well as bounding box prompt-induced perturbations. Although current test-time adaptation (TTA) methods for medical image segmentation may tackle this issue, partial (e.g., batch normalization) or whole parametric updates restrict their effectiveness due to limited update signals or catastrophic forgetting in large models. Meanwhile, these approaches ignore the computational complexity during adaptation, which is particularly significant for modern foundation models. To this end, our theoretical analyses reveal that directly refining image embeddings is feasible to approach the same goal as parametric updates under the MedSAM architecture, which enables us to realize high computational efficiency and segmentation performance without the risk of catastrophic forgetting. Under this framework, we propose to encourage maximizing factorized conditional probabilities of the posterior prediction probability using a proposed distribution-approximated latent conditional random field loss combined with an entropy minimization loss. Experiments show that we achieve about 3\% Dice score improvements across three datasets while reducing computational complexity by over 7 times.

医学分割测试时自适应无参数更新高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。