mlsanityData to 14 Oct 2026

LoCA: Forward-Only LLM Tuning after One-Shot Calibration with Local Credit Assignment

arXiv 2608.03020 ↗First posted 4 Aug 2026cs.AIcited by 0

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

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  3. 2024Qwen2.5 Technical Reportcited by 4,942
  4. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  5. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
  6. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  7. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
  8. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
  9. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,418
  10. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
  11. 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
  12. 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
20 more
  1. 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
  2. 2016Training Deep Nets with Sublinear Memory Costcited by 848
  3. 2020Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuningcited by 671
  4. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 463
  5. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
  6. 2023Fine-Tuning Language Models with Just Forward Passescited by 345
  7. 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
  8. 2022The Forward-Forward Algorithm: Some Preliminary Investigationscited by 266
  9. 2016Direct Feedback Alignment Provides Learning in Deep Neural Networkscited by 240
  10. 2013Optimal rates for zero-order convex optimization: the power of two function evaluationscited by 198
  11. 2019Training Neural Networks with Local Error Signalscited by 136
  12. 2023Full Parameter Fine-tuning for Large Language Models with Limited Resourcescited by 134
  13. 2018Greedy Layerwise Learning Can Scale to ImageNetcited by 115
  14. 2022Scaling Forward Gradient With Local Lossescited by 65
  15. 2020Direct Feedback Alignment Scales to Modern Deep Learning Tasks and Architecturescited by 56
  16. 2024Sparse MeZO: Less Parameters for Better Performance in Zeroth-Order LLM Fine-Tuningcited by 54
  17. 2024Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizercited by 53
  18. 2024Enhancing Zeroth-order Fine-tuning for Language Models with Low-rank Structurescited by 50
  19. 2024BAdam: A Memory Efficient Full Parameter Optimization Method for Large Language Modelscited by 27
  20. 2024Zeroth-Order Fine-Tuning of LLMs in Random Subspacescited by 26

Built on by

Not cited yet.