Stachenfeld, Kim

13 publications

ICLRW 2025 Brain-like Slot Representation for Sequence Working Memory in Recurrent Neural Networks Mingye Wang, Stefano Fusi, Kim Stachenfeld

ICML 2025 Discovering Symbolic Cognitive Models from Human and Animal Behavior Pablo Samuel Castro, Nenad Tomasev, Ankit Anand, Navodita Sharma, Rishika Mohanta, Aparna Dev, Kuba Perlin, Siddhant Jain, Kyle Levin, Noemi Elteto, Will Dabney, Alexander Novikov, Glenn C Turner, Maria K Eckstein, Nathaniel D. Daw, Kevin J Miller, Kim Stachenfeld

ICLR 2025 ReCogLab: A Framework Testing Relational Reasoning & Cognitive Hypotheses on LLMs Andrew Liu, Henry Prior, Gargi Balasubramaniam, Rivka Moroshko, Amir Zait, Ilia Labzovsky, Danny Karmon, Ishita Dasgupta, Kim Stachenfeld, Kenneth Marino

ICLR 2025 SIMPL: Scalable and Hassle-Free Optimisation of Neural Representations from Behaviour Tom George, Pierre Glaser, Kim Stachenfeld, Caswell Barry, Claudia Clopath

ICLR 2025 When Does Compositional Structure Yield Compositional Generalization? a Kernel Theory. Samuel Lippl, Kim Stachenfeld

ICLR 2024 Learning 3D Particle-Based Simulators from RGB-D Videos William F Whitney, Tatiana Lopez-Guevara, Tobias Pfaff, Yulia Rubanova, Thomas Kipf, Kim Stachenfeld, Kelsey R Allen

ICLR 2024 Predictive Auxiliary Objectives in Deep RL Mimic Learning in the Brain Ching Fang, Kim Stachenfeld

ICLRW 2024 The Impact of Task Structure, Representational Geometry, and Learning Mechanism on Compositional Generalization Samuel Lippl, Kim Stachenfeld

ICLRW 2023 Beyond Temporal Credit Assignment in Reinforcement Learning Sephora Madjiheurem, Kim Stachenfeld, Peter Battaglia, Jessica B Hamrick

ICMLW 2023 Diffusion Generative Inverse Design Marin Vlastelica, Tatiana Lopez-Guevara, Kelsey R Allen, Peter Battaglia, Arnaud Doucet, Kim Stachenfeld

NeurIPSW 2023 Memorization and Consolidation in Associative Memory Networks Danil Tyulmankov, Kim Stachenfeld, Dmitry Krotov, Larry Abbott

CoRL 2022 Graph Network Simulators Can Learn Discontinuous, Rigid Contact Dynamics Kelsey R Allen, Tatiana Lopez Guevara, Yulia Rubanova, Kim Stachenfeld, Alvaro Sanchez-Gonzalez, Peter Battaglia, Tobias Pfaff

ICLR 2022 Learned Simulators for Turbulence Kim Stachenfeld, Drummond Buschman Fielding, Dmitrii Kochkov, Miles Cranmer, Tobias Pfaff, Jonathan Godwin, Can Cui, Shirley Ho, Peter Battaglia, Alvaro Sanchez-Gonzalez