Titans Revisited: A Lightweight Reimplementation and Critical Analysis of a Test-Time Memory Model
Not on the map. Why?
Abstract (arXiv)
By the end of 2024, Google researchers introduced Titans: Learning at Test Time, a neural memory model achieving strong empirical results across multiple tasks. However, the lack of publicly available code and ambiguities in the original description hinder reproducibility. In this work, we present a lightweight reimplementation of Titans and conduct a comprehensive evaluation on Masked Language Modeling, Time Series Forecasting, and Recommendation tasks. Our results reveal that Titans does not always outperform established baselines due to chunking. However, its Neural Memory component consistently improves performance compared to attention-only models. These findings confirm the model's innovative potential while highlighting its practical limitations and raising questions for future research.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 12 references that are in the map. First posted on arXiv 10 Oct 2025.
Builds on
10 of 12 references.
References · 10
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,169
- 2019BERT4Rec: Sequential Recommendation with Bidirectional Encoder Representations from Transformercited by 964
- 2024Titans: Learning to Memorize at Test Timecited by 328
- 2024Human-inspired Episodic Memory for Infinite Context LLMscited by 62
- 2023Learning to (Learn at Test Time)cited by 17
Built on by
Not cited yet.