FwdLLM: Efficient Federated Finetuning of Large Language Models with Perturbed Inferences.

Mengwei Xu,Dongqi Cai,Yaozong Wu,Xiang Li,Shangguang Wang
2024-01-01
Abstract:Large Language Models (LLMs) are transforming the landscape of mobile intelligence. Federated Learning (FL), a method to preserve user data privacy, is often employed in fine-tuning LLMs to downstream mobile tasks, i.e., FedLLM. A vital challenge of FedLLM is the tension between LLM complexity and resource constraint of mobile devices. In response to this challenge, this work introduces FwdLLM, an innovative FL protocol designed to enhance the FedLLM efficiency. The key idea of FwdLLM is to employ backpropagation (BP)-free training methods, requiring devices only to execute "perturbed inferences". Consequently, FwdLLM delivers way better memory efficiency and time efficiency (expedited by mobile NPUs and an expanded array of participant devices). FwdLLM centers around three key designs: (1) it combines BP -free training with parameter-efficient training methods, an essential way to scale the approach to the LLM era; (2) it systematically and adaptively allocates computational loads across devices, striking a careful balance between convergence speed and accuracy; (3) it discriminatively samples perturbed predictions that are more valuable to model convergence. Comprehensive experiments illustrate FwdLLM'S significant advantages over conventional methods, including up to three orders of magnitude faster convergence and a 14.6 x reduction in memory footprint. Uniquely, FwdLLM paves the way for federated billion-parameter LLMs such as LLaMA on COTS mobile devices a feat previously unattained.
What problem does this paper attempt to address?