Skip to content
Kahn Lab Computational Cognitive Neuroscience

Research

How do we plan in complex environments?

To plan effectively, we need to predict not only one step into the future but the possible long-term outcomes associated with our choices. Doing so requires the brain to construct detailed predictive models of the world.

How do we build these mental models, and how do we choose the right model to use at any time? We draw on network science, control theory, and reinforcement learning, and combine behavior in humans and animals with human neuroimaging, in order to understand healthy cognition, its development, and how it can go awry.

Question 01

How do we build models of the world?

Human behavior, from infants to adults, is shaped by temporal regularity in sequences such as speech, music, or motor movements. We are sensitive to the transition probabilities between syllables, notes, or other items, and can use those probabilities to guide our perceptions and expectations of the immmediate future. But to plan further into the future, we either need to chain these single-step predictions (which becomes increasingly costly with planning depth) or somehow generate multi-step predictions that themselves reach beyond the next event

We ask how people extract that higher-level structure, starting from the mechanisms that support simple statistical regularities and asking how they can underpin multi-step temporal abstractions. We study this by studying how people learn to predict sequences of items. Formalizing such sequences as walks on a graph, encoding the possible transitions between items, allows us to hold local statistics fixed while varying the global structure that generates them, and examine how people depend on such global structure. We can then read out peoples’ internal predictions via reaction times and choices, and examine how these predictive models are formed trial-by-trial.

A network diagram of a modular graph, with densely connected clusters of nodes joined by a small number of edges between clusters.
We examine predictions to sequences of items, where the possible transitions between those items can be visualized as a graph.
Brain images showing regions where BOLD activity reflects the predictive structure of a learned sequence rather than the identity of individual stimuli.
Visual cortex representations differ when items show modular structure.

Question 02

How do we choose the right model to guide decision-making?

The brain is capable of both habitual and goal-directed behavior, typically mapped onto model-free planning, which caches the long-run values of actions, and model-based planning, which simulates outcomes through a world model, and these two strategies are though to adaptively trade off with one another. Planning in complex environments, however, requires predictions over temporally extended timescales, where the value of an action may depend on outcomes many minutes away. In such environments, habits are insensitive to changes in distant outcomes, while simulation can be too expensive for longer horizons.

Predictive models such as the successor representation offer a middle path, learning where actions lead separately from what those outcomes are worth, so distant values can change in the face of new information, without costly step-by-step replanning. Substantial evidence demonstrates that people rely on such predictive models in deicison making, and we have found that such models trade off against finer-grained model-based planning. In particular, the brain adaptively chooses finer-grained models only when coarser predictive models are behaviorally insufficient. However, understanding what signals the brain relies for this arbitration remains an ongoing question.

A screenshot of the sailing task, in which participants choose routes between islands laid out as a graph of possible transitions.
We examine how people adaptively choose when to plan step-by-step versus mental models that allow them to skip further into the future.

Also ongoing

Network structure and brain dynamics

The network science underpinning our work on graph learning applies equally to the brain’s own architecture. Diffusion imaging lets us treat white matter connectivity as a network, and ask what dynamics that network can support: how controllable it is, how easily activity can be steered between states, and how those properties change over development.

We are particularly interested in measures of indirect connectivity, such as communicability, since the regions that cooperate during learning are often not directly connected. This work runs alongside the questions above rather than apart from them — the same tools that characterize a learned graph also characterize the structural graph doing the learning.

A diffusion MRI reconstruction of white matter tracts, with streamlines coloured by their orientation through the brain.

Interested in this work?

The lab is recruiting graduate students, postdocs, and Arizona undergraduates.

How to join