arXiv:2603.05873cs.CV2026-03

用记忆代替权重更新,实现医疗图像分割的高效自适应。

Shifting Adaptation from Weight Space to Memory Space: A Memory-Augmented Agent for Medical Image Segmentation

  • 将模型适应从修改权重转为动态管理记忆单元
  • 静态记忆即达基准水平,测试时记忆进一步提升性能
  • 适合需要低通信开销的联邦学习与持续部署场景

医疗图像分割对临床工作至关重要,但单一数据集训练的模型在不同机构、扫描仪或患者群体间泛化能力差。尽管视觉基础模型展现出潜力,其部署通常需任务特异性微调,在联邦学习中带来巨大通信开销,并阻碍部署期间的知识持续演进。本文提出一种记忆增强分割代理(MemSeg-Agent),将适应从权重空间转向记忆空间,实现少样本学习、联邦监督学习和测试时自适应的统一架构。MemSeg-Agent以固定主干网络为基础,通过智能控制器动态组合轻量级静态、少样本及测试时工作记忆。在联邦设置下,仅更新紧凑的记忆单元而非模型参数,显著降低通信开销。在四个公开数据集上的实验表明,该方法具备强性能和鲁棒性:仅使用静态记忆即达到或超过强监督基线,且测试时记忆进一步提升域内与跨域表现,无需微调。总体而言,MemSeg-Agent为智能体时代可扩展、自适应的医疗图像分割提供了新范式。

原文摘要 · Abstract (English)

Medical image segmentation is fundamental to clinical workflows, yet models trained on a single dataset often fail to generalize across institutions, scanners, or patient populations. While vision foundation models have shown great promise in addressing this challenge, their deployment typically requires task-specific fine-tuning, which introduces substantial communication overhead in federated learning and prevents continuous knowledge evolution during deployment. In this work, we propose a memory-augmented segmentation agent (MemSeg-Agent) that shifts adaptation from weight space to memory space, enabling few-shot learning, federated supervised learning, and test-time adaptation within a unified architecture. MemSeg-Agent conditions a fixed backbone with lightweight static, few-shot, and test-time working memories, which are dynamically composed by an agentic controller. In federated settings, we update compact memory units instead of model parameters, substantially reducing communication overhead. Experiments on four public datasets demonstrate strong performance and robustness to domain shift: Static memory alone matches or surpasses strong supervised baselines with high parameter efficiency, and test-time working memory further improves in-domain and cross-domain performance without fine-tuning. Overall, MemSeg-Agent introduces a new paradigm for scalable and adaptive medical image segmentation in the era of agentic AI.

医疗图像记忆机制联邦学习自适应

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