Retrievable Gradients: Continual Post-Training Without Cumulative Weight Drift

arXiv:2606.15734 · cs.CL, cs.AI, cs.IR, cs.LG · Submitted 2026-06-14 · Read on arXiv

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

Related papers