Memory-Efficient Structured Backpropagation for On-Device LLM Fine-Tuning
Not on the map. Why?
Abstract (arXiv)
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 = xA can be recomputed during backward at minimal cost since rank r ll din, 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 ≈0.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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 10 of its 14 references that are in the map. First posted on arXiv 13 Feb 2026.
Builds on
10 of 14 references.
References · 10
- 2017Attention Is All You Needcited by 59,932
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2016Training Deep Nets with Sublinear Memory Costcited by 825
- 2023Fine-Tuning Language Models with Just Forward Passescited by 337
- 2022On-Device Training Under 256KB Memorycited by 158
- 2023PockEngine: Sparse and Efficient Fine-tuning in a Pocketcited by 22
Built on by
Not cited yet.