大模型微调只需一次通信,性能不降反升
Revisiting Federated Fine-Tuning: A Single Communication Round is Enough for Foundation Models
- 采用单轮聚合的联邦微调方法
- 单轮即可达到多轮效果,通信量大幅降低
- 适合大规模分布式隐私保护场景
基础模型(FMs)的快速发展催生了在跨领域大规模数据集上进行微调的需求。为应对这一挑战,联邦微调应运而生,可在保护数据隐私的前提下,在多个设备的分布式数据上对模型进行微调。然而,由于参数量庞大及传统联邦学习算法需多轮通信,导致通信成本过高,限制了其实际应用。本文通过理论与实证分析发现,传统多轮聚合算法对大模型联邦微调并非必需。实验表明,仅需一轮聚合(即单次联邦微调)即可获得与多轮聚合相当的全局模型性能。数学与实证分析进一步证明,因参数量大且预训练任务广泛,大模型在单轮联邦微调中训练损失显著低于小模型。大量实验显示,单轮联邦微调可显著降低通信开销,支持异步聚合,提升隐私性,并在文本生成与文生图任务中保持性能一致性。研究成果为实践中的联邦微调带来革新,提升效率、降低成本、扩大基础模型的应用边界。
原文摘要 · Abstract (English)
The recent advancement of foundation models (FMs) has increased the demand for fine-tuning these models on large-scale cross-domain datasets. To address this, federated fine-tuning has emerged, allowing FMs to be fine-tuned on distributed datasets across multiple devices while ensuring data privacy. However, the substantial parameter size and the multi-round communication in federated learning algorithms result in prohibitively high communication costs, challenging the practicality of federated fine-tuning. In this paper, we identify and analyze, both theoretically and empirically, that the traditional multi-round aggregation algorithms may not be necessary for federated fine-tuning large FMs. Our experiments reveal that a single round of aggregation (i.e., one-shot federated fine-tuning) yields a global model performance comparable to that achieved through multiple rounds of aggregation. Through rigorous mathematical and empirical analyses, we demonstrate that large FMs, due to their extensive parameter sizes and pre-training on general tasks, achieve significantly lower training loss in one-shot federated fine-tuning compared to smaller models. Our extensive experiments show that one-shot federated fine-tuning significantly reduces communication costs. It also has the potential to enable asynchronous aggregation, enhances privacy, and maintains performance consistency with multi-round federated fine-tuning on both text generation and text-to-image generation tasks. Our findings provide insights to revolutionize federated fine-tuning in practice, enhancing efficiency, reducing costs, and expanding accessibility for FMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。