arXiv:2410.00362cs.CLcs.AI2024-10被引 4

让边缘设备在不暴露数据的情况下高效微调大模型。

FedPT: Federated Proxy-Tuning of Large Language Models on Resource-Constrained Edge Devices

  • 用小型模型协作学习,再生成大型模型的代理微调版本。
  • 计算、通信和内存开销降低,性能接近直接微调大模型。
  • 适合隐私敏感、资源有限的边缘设备场景。

尽管预训练大语言模型在多种语言任务中表现优异,但针对下游任务进行微调通常需要收集个人数据,引发严重隐私问题。联邦学习(FL)成为主流解决方案,可在不共享原始数据的前提下实现协同训练。然而,大模型的联邦微调面临参数访问受限及高计算、通信与内存开销等挑战。本文提出联邦代理微调(FedPT),一种仅需访问大模型输出词汇预测结果即可完成微调的新框架,无需获取其内部参数。具体而言,各设备先协同训练一个小型语言模型,服务器随后将小模型学习的知识与大模型知识融合,构建出性能接近直接微调大模型的大型代理微调模型。实验表明,FedPT显著降低了计算、通信和内存开销,同时保持与直接联邦微调大模型相当的性能。该方法为资源受限设备上的高效、隐私保护型大模型微调提供了可行方案,拓展了先进大模型的应用边界。

原文摘要 · Abstract (English)

Despite demonstrating superior performance across a variety of linguistic tasks, pre-trained large language models (LMs) often require fine-tuning on specific datasets to effectively address different downstream tasks. However, fine-tuning these LMs for downstream tasks necessitates collecting data from individuals, which raises significant privacy concerns. Federated learning (FL) has emerged as the de facto solution, enabling collaborative model training without sharing raw data. While promising, federated fine-tuning of large LMs faces significant challenges, including restricted access to model parameters and high computation, communication, and memory overhead. To address these challenges, this paper introduces \textbf{Fed}erated \textbf{P}roxy-\textbf{T}uning (FedPT), a novel framework for federated fine-tuning of black-box large LMs, requiring access only to their predictions over the output vocabulary instead of their parameters. Specifically, devices in FedPT first collaboratively tune a smaller LM, and then the server combines the knowledge learned by the tuned small LM with the knowledge learned by the larger pre-trained LM to construct a large proxy-tuned LM that can reach the performance of directly tuned large LMs. The experimental results demonstrate that FedPT can significantly reduce computation, communication, and memory overhead while maintaining competitive performance compared to directly federated fine-tuning of large LMs. FedPT offers a promising solution for efficient, privacy-preserving fine-tuning of large LMs on resource-constrained devices, broadening the accessibility and applicability of state-of-the-art large LMs.

联邦学习大模型微调边缘计算

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