mlsanityData to 7 Oct 2026

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

arXiv 2602.13069 ↗First posted 13 Feb 2026cs.LG cs.CLcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  4. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
  5. 2016Pointer Sentinel Mixture Modelscited by 2,992
  6. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  7. 2016Training Deep Nets with Sublinear Memory Costcited by 825
  8. 2023Fine-Tuning Language Models with Just Forward Passescited by 337
  9. 2022On-Device Training Under 256KB Memorycited by 158
  10. 2023PockEngine: Sparse and Efficient Fine-tuning in a Pocketcited by 22

Built on by

Not cited yet.