Abstract:The mammalian hippocampus contains a cognitive map that represents an animal's position in the environment and generates offline "replay" for the purposes of recall, planning, and forming long term memories. Recently, it's been found that artificial neural networks trained to predict sensory inputs develop spatially tuned cells, aligning with predictive theories of hippocampal function. However, whether predictive learning can also account for the ability to produce offline replay is unknown. Here, we find that spatially tuned cells, which robustly emerge from all forms of predictive learning, do not guarantee the presence of a cognitive map with the ability to generate replay. Offline simulations only emerged in networks that used recurrent connections and head-direction information to predict multi-step observation sequences, which promoted the formation of a continuous attractor reflecting the geometry of the environment. These offline trajectories were able to show wake-like statistics, autonomously replay recently experienced locations, and could be directed by a virtual head direction signal. Further, we found that networks trained to make cyclical predictions of future observation sequences were able to rapidly learn a cognitive map and produced sweeping representations of future positions reminiscent of hippocampal theta sweeps. These results demonstrate how hippocampal-like representation and replay can emerge in neural networks engaged in predictive learning, and suggest that hippocampal theta sequences reflect a circuit that implements a data-efficient algorithm for sequential predictive learning. Together, this framework provides a unifying theory for hippocampal functions and hippocampal-inspired approaches to artificial intelligence.

Learning offline: memory replay in biological and artificial reinforcement learning

Augmented Replay Memory in Reinforcement Learning With Continuous Control

Replay in Deep Learning: Current Approaches and Missing Biological Elements

Brain-Like Replay Naturally Emerges in Reinforcement Learning Agents

A unifying account of replay as context-driven memory reactivation

A model of hippocampal replay driven by experience and environmental structure facilitates spatial learning

Model-Based and Model-Free Replay Mechanisms for Reinforcement Learning in Neurorobotics

Replay-enhanced Continual Reinforcement Learning

Dual Memory Model for Experience-Once Task-Incremental Lifelong Learning.

Distinct replay signatures for prospective decision-making and memory preservation

Research on Online Reinforcement Learning Method Based on Experience-Replay

Replay in human visual cortex is linked to the formation of successor representations and independent of consciousness

Generative Memory for Lifelong Reinforcement Learning

A Robotic Model of Hippocampal Reverse Replay for Reinforcement Learning

Memory Efficient Experience Replay for Streaming Learning

Adaptive Memory Replay for Continual Learning

Paradoxical replay can protect contextual task representations from destructive interference when experience is unbalanced

Saliency-Guided Hidden Associative Replay for Continual Learning

Sequential predictive learning is a unifying theory for hippocampal representation and replay

Memory Enhanced Replay for Continual Learning

Map-based Experience Replay: A Memory-Efficient Solution to Catastrophic Forgetting in Reinforcement Learning