← Search

Kim Stachenfeld

11 accepted papers

2025

Discovering Symbolic Cognitive Models from Human and Animal Behavior

ICML 2025spotlight

Symbolic models play a key role in cognitive science, expressing computationally precise hypotheses about how the brain implements a cognitive process. Identifying an appropriate model typically requires a great deal of effort and ingenuity on the part of a human scientist. Here, we adapt FunSearch…

Cited by 1SourcePDFScholar
2025

ReCogLab: a framework testing relational reasoning & cognitive hypotheses on LLMs

ICLR 2025poster

A fundamental part of human cognition is the ability to not only recall previous memories, but also reason across them to draw conclusions. In cognitive science and psychology, this is termed relational reasoning and a number of effects and biases have been observed in human cognition. Designing exp…

2025

SIMPL: Scalable and hassle-free optimisation of neural representations from behaviour

ICLR 2025poster

Neural activity in the brain is known to encode low-dimensional, time-evolving, behaviour-related variables. A long-standing goal of neural data analysis has been to identify these variables and their mapping to neural activity. A productive and canonical approach has been to simply visualise neural…

Cited by 0SourcePDFScholar
2025

When does compositional structure yield compositional generalization? A kernel theory.

ICLR 2025poster

Compositional generalization (the ability to respond correctly to novel combinations of familiar components) is thought to be a cornerstone of intelligent behavior. Compositionally structured (e.g. disentangled) representations support this ability; however, the conditions under which they are suffi…

2024

Learning 3D Particle-based Simulators from RGB-D Videos

ICLR 2024poster

Realistic simulation is critical for applications ranging from robotics to animation. Traditional analytic simulators sometimes struggle to capture sufficiently realistic simulation which can lead to problems including the well known "sim-to-real" gap in robotics. Learned simulators have emerged as…

Cited by 10SourcePDFScholar
2024

Learning rigid-body simulators over implicit shapes for large-scale scenes and vision

NeurIPS 2024oral

Simulating large scenes with many rigid objects is crucial for a variety of applications, such as robotics, engineering, film and video games. Rigid interactions are notoriously hard to model: small changes to the initial state or the simulation parameters can lead to large changes in the final stat…

Cited by 2SourcePDFScholar
2023

A generative model of the hippocampal formation trained with theta driven local learning rules

NeurIPS 2023poster

Advances in generative models have recently revolutionised machine learning. Meanwhile, in neuroscience, generative models have long been thought fundamental to animal intelligence. Understanding the biological mechanisms that support these processes promises to shed light on the relationship betwee…

Cited by 14SourcePDFScholar
2022

Graph network simulators can learn discontinuous, rigid contact dynamics

CoRL 2022poster

Recent years have seen a rise in techniques for modeling discontinuous dynamics, such as rigid contact or switching motion modes, using deep learning. A common claim is that deep networks are incapable of accurately modeling rigid-body dynamics without explicit modules for handling contacts, due to…

Cited by 45SourceScholar
2022

Inverse Design for Fluid-Structure Interactions using Graph Network Simulators

NeurIPS 2022accept

Designing physical artifacts that serve a purpose---such as tools and other functional structures---is central to engineering as well as everyday human behavior. Though automating design using machine learning has tremendous promise, existing methods are often limited by the task-dependent distribut…

Cited by 20SourcePDFScholar
2022

Learned Simulators for Turbulence

ICLR 2022poster

Turbulence simulation with classical numerical solvers requires high-resolution grids to accurately resolve dynamics. Here we train learned simulators at low spatial and temporal resolutions to capture turbulent dynamics generated at high resolution. We show that our proposed model can simulate tur…

Cited by 47SourcePDFScholar