mlsanityData to 14 Oct 2026

TFGN: Task-Free, Replay-Free Continual Pre-Training Without Catastrophic Forgetting at LLM Scale

arXiv 2605.15053 ↗First posted 14 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Continually pre-training a large language model on heterogeneous text domains, without replay or task labels, has remained an unsolved architectural problem at LLM scale. Existing methods rely on replay buffers, task identifiers, regularization penalties that scale poorly, or sentence-classification-scale evaluation. We introduce TFGN, an architectural overlay for transformer language models that produces input-conditioned, parameter-efficient updates while leaving the rest of the transformer unchanged. On six heterogeneous text domains (Prose, Python, Math, Biomedical, Chinese, JavaScript) at 1B tokens per phase across three model scales (~398M, ~739M, ~9B) and two regimes (From-Scratch and Retrofit), TFGN achieves backward transfer of -0.007 at LLaMA 3.1 8B Retrofit, HellaSwag retention 0.506/0.504/0.510, and >=99.59% L2-orthogonal gradient separation between domain pairs - with no replay, no task IDs, no Fisher penalty. The same matrices show positive cross-domain forward transfer: held-out JavaScript PPL drops 26.8% at LLaMA-8B Retrofit and 62.0% at GPT-2 Medium From-Scratch purely from Python training. Two extensions on the same substrate close further open problems. A closed-loop meta-control layer (Extension A) reduces forgetting by an additional 81% at ~398M, mapping onto the System A and System M roles of Dupoux et al. (arXiv:2603.15381). An operator-level plan vector (Extension B) reshapes forward-pass behavior at 99.96% cosine fidelity over 30 source->target pairs. The architectural insight is a Read/Write decomposition: the forward pass is fully dense, while cross-domain parameter updates are structured so prior-domain subspaces are not written to. To our knowledge, TFGN is the first architecture that simultaneously closes catastrophic forgetting at LLM scale, realizes a closed-loop autonomous-learning meta-controller, and carries an operator-level latent planner.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 71 references that are in the map. First posted on arXiv 14 May 2026.

Builds on

38 of 71 references.

References · 38

  1. 2023GPT-4 Technical Reportcited by 18,396
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  3. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  5. 2016Progressive Neural Networkscited by 1,361
  6. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
  7. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
  8. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  9. 2023Mastering Diverse Domains through World Modelscited by 1,253
  10. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  11. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,085
  12. 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
26 more
  1. 2024Refusal in Language Models Is Mediated by a Single Directioncited by 1,032
  2. 2023Steering Language Models With Activation Engineeringcited by 922
  3. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
  4. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
  5. 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
  6. 2023StarCoder: may the source be with you!cited by 799
  7. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
  8. 2023Orthogonal Subspace Learning for Language Model Continual Learningcited by 312
  9. 2023Function Vectors in Large Language Modelscited by 308
  10. 2019On Tiny Episodic Memories in Continual Learningcited by 307
  11. 2023CulturaX: A Cleaned, Enormous, and Multilingual Dataset for Large Language Models in 167 Languagescited by 174
  12. 2023OpenWebMath: An Open Dataset of High-Quality Mathematical Web Textcited by 155
  13. 2023Lifelong Language Pretraining with Distribution-Specialized Expertscited by 66
  14. 2023TRACE: A Comprehensive Benchmark for Continual Learning in Large Language Modelscited by 66
  15. 2025Do Latent Tokens Think? A Causal and Adversarial Analysis of Chain-of-Continuous-Thoughtcited by 23
  16. 2024Neuron-Level Sequential Editing for Large Language Modelscited by 20
  17. 2024Examining Forgetting in Continual Pre-training of Aligned Large Language Modelscited by 18
  18. 2024Towards Effective and Efficient Continual Pre-training of Large Language Modelscited by 15
  19. 2025Revisiting Replay and Gradient Alignment for Continual Pre-Training of Large Language Modelscited by 14
  20. 2026Mechanistic Analysis of Catastrophic Forgetting in Large Language Models During Continual Fine-tuningcited by 14
  21. 2025TreeLoRA: Efficient Continual Learning via Layer-Wise LoRAs Guided by a Hierarchical Gradient-Similarity Treecited by 12
  22. 2026Why AI systems don't learn and what to do about it: Lessons on autonomous learning from cognitive sciencecited by 11
  23. 2022Short-Term Plasticity Neurons Learning to Learn and Forgetcited by 6
  24. 2025Full-Parameter Continual Pretraining of Gemma2: Insights into Fluency and Domain Knowledgecited by 4
  25. 2025STABLE: Gated Continual Learning for Large Language Modelscited by 3
  26. 2026HiEdit: Lifelong Model Editing with Hierarchical Reinforcement Learningcited by 1

Built on by

Not cited yet.