A complementary learning system for continual episodic memory in large language models

Abstract

Humans retain memories of individual experiences for a lifetime, an ability attributed to a complementary learning system in which a fast process encodes episodes and a slow process integrates them into semantic knowledge. In classical Hebbian models such as Hopfield networks, memory traces are superposed in shared weights. This makes learning naturally continual but causes strong interference among correlated memories, a failure that reappears as catastrophic forgetting in deep networks. Here we use a large language model as a model system for continual episodic memory, with its pretrained weights supplying the semantic context in which new episodes are embedded. Fast learning is implemented by a hippocampus-like module that assigns each episode to a dedicated, extremely sparse low-rank adapter; competitive gating then selects among these separated traces during recall. Across streams of up to 1,000 factual and autobiographical episodes, each adapter requires only 2-3 parameters per token while preserving excellent recall. An internal retrieval-augmented generation mechanism reconstructs the selected episode in context and supports high-accuracy question answering over stored memories. Finally, slow cortical consolidation is modeled by fine-tuning the base weights through batch replay, enabling reconstruction and direct question answering without episodic adapters. Together, fast storage and slow consolidation implement both components of a complementary learning system within a single language model, yielding a neural-network model that stores, recalls, and consolidates naturalistic episodic memories, thereby capturing key functional features of human memory.