让边缘设备用更少资源训练大模型,精度不降反升。
Lean Clients, Full Accuracy: Hybrid Zeroth- and First-Order Split Federated Learning
- 客户端用零阶优化替代反向传播,免去梯度计算和缓存
- 在ResNet和语言模型上实现与基准相当的精度,内存减少64%
- 理论证明收敛速度与模型规模无关,适合低算力设备部署
分割联邦学习(SFL)使资源受限的边缘设备与计算能力强的服务器协同训练成为可能。通信开销是SFL的核心问题,可通过辅助网络缓解。然而,客户端计算成本仍是根本挑战,因反向传播需大量内存和计算资源,严重限制了边缘设备支持的模型规模。为此,我们提出HERON-SFL,一种融合零阶(ZO)优化用于本地客户端训练、保留一阶(FO)优化于服务器端的新型混合优化框架。借助辅助网络,ZO更新使客户端通过每步扰动前向评估近似局部梯度,无需激活缓存且避免传统训练中的显式梯度计算。基于低有效秩假设,我们理论证明了HERON-SFL的收敛率与模型维度无关,解决了零阶算法常见的可扩展性问题。实验表明,在ResNet训练和语言模型微调任务中,HERON-SFL达到基准精度,客户端峰值内存降低高达64%,客户端每步计算成本减少33%,显著拓展了资源受限设备上可训练或适配的模型范围。
原文摘要 · Abstract (English)
Split Federated Learning (SFL) enables collaborative training between resource-constrained edge devices and a compute-rich server. Communication overhead is a central issue in SFL and can be mitigated with auxiliary networks. Yet, the fundamental client-side computation challenge remains, as back-propagation requires substantial memory and computation costs, severely limiting the scale of models that edge devices can support. To enable more resource-efficient client computation and reduce the client-server communication, we propose HERON-SFL, a novel hybrid optimization framework that integrates zeroth-order (ZO) optimization for local client training while retaining first-order (FO) optimization on the server. With the assistance of auxiliary networks, ZO updates enable clients to approximate local gradients using perturbed forward-only evaluations per step, eliminating memory-intensive activation caching and avoiding explicit gradient computation in the traditional training process. Leveraging the low effective rank assumption, we theoretically prove that HERON-SFL's convergence rate is independent of model dimensionality, addressing a key scalability concern common to ZO algorithms. Empirically, on ResNet training and language model (LM) fine-tuning tasks, HERON-SFL matches benchmark accuracy while reducing client peak memory by up to 64% and client-side compute cost by up to 33% per step, substantially expanding the range of models that can be trained or adapted on resource-limited devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。