mlsanityData to 7 Oct 2026

Retrievable Gradients: Continual Post-Training Without Cumulative Weight Drift

arXiv 2606.15734 ↗First posted 14 Jun 2026cs.CL cs.AI cs.IRcited by 0

Not on the map. Why?

Abstract (arXiv)

Continual post-training enables models to absorb emerging knowledge after deployment, but repeatedly updating shared parameters can accumulate weight drift, potentially causing catastrophic forgetting and degrading general capabilities. Retrieval-augmented generation avoids such parameter drift, yet often lacks the depth of parametric knowledge integration. In this paper, we propose ReGrad (Retrievable Gradients), a new paradigm that treats gradients as retrievable units of knowledge. ReGrad pre-computes document-specific gradients offline, stores them in an indexed Gradient Bank, and retrieves only query-relevant gradients at inference time for temporary weight adaptation. However, raw language-modeling gradients are optimized for token-level document reconstruction rather than for query-driven knowledge use. We therefore introduce a bi-level meta-learning objective that reshapes document-derived gradients into generalizable adaptation signals for downstream tasks. Experiments across general and domain-specific settings show that textscReGrad outperforms CPT and RAG baselines, enabling scalable and reversible parametric knowledge injection without accumulating weight drift.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 45 of its 68 references that are in the map. First posted on arXiv 14 Jun 2026.

Builds on

45 of 68 references.

References · 45

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
  3. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
  4. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  5. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
  6. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  7. 2016Learning without Forgettingcited by 2,465
  8. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  9. 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,126
  10. 2019Language Models as Knowledge Bases?cited by 1,922
  11. 2017Gradient Episodic Memory for Continual Learningcited by 1,871
  12. 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
33 more
  1. 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarizationcited by 1,411
  2. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  3. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
  4. 2021Improving language models by retrieving from trillions of tokenscited by 1,278
  5. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  6. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
  7. 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,103
  8. 2022Mass-Editing Memory in a Transformercited by 976
  9. 2023In-Context Retrieval-Augmented Language Modelscited by 694
  10. 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
  11. 2021Fast Model Editing at Scalecited by 491
  12. 2022Memory-Based Model Editing at Scalecited by 443
  13. 2018The Web as a Knowledge-base for Answering Complex Questionscited by 409
  14. 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 264
  15. 2021Mind the Gap: Assessing Temporal Generalization in Neural Language Modelscited by 226
  16. 2021When Does Pretraining Help? Assessing Self-Supervised Learning for Law and the CaseHOLD Datasetcited by 199
  17. 2023Mitigating the Alignment Tax of RLHFcited by 181
  18. 2021Towards Continual Knowledge Learning of Language Modelscited by 172
  19. 2023S-LoRA: Serving Thousands of Concurrent LoRA Adapterscited by 146
  20. 2022Pile of Law: Learning Responsible Data Filtering from the Law and a 256GB Open-Source Legal Datasetcited by 113
  21. 2024Fine-tuning large language models for domain adaptation: Exploration of training strategies, scaling, model merging and synergistic capabilitiescited by 67
  22. 2025Overcoming catastrophic forgetting in neural networkscited by 52
  23. 2026Skill Retrieval Augmentation for Agentic AIcited by 41
  24. 2025RbFT: Robust Fine-tuning for Retrieval-Augmented Generation against Retrieval Defectscited by 27
  25. 2024Knowledge Editing through Chain-of-Thoughtcited by 26
  26. 2025Dynamic Parametric Retrieval Augmented Generation for Test-time Knowledge Enhancementcited by 24
  27. 2023RECKONING: Reasoning through Dynamic Knowledge Encodingcited by 17
  28. 2025SurGE: A Benchmark and Evaluation Framework for Scientific Survey Generationcited by 14
  29. 2025Decoupling Reasoning and Knowledge Injection for In-Context Knowledge Editingcited by 12
  30. 2024STARD: A Chinese Statute Retrieval Dataset with Real Queries Issued by Non-professionalscited by 10
  31. 2024Balancing Continuous Pre-Training and Instruction Fine-Tuning: Optimizing Instruction-Following in LLMscited by 10
  32. 2023Caseformer: Pre-training for Legal Case Retrieval Based on Inter-Case Distinctionscited by 9
  33. 2026Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimizationcited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Revisiting Replay and Gradient Alignment for Continual Pre-Training of Large Language Models
  2. 2025Direct Retrieval-augmented Optimization: Synergizing Knowledge Selection and Language Models
  3. 2026Inject, Align, Recover: Staged Post-Training for Retrieval-Free Document Knowledge Internalization
  4. 2026TASR: Training-Free Adaptive Stopping for Iterative Retrieval
  5. 2025From RAG to Memory: Non-Parametric Continual Learning for Large Language Models
  6. 2025KARE-RAG: Knowledge-Aware Refinement and Enhancement for RAG
  7. 2025Resisting Contextual Interference in RAG via Parametric-Knowledge Reinforcement
  8. 2026ORBIT: Preserving Foundational Language Capabilities in GenRetrieval via Origin-Regulated Merging