Sakana AI Introduces Fast-Weight Product Key Memory Method
SakanaAILabs · x · 2026-07-04
Sakana AI's paper "Fast-weight Product Key Memory" (FwPKM) is slated for an oral presentation at the ICML 2026 "Towards Sustainable AI: Continuous Adaptation" workshop. The method combines Test-Time Training (TTT) with Product Key Memory (PKM) to create sparse fast weights, enabling high-capacity memory construction during inference without updating all weights. It offers large storage capacity, sparse access, and supports test-time writing. Under a 128K context length, it achieves about 75% accuracy on the 5-needle NIAH benchmark, and rapidly adapts to new domains after reading 4 million tokens from a target domain, with ΔNLL dropping by up to 0.4.
More from Research
- Kimi K3 may be strong on cyber, but token efficiency keeps it off UK AISIS — teortaxesTex · 2026-07-27
- ARC AGI 3 should have stayed private, with no examples or public dataset — flowersslop · 2026-07-27
- ExploitGym may have only 60–70% solvable tasks, fueling the OpenAI cheating debate — max_paperclips · 2026-07-27
- RTX 5090 local tests show Qwen Q6 can drop to 15 tok/s at 80k context — LFAdvice7984 · 2026-07-27
- Noahpinion quotes Chollet: intelligence may hit a hard ceiling — binarybits · 2026-07-27
- Paper argues graph topology can become the core operating system for AI agents — theomitsa · 2026-07-27