Retrievable Gradients: Continual Post-Training Without Cumulative Weight Drift
cs.CL, cs.AI, cs.IR, cs.LG
Submitted: 2026-06-14
Updated: 2026-10-11
Code: https://github.com/oneal2000/ReGrad
License: http://creativecommons.org/licenses/by-sa/4.0/
The gist: Continual post-training enables models to absorb emerging knowledge after deployment, but repeatedly updating shared parameters can accumulate weight drift, potentially causing catastrophic
Terminology
Abstract
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 ReGrad outperforms CPT and RAG baselines, enabling scalable and reversible parametric knowledge injection without accumulating weight drift.
Sources
- From Local to Global: A Graph RAG Approach to Query-Focused Summarization
- Scaling Laws For Dense Retrieval
- The Pile: An 800GB Dataset of Diverse Text for Language Modeling
- Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Steps
- Towards Continual Knowledge Learning of Language Models
- Active Retrieval Augmented Generation
- Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learning
- Balancing Continuous Pre-Training and Instruction Fine-Tuning: Optimizing Instruction-Following in LLMs
- Dense Passage Retrieval for Open-Domain Question Answering
- Mass-Editing Memory in a Transformer
- Fast Model Editing at Scale
- In-Context Retrieval-Augmented Language Models
- S-LoRA: Serving Thousands of Concurrent LoRA Adapters
- Caseformer: Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions
- Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimization
- STARD: A Chinese Statute Retrieval Dataset with Real Queries Issued by Non-professionals
- Skill Retrieval Augmentation for Agentic AI
- Towards Unification of Hallucination Detection and Fact Verification for Large Language Models
- DRAGIN: Dynamic Retrieval Augmented Generation based on the Information Needs of Large Language Models
- Unsupervised Real-Time Hallucination Detection based on the Internal States of Large Language Models
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering