arXiv:2411.15837cs.CV2024-11被引 2

通过模态对齐提升异构联邦学习的性能与收敛性。

Modality Alignment Meets Federated Broadcasting

  • 服务器用文本编码器,客户端用图像编码器,实现跨模态通信对齐。
  • 在极端数据异构下仍保持模型泛化能力,测试准确率超基线10%以上。
  • 采用低秩适配和查询聚合,适合资源受限设备部署。

联邦学习(FL)通过在分布式边缘设备上训练模型而不集中本地数据,成为保护数据隐私的强大方法。尽管在同质数据场景中取得进展,但在异构数据下维持全局与本地客户端间的性能仍具挑战,因数据分布差异导致模型收敛下降且计算成本上升。本文提出一种新型联邦学习框架,融合模态对齐机制:服务器部署文本编码器,本地设备运行图像编码器。受CLIP等多模态学习范式启发,将服务器-客户端通信类比为多模态广播。通过预训练模型初始化,结合低秩适配(LoRA)更新部分参数,兼顾计算效率与性能。本地模型独立训练,参数更新经查询式聚合后上传至服务器,促进跨客户端知识共享,显著提升高度异构环境下的表现。在多个基准数据集上的实验验证了该方法在保持泛化性和鲁棒性方面的有效性。

原文摘要 · Abstract (English)

Federated learning (FL) has emerged as a powerful approach to safeguard data privacy by training models across distributed edge devices without centralizing local data. Despite advancements in homogeneous data scenarios, maintaining performance between the global and local clients in FL over heterogeneous data remains challenging due to data distribution variations that degrade model convergence and increase computational costs. This paper introduces a novel FL framework leveraging modality alignment, where a text encoder resides on the server, and image encoders operate on local devices. Inspired by multi-modal learning paradigms like CLIP, this design aligns cross-client learning by treating server-client communications akin to multi-modal broadcasting. We initialize with a pre-trained model to mitigate overfitting, updating select parameters through low-rank adaptation (LoRA) to meet computational demand and performance efficiency. Local models train independently and communicate updates to the server, which aggregates parameters via a query-based method, facilitating cross-client knowledge sharing and performance improvement under extreme heterogeneity. Extensive experiments on benchmark datasets demonstrate the efficacy in maintaining generalization and robustness, even in highly heterogeneous settings.

联邦学习模态对齐异构数据低秩适配

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