Spaced Repetition Improves Continual LLM Pre-Training

Alankar Atreya, Devesh Batra, Yoages Kumar Mantri, Geremy Bantug, Greig A Cowan, Raad Khraishi· August 19, 2026 View original

Key takeaways

  • SRT uses spaced repetition to optimize continual LLM pre-training.
  • It addresses catastrophic forgetting by adaptively scheduling example reviews.
  • SRT significantly improves the stability-plasticity trade-off in LLMs.
  • The method works across model scales and preserves broad benchmark performance.

Who benefits

AI ResearchSoftware DevelopmentEdTechContent CreationData Science

Summary

Spaced Repetition Training (SRT) is a new continual learning framework inspired by cognitive science that uses the SuperMemo-2 algorithm to schedule sample rehearsal, significantly improving the stability-plasticity trade-off in language model pre-training.

Continual pre-training of large language models (LLMs) faces a fundamental challenge: acquiring new information without forgetting previously learned knowledge. Existing replay methods often use uniform sampling, which overlooks the varying rates at which different examples are forgotten. This paper introduces Spaced Repetition Training (SRT), a novel continual learning framework that draws inspiration from cognitive science principles. SRT re-frames continual pre-training as an adaptive review scheduling problem, deciding not only how much history to replay but also which specific examples to revisit at each step. By maintaining per-example review states and mapping perplexity to a recall-quality signal, SRT schedules historical examples for retention and new ones for consolidation, all while keeping the core model, objective, and optimizer unchanged. Experiments on Wikipedia and code corpora show SRT recovering substantial amounts of old-knowledge accuracy lost by naive methods, while preserving or enhancing new-knowledge acquisition, even at larger model scales.

Why it matters

This research offers a more efficient and effective way to continually update large language models, ensuring they remain current with new information without suffering from catastrophic forgetting, which is crucial for long-term model utility.

How to implement this in your domain

  1. 1Integrate the SuperMemo-2 algorithm or similar spaced repetition logic into existing continual learning pipelines for LLMs.
  2. 2Develop mechanisms to track per-example review states and perplexity as a recall signal.
  3. 3Experiment with SRT on internal datasets to evaluate its impact on model stability and plasticity.
  4. 4Consider applying spaced repetition principles to other machine learning tasks beyond language models.

Original post by Alankar Atreya, Devesh Batra, Yoages Kumar Mantri, Geremy Bantug, Greig A Cowan, Raad Khraishi

"arXiv:2608.17530v1 Announce Type: new Abstract: Continual pre-training of large language models must acquire new information without erasing old knowledge. Existing replay methods often choose a global old/new mixture and sample uniformly, ignoring that examples differ in how qui…"

View on X

Originally posted by Alankar Atreya, Devesh Batra, Yoages Kumar Mantri, Geremy Bantug, Greig A Cowan, Raad Khraishi on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Engineering & DevTools