*Source: zhichai.net forum post — auto-collected 2026-05-14*
Paper Overview
Field: Machine Learning
Authors: Rishabh Tiwari, Kusha Sareen, Lakshya A Agrawal, Joseph E. Gonzalez, Matei Zaharia, Kurt Keutzer, Inderjit S Dhillon, Rishabh Agarwal, Devvrit Khatri
Published: 2026-05-12
arXiv: 2605.12484
Abstract
Large language models (LLMs) are trained for downstream tasks by updating their parameters (e.g., via RL). However, updating parameters forces them to absorb task-specific information, which can result in catastrophic forgetting and loss of plasticity. In contrast, in-context learning with fixed LLM parameters can cheaply and rapidly adapt to task-specific requirements (e.g., prompt optimization), but cannot by itself typically match the performance gains available through updating LLM parameters. There is no good reason for restricting learning to being in-context or in-weights. Moreover, humans also likely learn at different time scales (e.g., System 1 vs 2).
Key Findings
- Fast-slow learning framework: Model parameters act as "slow" weights, while an optimized context serves as "fast" weights. The fast weights are learned from textual feedback to absorb task-specific information, allowing the slow weights to remain closer to the base model and maintain general reasoning behavior.
- Sample efficiency: Fast-slow training (FST) is up to 3x more sample-efficient than slow-only learning (RL) on reasoning tasks, while consistently reaching higher performance asymptotes.
- Less forgetting: FST-trained models show up to 70% lower KL divergence from the base LLM compared to RL training, resulting in substantially less catastrophic forgetting.
- Preserved plasticity: The reduced drift retains plasticity — after training on one task, FST models adapt to subsequent tasks more effectively than parameter-only trained models.
- Continual learning: In continual learning scenarios, FST continues to acquire each new task, while parameter-only RL stagnates.
*Auto-collected on 2026-05-14*