arXiv:2503.01164cs.CV2025-03

用积木式组合方法,零训练实现通用医学影像诊断模型的快速更新。

Med-LEGO: Editing and Adapting toward Generalist Medical Image Diagnosis

  • 通过SVD改进LoRA,用极少量参数捕捉专科模型专长。
  • 仅用0.18%参数量就超越现有方法,跨域与本域任务均表现优异。
  • 无需原始数据或重训练,适合医疗场景下隐私敏感的模型迭代。

视觉基础模型在计算机辅助诊断(CAD)中已广泛应用。尽管这些基础模型为通用医学AI提供了可行方案,但隐私问题使得跨领域、跨数据集的预训练或持续更新难以实施,导致多数研究聚焦于专科模型。为此,我们提出Med-LEGO,一种无需训练的框架,可像拼积木一样将多个专科模型无缝整合或更新通用CAD模型。Med-LEGO通过引入奇异值分解(SVD)改进LoRA,以极少额外参数高效捕捉各专科模型的领域专长。通过简单运算组合这些适配权重,可在不依赖原始数据或重训练的前提下,轻松集成或修改特定诊断能力。最终合并模型还可进一步适应新诊断任务,形成灵活通用的医学AI系统。大量实验表明,Med-LEGO在跨域与本域任务上均优于现有方法,仅使用0.18%全模型参数量,且具备更优收敛性与泛化能力,为通用医学AI提供有效路径。

原文摘要 · Abstract (English)

The adoption of visual foundation models has become a common practice in computer-aided diagnosis (CAD). While these foundation models provide a viable solution for creating generalist medical AI, privacy concerns make it difficult to pre-train or continuously update such models across multiple domains and datasets, leading many studies to focus on specialist models. To address this challenge, we propose Med-LEGO, a training-free framework that enables the seamless integration or updating of a generalist CAD model by combining multiple specialist models, similar to assembling LEGO bricks. Med-LEGO enhances LoRA (low-rank adaptation) by incorporating singular value decomposition (SVD) to efficiently capture the domain expertise of each specialist model with minimal additional parameters. By combining these adapted weights through simple operations, Med-LEGO allows for the easy integration or modification of specific diagnostic capabilities without the need for original data or retraining. Finally, the combined model can be further adapted to new diagnostic tasks, making it a versatile generalist model. Our extensive experiments demonstrate that Med-LEGO outperforms existing methods in both cross-domain and in-domain medical tasks while using only 0.18% of full model parameters. These merged models show better convergence and generalization to new tasks, providing an effective path toward generalist medical AI.

医学AI模型融合LoRA

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