arXiv:2409.06067cs.AIcs.CL2024-09中稿 · WACV 2025被引 3

用大模型辅助联邦学习,解决数据异构难题。

MLLM-LLaVA-FL: Multimodal Large Language Model Assisted Federated Learning

  • 服务器端用多模态大模型处理跨模态信息,提升全局一致性
  • 在多个基准测试中,显著改善异构与长尾分布下的性能表现
  • 不增加本地计算负担,保护隐私,适合资源受限场景

以往联邦学习因客户端数据异构常导致性能下降。针对多模态大模型(如GPT-4v、LLaVA)在图像描述与多模态问答中的出色表现,本文提出一种新型联邦学习框架——MLLM-LLaVA-FL。该框架在服务器端引入强大多模态大模型,以应对数据异构与长尾分布挑战。凭借其跨模态表征能力与丰富的开放词汇先验知识,框架充分利用网络上广泛存在的开源数据及服务器侧算力,有效挖掘未被利用的外部信息。整个流程包含三个阶段:首先进行全局视觉-文本预训练,利用在线开源数据并借助多模态大模型;其次将预训练模型分发至各客户端进行本地训练;最后在模型回传后,由多模态大模型监督下进行全局对齐,进一步提升性能。在多个标准基准上的实验表明,该框架在典型异构与长尾分布场景下表现优异,同时避免了隐私泄露风险和本地设备的额外计算负担。

原文摘要 · Abstract (English)

Previous studies on federated learning (FL) often encounter performance degradation due to data heterogeneity among different clients. In light of the recent advances in multimodal large language models (MLLMs), such as GPT-4v and LLaVA, which demonstrate their exceptional proficiency in multimodal tasks, such as image captioning and multimodal question answering. We introduce a novel federated learning framework, named Multimodal Large Language Model Assisted Federated Learning (MLLM-LLaVA-FL), which employs powerful MLLMs at the server end to address the heterogeneous and long-tailed challenges. Owing to the advanced cross-modality representation capabilities and the extensive open-vocabulary prior knowledge of MLLMs, our framework is adept at harnessing the extensive, yet previously underexploited, open-source data accessible from websites and powerful server-side computational resources. Hence, the MLLM-LLaVA-FL not only enhances the performance but also avoids increasing the risk of privacy leakage and the computational burden on local devices, distinguishing it from prior methodologies. Our framework has three key stages. Initially, we conduct global visual-text pretraining of the model. This pretraining is facilitated by utilizing the extensive open-source data available online, with the assistance of MLLMs. Subsequently, the pretrained model is distributed among various clients for local training. Finally, once the locally trained models are transmitted back to the server, a global alignment is carried out under the supervision of MLLMs to further enhance the performance. Experimental evaluations on established benchmarks, show that our framework delivers promising performance in the typical scenarios with data heterogeneity and long-tail distribution across different clients in FL.

联邦学习多模态大模型数据异构隐私保护

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