3 papers
cs.AI2026
Learning Abstractions for Hierarchical Planning in Program-Synthesis Agents
Zergham Ahmed, Kazuki Irie, Joshua B. Tenenbaum +2
Humans learn abstractions and use them to plan efficiently to quickly generalize across tasks -- an ability that remains challenging for state-of-the-art large language model (LLM)…
cs.LG2025
Blending Complementary Memory Systems in Hybrid Quadratic-Linear Transformers
Kazuki Irie, Morris Yau, Samuel J. Gershman
We develop hybrid memory architectures for general-purpose sequence processing neural networks, that combine key-value memory using softmax attention (KV-memory) with fast weight m…
q-bio.NC2025
Key-value memory in the brain
Samuel J. Gershman, Ila Fiete, Kazuki Irie
Classical models of memory in psychology and neuroscience rely on similarity-based retrieval of stored patterns, where similarity is a function of retrieval cues and the stored pat…