让手机高效协作微调大模型,降低算力与通信负担。
Fed MobiLLM: Efficient Federated LLM Fine-Tuning over Heterogeneous Mobile Devices via Server Assisted Side-Tuning
- 服务器辅助侧网络训练,客户端仅传中间激活值。
- 计算开销减少95.2%,通信成本降93.2%,收敛快5.1倍。
- 适合异构手机设备上个性化大模型快速适配。
在异构移动设备上协同微调大语言模型(LLM)可推动个性化智能的广泛应用,但面临严峻系统挑战:传统联邦微调方法对移动硬件造成过重的计算与内存负担,且同步聚合协议易被慢设备拖累。本文提出 Fed MobiLLM,一种新型设计,实现跨异构移动设备的高效联邦 LLM 微调。其核心是首创的服务器辅助联邦侧调优范式:客户端使用冻结的预缩放骨干 LLM 对本地数据进行轻量前向计算,仅上传选定的中间激活值;服务器独立训练共享侧网络,消除客户端反向传播,支持异步更新。为应对设备间模型异构性,引入自适应分层特征对齐方法,确保共享侧网络协同训练时表示一致。大量实验表明,Fed MobiLLM 在保持鲁棒微调性能的同时,实现极低设备端内存占用,相较现有方法计算开销降低至少95.2%,通信成本减少93.2%,收敛速度提升5.1倍,验证了其在异构移动设备上实用化大模型适配的有效性。
原文摘要 · Abstract (English)
Collaboratively fine-tuning (FT) large language models (LLMs) over heterogeneous mobile devices fosters immense potential applications of personalized intelligence. However, such a vision faces critical system challenges. Conventional federated LLM FT approaches place prohibitive computational and memory burdens on mobile hardware, and their synchronous model aggregation protocols stall for slower devices. In this paper, we propose Fed MobiLLM, a novel design to facilitate efficient federated LLM FT across mobile devices with diverse computing/communication speeds and local model architectures. In particular, Fed MobiLLM implements a pioneering server-assisted federated side-tuning paradigm. Briefly, mobile devices perform lightweight forward propagation computations on local data using their frozen pre-scaled backbone LLMs, and then upload selected intermediate activations. The server trains a shared side-network independently, eliminating client-side backpropagation and enabling asynchronous updates. To bridge model heterogeneity across different devices, we introduce an adaptive layer-wise feature alignment method, which ensures consistent representations for collaboratively tuning a shared side network. Extensive experimental results demonstrate that Fed MobiLLM can maintain robust fine-tuning performance while achieving extremely low on-device memory, with at least 95.2% reduction in computation overhead, 93.2% reduction in communication costs and 5.1x faster convergence compared to existing methods, validating its efficacy for practical LLM adaptation over heterogeneous mobile devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。