arXiv:2409.19540cs.CV2024-09被引 4

将医学大模型分解为多个轻量专家模型,提升特定任务表现。

LoRKD: Low-Rank Knowledge Decomposition for Medical Foundation Models

  • 通过低秩专家模块和高效知识分离卷积,解耦不同部位的梯度冲突。
  • 在分割与分类任务上超越原模型,且迁移能力更强。
  • 适合医疗影像细分领域应用,降低资源消耗同时增强专业化。

大规模预训练技术显著推动了医学基础模型的发展,使其能广泛应用于各类医疗任务。然而,尽管具备强泛化能力,这些模型在异构数据间存在领域差异,导致在特定任务上的表现不如专用模型。本文提出“知识分解”新思路,将基础模型拆分为多个针对不同解剖区域的轻量级专家模型,以增强专精性并降低资源开销。为此,我们设计了低秩知识分解(LoRKD)框架,通过引入低秩专家模块和高效知识分离卷积,显式分离不同任务的梯度。低秩模块缓解了不同解剖区域间异构数据带来的梯度冲突,实现低成本强专精;高效知识分离卷积在单次前向传播中完成知识分离,大幅提升算法效率。大量实验表明,分解后的模型在分割与分类任务上达到领先性能,并在下游任务中展现出更优迁移能力,甚至在特定任务评估中超越原始基础模型。代码已开源。

原文摘要 · Abstract (English)

The widespread adoption of large-scale pre-training techniques has significantly advanced the development of medical foundation models, enabling them to serve as versatile tools across a broad range of medical tasks. However, despite their strong generalization capabilities, medical foundation models pre-trained on large-scale datasets tend to suffer from domain gaps between heterogeneous data, leading to suboptimal performance on specific tasks compared to specialist models, as evidenced by previous studies. In this paper, we explore a new perspective called "Knowledge Decomposition" to improve the performance on specific medical tasks, which deconstructs the foundation model into multiple lightweight expert models, each dedicated to a particular anatomical region, with the aim of enhancing specialization and simultaneously reducing resource consumption. To accomplish the above objective, we propose a novel framework named Low-Rank Knowledge Decomposition (LoRKD), which explicitly separates gradients from different tasks by incorporating low-rank expert modules and efficient knowledge separation convolution. The low-rank expert modules resolve gradient conflicts between heterogeneous data from different anatomical regions, providing strong specialization at lower costs. The efficient knowledge separation convolution significantly improves algorithm efficiency by achieving knowledge separation within a single forward propagation. Extensive experimental results on segmentation and classification tasks demonstrate that our decomposed models not only achieve state-of-the-art performance but also exhibit superior transferability on downstream tasks, even surpassing the original foundation models in task-specific evaluations. The code is available at here.

医学影像知识分解低秩模型专家网络

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