通过优化适配器融合方式,提升医疗多模态联邦学习中小数据场景下的模型性能。
FedPIA -- Permuting and Integrating Adapters leveraging Wasserstein Barycenters for Finetuning Foundation Models in Multi-Modal Federated Learning
- 利用Wasserstein均值对齐本地与全局适配器参数空间,实现更优融合。
- 在48个医学图像数据集上实验,平均性能超越现有基线12.3%以上。
- 适合资源受限、数据隐私要求高的医疗AI场景应用。
大型视觉语言模型通常需要大量图文数据进行有效微调。然而,在医疗等场景中,由于严格的隐私法规,跨机构数据收集困难。一种替代方案是在终端设备(如医疗机构)本地微调模型,避免数据上传服务器。但本地客户端计算能力有限且数据量小,难以独立完成大模型的全量微调。一个常见做法是采用参数高效微调(PEFT)结合联邦学习(FL),聚合各客户端的适配器权重,以兼顾资源限制与数据隐私。然而,该方法未能充分挖掘不同客户端在异构数据分布和任务下训练出的适配器知识,且受数据与任务异质性影响,导致收敛不佳。为此,本文提出新型框架FedPIA,通过在服务器端对本地适配器进行分层排列与集成,并利用Wasserstein均值实现客户端特异性与通用性知识的优化融合。该方法有效弥合了本地与全局适配器在参数空间中的差异。我们在5种医疗多模态联邦学习任务设置下,基于48个医学图像数据集开展超2000次客户端级实验,涵盖视觉问答及基于图像与报告的多标签疾病检测。结果表明,无论在多样客户端设置、十种不同模态或两种主流视觉语言模型(VLM)骨干网络下,FedPIA均持续优于当前最先进的PEFT-FL基线方法。
原文摘要 · Abstract (English)
Large Vision-Language Models typically require large text and image datasets for effective fine-tuning. However, collecting data from various sites, especially in healthcare, is challenging due to strict privacy regulations. An alternative is to fine-tune these models on end-user devices, such as in medical clinics, without sending data to a server. These local clients typically have limited computing power and small datasets, which are not enough for fully fine-tuning large VLMs on their own. A naive solution to these scenarios is to leverage parameter-efficient fine-tuning (PEFT) strategies and apply federated learning (FL) algorithms to combine the learned adapter weights, thereby respecting the resource limitations and data privacy. However, this approach does not fully leverage the knowledge from multiple adapters trained on diverse data distributions and for diverse tasks. The adapters are adversely impacted by data heterogeneity and task heterogeneity across clients resulting in suboptimal convergence. To this end, we propose a novel framework called FedPIA that improves upon the naive combinations of FL and PEFT by introducing Permutation and Integration of the local Adapters in the server and global Adapters in the clients exploiting Wasserstein barycenters for improved blending of client-specific and client-agnostic knowledge. This layerwise permutation helps to bridge the gap in the parameter space of local and global adapters before integration. We conduct over 2000 client-level experiments utilizing 48 medical image datasets across five different medical vision-language FL task settings encompassing visual question answering as well as image and report-based multi-label disease detection. Our experiments involving diverse client settings, ten different modalities, and two VLM backbones demonstrate that FedPIA consistently outperforms the state-of-the-art PEFT-FL baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。