LoCA: Forward-Only LLM Tuning after One-Shot Calibration with Local Credit Assignment
Not on the map. Why?
Abstract (arXiv)
Parameter-efficient post-training reduces the number of trainable parameters, but still requires repeated end-to-end backpropagation through the frozen backbone. Every adaptation step therefore needs backward-capable hardware and must store or recompute activations. We ask whether this repeated backward chain can be replaced by a one-time calibration. We introduce Local Credit Assignment (LoCA), a two-stage method for small-shift adaptation. One probe backward pass fits a low-rank map at each transformer block from the final prediction error to a local hidden-state correction. LoCA then reuses these maps to form blockwise regression targets from forward activations and fits low-rank adapters with closed-form ridge solves. No further backbone backward pass is required. We evaluate LoCA on five discriminative benchmarks with Qwen2.5 models from 0.5B to 14B. In 16 of 25 reported task--scale comparisons, LoCA yields lower evaluation cross-entropy than the corresponding LoRA run. Its measured full-run GPU peak, including calibration, is 26--29% lower than LoRA's. After calibration, its CPU steady-state memory is 36--52% lower and its per-pass time is 43--48% lower. A shared scale-normalized candidate set is reused across all tested Qwen2.5 sizes and on SmolLM2-1.7B. LoCA thus amortizes global credit assignment into one calibration and enables later forward-only tuning when repeated backpropagation is impractical. The code associated with this paper is available hrefhttps://github.com/Xia12121/LoCAhere.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 32 of its 39 references that are in the map. First posted on arXiv 4 Aug 2026.
Builds on
32 of 39 references.
References · 32
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
20 more
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
- 2016Training Deep Nets with Sublinear Memory Costcited by 848
- 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
- 2023Fine-Tuning Language Models with Just Forward Passescited by 345
- 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
- 2022The Forward-Forward Algorithm: Some Preliminary Investigationscited by 266
- 2016Direct Feedback Alignment Provides Learning in Deep Neural Networkscited by 240
- 2013Optimal rates for zero-order convex optimization: the power of two function evaluationscited by 198
- 2019Training Neural Networks with Local Error Signalscited by 136
- 2023Full Parameter Fine-tuning for Large Language Models with Limited Resourcescited by 134
- 2018Greedy Layerwise Learning Can Scale to ImageNetcited by 115
- 2022Scaling Forward Gradient With Local Lossescited by 65
- 2020Direct Feedback Alignment Scales to Modern Deep Learning Tasks and Architecturescited by 56
- 2024Sparse MeZO: Less Parameters for Better Performance in Zeroth-Order LLM Fine-Tuningcited by 54
- 2024Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizercited by 53
- 2024Enhancing Zeroth-order Fine-tuning for Language Models with Low-rank Structurescited by 50
- 2024BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Modelscited by 27
- 2024Zeroth-Order Fine-Tuning of LLMs in Random Subspacescited by 26
Built on by
Not cited yet.