arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.10967cs.CV

Med-DualLoRA: 3D心脏MRI基础模型的局部适应

Med-DualLoRA: Local Adaptation of Foundation Models for 3D Cardiac MRI

Artificial Intelligence in Medicine Lab (BCN-AIM), Departament de Matemàtiques i Informàtica, Universitat de Barcelona, Spain(巴塞罗那大学人工智能医学实验室(BCN-AIM)、数学与计算机科学系,西班牙);Institució Catalana de Recerca i Estudis Avançats (ICREA), Barcelona, Spain(加泰罗尼亚高级研究机构(ICREA),巴塞罗那,西班牙)

Joan Perramon-Llussà, Amelia Jiménez-Sánchez, Grzegorz Skorupko, Fotis Avgoustidis, Carlos Martín-Isla, Karim Lekadir, Polyxeni Gkontra

AI总结:

本文提出Med-DualLoRA框架,通过解耦全局共享和局部低秩适应,实现3D心脏MRI疾病检测的联邦学习,提升个性化与通信效率。

摘要:

基础模型(FMs)在医疗影像任务和模态中表现出强大的鲁棒性,尤其在心脏磁共振(CMR)中,通过任务特定的适应。然而,使用单个站点数据进行适应可能导致性能不佳和模型偏见,而集中化微调临床数据在隐私约束下往往不可行。联邦微调提供了一种隐私保护的替代方案;然而,传统方法在异构、非独立同分布(non-IID)多中心数据下表现不佳,并且在适应大模型时通信开销巨大。在本文中,我们研究了联邦FM微调用于3D CMR疾病检测,并提出Med-DualLoRA,一种客户端感知的参数高效微调(PEFT)联邦框架,通过加法分解解耦全局共享和局部低秩适应(LoRA)。全局和局部LoRA模块在本地训练,但只有全局组件在各站点之间共享和聚合,保持本地适配器的隐私。这种设计提高了个性化能力,同时显著降低了通信成本,实验表明仅适应两个transformer块可以保持性能并进一步提高效率。我们评估了我们的方法在多中心最先进的 cine 3D CMR FM 上,使用ACDC和结合M\&Ms数据集进行疾病检测的联邦学习,将每个供应商视为联邦客户端。Med-DualLoRA在平衡准确率(0.768)和特异性(0.612)上优于其他联邦PEFT基线,同时保持通信效率。我们的方法在现实临床约束下为医学FM的本地联邦适应提供了可扩展的解决方案。

英文摘要:

Foundation models (FMs) show great promise for robust downstream performance across medical imaging tasks and modalities, including cardiac magnetic resonance (CMR), following task-specific adaptation. However, adaptation using single-site data may lead to suboptimal performance and increased model bias, while centralized fine-tuning on clinical data is often infeasible due to privacy constraints. Federated fine-tuning offers a privacy-preserving alternative; yet conventional approaches struggle under heterogeneous, non-IID multi-center data and incur substantial communication overhead when adapting large models. In this work, we study federated FM fine-tuning for 3D CMR disease detection and propose Med-DualLoRA, a client-aware parameter-efficient fine-tuning (PEFT) federated framework that disentangles globally shared and local low-rank adaptations (LoRA) through additive decomposition. Global and local LoRA modules are trained locally, but only the global component is shared and aggregated across sites, keeping local adapters private. This design improves personalization while significantly reducing communication cost, and experiments show that adapting only two transformer blocks preserves performance while further improving efficiency. We evaluate our method on a multi-center state-of-the-art cine 3D CMR FM fine-tuned for disease detection using ACDC and combined M\&Ms datasets, treating each vendor as a federated client. Med-DualLoRA achieves statistically significant improved performance (balanced accuracy 0.768, specificity 0.612) compared to other federated PEFT baselines, while maintaining communication efficiency. Our approach provides a scalable solution for local federated adaptation of medical FMs under realistic clinical constraints.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑