解决联邦学习中多模态数据缺失与计算资源不均问题
FediLoRA: Practical Federated Fine-Tuning of Foundation Models Under Missing-Modality Constraints
- 采用轻量级聚合框架,结合简单平均与结构化编辑
- 在多领域数据集上保持强性能,医疗数据表现尤佳
- 适合真实场景部署,尤其适用于医疗等敏感领域
联邦学习结合LoRA微调为机构协作利用大规模数据训练视觉语言大模型提供了高效且保护隐私的方案。然而,参与机构常面临异构计算资源,导致LoRA秩不平衡,严重阻碍有效协作。此外,医疗、交通等领域因用户误操作或设备故障频繁出现数据模态缺失,显著降低联邦设置下的全局模型性能。目前尚无研究同时应对这两类挑战。为此,我们提出FediLoRA,一种轻量级联邦LoRA聚合框架,能有效缓解异构环境下的模态缺失影响。该方法基于观察:简单平均与结构化编辑可协同提升全局与个性化模型表现。FediLoRA在多个通用领域与医疗领域基准数据集上均表现优异。对医疗数据的额外实验进一步证明其适用于实际部署场景。代码已开源:https://github.com/gotobcn8/FediLoRA。
原文摘要 · Abstract (English)
Federated Learning with LoRA fine-tuning offers an efficient and privacy-aware solution for institutions to collaboratively leverage their large datasets to train VLLMs. However, participating institutions often possess heterogeneous computational resources, resulting in imbalanced LoRA ranks, which pose a major challenge for effective collaboration. In addition, real-world applications in domains such as healthcare and transportation frequently suffer from missing modalities due to user mistakes or device failures, which significantly degrade global model performance in federated settings. To the best of our knowledge, no prior work has addressed these two challenges simultaneously in federated VLLMs. To tackle these issues, we propose FediLoRA, a lightweight federated LoRA aggregation framework that effectively mitigates the impact of missing modalities in heterogeneous environment. FediLoRA is explicitly motivated by the observation that simple averaging and structured editing can jointly benefit both global and personalized models. Our approach achieves strong performance across multiple general-domain and medical-domain benchmark datasets. Additional experiments on healthcare data further demonstrate that FediLoRA is well-suited for practical, real-world deployment scenarios. Our code is released at https://github.com/gotobcn8/FediLoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。