Abstract:Abstract Many cognitive processes involve transformations of distributed representations in neural populations, creating a need for population-level models. Recurrent neural network models fulfill this need, but there are many open questions about how their connectivity gives rise to dynamics that solve a task. Here, we present a method for finding the connectivity of networks for which the dynamics are specified to solve a task in an interpretable way. We apply our method to a working memory task by synthesizing a network that implements a drift-diffusion process over a ring-shaped manifold. We also use our method to demonstrate how inputs can be used to control network dynamics for cognitive flexibility and explore the relationship between representation geometry and network capacity. Our work fits within the broader context of understanding neural computations as dynamics over relatively low-dimensional manifolds formed by correlated patterns of neurons. Author Summary Neurons in the brain form intricate networks that can produce a vast array of activity patterns. To support goal-directed behavior, the brain must adjust the connections between neurons so that network dynamics can perform desirable computations on behaviorally relevant variables. A fundamental goal in computational neuroscience is to provide an understanding of how network connectivity aligns the dynamics in the brain to the dynamics needed to track those variables. Here, we develop a mathematical framework for creating recurrent neural network models that can address this problem. Specifically, we derive a set of linear equations that constrain the connectivity to afford a direct mapping of task-relevant dynamics onto network activity. We demonstrate the utility of this technique by creating and analyzing a set of network models that can perform a simple working memory task. We then extend the approach to show how additional constraints can furnish networks whose dynamics are controlled flexibly by external inputs. Finally, we exploit the flexibility of this technique to explore the robustness and capacity limitations of recurrent networks. This network synthesis method provides a powerful means for generating and validating hypotheses about how task-relevant computations can emerge from network dynamics.

Geometry of neural computation unifies working memory and planning

A Geometric Framework for Understanding Dynamic Information Integration in Context-Dependent Computation

Task representations in neural networks trained to perform many cognitive tasks

Geometry of naturalistic object representations in recurrent neural network models of working memory

A unified neural representation model for spatial and semantic computations

Reconciling Flexibility and Efficiency: Medial Entorhinal Cortex Represents a Compositional Cognitive Map

Geometry and dynamics of representations in a precisely balanced memory network related to olfactory cortex

Humans actively reconfigure neural task states

2-D Neural Geometry Underpins Hierarchical Organization of Sequence in Human Working Memory

Bifurcation in Space: Emergence of Function Modularity in the Neocortex

Memory, Space, and Planning: Multiscale Predictive Representations

Two-dimensional neural geometry underpins hierarchical organization of sequence in human working memory

Geometric constraints on human brain function

Unity and Diversity of Neural Representation in Executive Functions.

Engineering recurrent neural networks from task-relevant manifolds and dynamics

Neural population geometry and optimal coding of tasks with shared latent structure

Working memory control dynamics follow principles of spatial computing

A quasi-comprehensive exploration of the mechanisms of spatial working memory

The Task-Dependent Modular Covariance Networks Unveiled by Multiple-Way Fusion-Based Analysis

From cognitive to neural models of working memory

How Working Memory and Reinforcement Learning Are Intertwined: A Cognitive, Neural, and Computational Perspective