Open WorldDiscovery GitHub
← Research library
2025NeurIPS

Titans: Learning to Memorize at Test Time

Ali Behrouz, Peilin Zhong, Vahab Mirrokni

What this work does

Updates a neural associative memory with gradient-based surprise, momentum and forgetting, combined with attention and persistent learned memory.

Connection to discovery

A memory-design option for long evidence histories and selective retention.

Evidence

The consulted equations specify memory updates and a parallelizable training form; architectural claims must be distinguished from any particular pretrained release.

Scope and limitations

Surprise is measured by the memory objective rather than causal information gain. The review does not infer an official code or checkpoint release from similarly named community implementations.

THE RESEARCH CONVERSATION

Discussion & reactions

Share a question, an insight, or a replication experience. Sign in with GitHub to join the conversation.

Open GitHub discussions ↗