arXiv:2511.11362cs.LGcs.CL2025-11

无需反向传播,用前向评估实现边缘设备大模型微调。

On-Device Fine-Tuning via Backprop-Free Zeroth-Order Optimization

  • 用前向评估估算梯度,避免存储中间激活和优化器状态。
  • 在内存受限条件下,可支持比传统方法大数倍的模型。
  • 适合资源受限的边缘设备,尤其对长时微调有优势。

边缘AI系统需在严格内存约束下适应不同代理任务,而传统基于反向传播的微调需存储层激活值和优化器状态,难以完全通过检查点缓解。在模型权重必须全部驻留设备内存的场景中,此开销严重限制了可部署模型的最大规模。内存高效零阶优化(MeZO)通过仅使用前向评估估计梯度,消除对中间激活和优化器状态的存储需求,从而显著提升片上内存下的模型容量,尽管可能带来更长的微调耗时。本文首次提供BP与MeZO训练下可容纳模型规模的理论估算,并通过数值验证表明,在有足够微调时间的前提下,MeZO在内存受限条件下具有精度优势。

原文摘要 · Abstract (English)

On-device fine-tuning is a critical capability for edge AI systems, which must support adaptation to different agentic tasks under stringent memory constraints. Conventional backpropagation (BP)-based training requires storing layer activations and optimizer states, a demand that can be only partially alleviated through checkpointing. In edge deployments in which the model weights must reside entirely in device memory, this overhead severely limits the maximum model size that can be deployed. Memory-efficient zeroth-order optimization (MeZO) alleviates this bottleneck by estimating gradients using forward evaluations alone, eliminating the need for storing intermediate activations or optimizer states. This enables significantly larger models to fit within on-chip memory, albeit at the cost of potentially longer fine-tuning wall-clock time. This paper first provides a theoretical estimate of the relative model sizes that can be accommodated under BP and MeZO training. We then numerically validate the analysis, demonstrating that MeZO exhibits accuracy advantages under on-device memory constraints, provided sufficient wall-clock time is available for fine-tuning.

边缘计算微调零阶优化

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