arXiv:2605.11857cs.LG2026-05

用输出共识替代参数聚合,大幅降低大模型联邦微调通信开销

Beyond Parameter Aggregation: Semantic Consensus for Federated Fine-Tuning of LLMs

论文配图:Beyond Parameter Aggregation: Semantic Consensus for Federated Fine-Tuning of LLMs
图 1 · 摘自论文原文
  • 客户端在私有数据上微调模型,通过共享公共提示的输出进行协作
  • 通信量仅与提示集规模相关,比传统方法减少1006倍(如Llama3.1-405B)
  • 适用于异构架构和开放生成任务,适合资源受限场景下的模型协同

大型语言模型的联邦微调通常被建模为参数聚合问题。然而,即使高效的参数方法也需传输大量可训练权重,假设模型结构一致,并依赖对模型参数的白盒访问。随着模型规模增长和部署环境日益异构,这些假设越来越脱离实际。本文提出一种新范式:以模型行为而非参数作为协作媒介。客户端在私有数据上微调本地模型,并交换在共享公共提示集上的生成输出。服务器将这些输出映射到语义表示空间,形成每个提示的语义共识,并返回伪标签用于进一步本地微调。该方法彻底改变了联邦大模型微调的通信规模:通信量仅取决于公共提示预算和输出大小,与模型规模无关。因此,协议天然支持异构架构,且直接适用于开放式文本生成。我们提供了理论分析和实证结果,表明该方法可媲美强基线,同时在通信量、运行时间和能耗上实现数量级降低(例如,针对Llama3.1-405B理论上减少1006倍)。结果表明,对于生成型基础模型,行为层面的共识比参数聚合更适合作为联邦适配的抽象。

原文摘要 · Abstract (English)

Federated fine-tuning of large language models is commonly formulated as a parameter aggregation problem. However, even parameter-efficient methods require transmitting large collections of trainable weights, assume aligned architectures, and rely on white-box access to model parameters. As model sizes continue to grow and deployments become increasingly heterogeneous, these assumptions become progressively misaligned with practical constraints. We consider an alternative formulation in which collaboration is mediated through model behavior rather than parameters. Clients fine-tune local models on private data and exchange generated outputs on a shared, public prompt set. The server maps these outputs into a semantic representation space, forms a per-prompt semantic consensus, and returns pseudo-labels for further local fine-tuning. This formulation fundamentally changes the communication scaling of federated LLM fine-tuning. The amount of information exchanged depends only on the public prompt budget and the size of the communicated behaviors, independent of model size. As a consequence, the protocol naturally accommodates heterogeneous architectures and applies directly to open-ended text generation. We present a theoretical analysis and empirical results demonstrating that this approach can match strong federated fine-tuning baselines while substantially reducing communication by orders of magnitude (e.g., analytically by a factor of $1006$ for Llama3.1-405B), as well as reductions in runtime and energy consumption. These results suggest that, for generative foundation models, behavior-level consensus provides a more appropriate abstraction for federated adaptation than parameter aggregation.

联邦学习大模型微调低通信语义共识

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