arXiv:2503.10217cs.LGcs.AI2025-03被引 1

通过随机丢弃层加速大模型联邦微调,兼顾效率与隐私。

Efficient Federated Fine-Tuning of Large Language Models with Layer Dropout

  • 用随机丢弃模型层减少计算和内存占用。
  • 比现有方法快1.3到6.3倍,内存降低40%至67%。
  • 适合资源受限设备上的高效个性化模型训练。

微调在使预训练大语言模型从通用语言理解进化为特定任务专长中起关键作用。为保护用户数据隐私,联邦微调已成为主流范式。然而,由于大模型复杂性与终端设备资源限制之间的矛盾,联邦微调效率极低,带来不可接受的微调开销。现有研究主要采用参数高效微调技术以降低通信成本,但计算和内存负担仍对开发者构成重大挑战。本文提出 DropPEFT,一种创新的联邦 PEFT 框架,采用新颖的随机变换器层丢弃方法,使设备在训练时可关闭大量模型层,从而消除相应计算负载与内存占用。在 DropPEFT 中,层丢弃比例的合理配置是关键挑战,因开销与训练性能对此设置高度敏感。为此,我们通过探索-利用策略自适应地为设备分配最优丢弃比例配置,实现高效且有效的微调。大量实验表明,相比最先进方法,DropPEFT 可实现 1.3–6.3× 的模型收敛速度提升,内存占用降低 40%–67%。

原文摘要 · Abstract (English)

Fine-tuning plays a crucial role in enabling pre-trained LLMs to evolve from general language comprehension to task-specific expertise. To preserve user data privacy, federated fine-tuning is often employed and has emerged as the de facto paradigm. However, federated fine-tuning is prohibitively inefficient due to the tension between LLM complexity and the resource constraint of end devices, incurring unaffordable fine-tuning overhead. Existing literature primarily utilizes parameter-efficient fine-tuning techniques to mitigate communication costs, yet computational and memory burdens continue to pose significant challenges for developers. This work proposes DropPEFT, an innovative federated PEFT framework that employs a novel stochastic transformer layer dropout method, enabling devices to deactivate a considerable fraction of LLMs layers during training, thereby eliminating the associated computational load and memory footprint. In DropPEFT, a key challenge is the proper configuration of dropout ratios for layers, as overhead and training performance are highly sensitive to this setting. To address this challenge, we adaptively assign optimal dropout-ratio configurations to devices through an exploration-exploitation strategy, achieving efficient and effective fine-tuning. Extensive experiments show that DropPEFT can achieve a 1.3-6.3\times speedup in model convergence and a 40%-67% reduction in memory footprint compared to state-of-the-art methods.

联邦学习大模型微调参数效率内存优化

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