English

Memory-Efficient Structured Backpropagation for On-Device LLM Fine-Tuning

Machine Learning 2026-02-16 v1 Computation and Language

Abstract

On-device fine-tuning enables privacy-preserving personalization of large language models, but mobile devices impose severe memory constraints, typically 6--12GB shared across all workloads. Existing approaches force a trade-off between exact gradients with high memory (MeBP) and low memory with noisy estimates (MeZO). We propose Memory-efficient Structured Backpropagation (MeSP), which bridges this gap by manually deriving backward passes that exploit LoRA's low-rank structure. Our key insight is that the intermediate projection h=xAh = xA can be recomputed during backward at minimal cost since rank rdinr \ll d_{in}, eliminating the need to store it. MeSP achieves 49\% average memory reduction compared to MeBP on Qwen2.5 models (0.5B--3B) while computing mathematically identical gradients. Our analysis also reveals that MeZO's gradient estimates show near-zero correlation with true gradients (cosine similarity \approx0.001), explaining its slow convergence. MeSP reduces peak memory from 361MB to 136MB for Qwen2.5-0.5B, enabling fine-tuning scenarios previously infeasible on memory-constrained devices.

Keywords

Cite

@article{arxiv.2602.13069,
  title  = {Memory-Efficient Structured Backpropagation for On-Device LLM Fine-Tuning},
  author = {Juneyoung Park and Yuri Hong and Seongwan Kim and Jaeho Lee},
  journal= {arXiv preprint arXiv:2602.13069},
  year   = {2026}
}

Comments

Under the review, 11 pages

R2 v1 2026-07-01T10:35:32.337Z