Building Compositional Tasks with Shared Neural Subspaces

How the brain flexibly performs multiple tasks by compositionally combining task-relevant neural representations in shared subspaces of neural activity.

Sina Tafazoli, Flora M. Bouchacourt, Adel Ardalan, Nikola T. Markov, Motoaki Uchimura, Marcelo G. Mattar, Nathaniel D. Daw & Timothy J. Buschman

Nature · Volume 650 · 2026

Key Findings

Core discoveries from neural recordings in five brain regions during three compositional tasks

Shared Sensory Subspaces

The same subspaces of LPFC neural activity represent colour category information across the C1 and C2 tasks. Classifiers trained on one task generalize to the other, indicating a shared representational geometry.

Shared Motor Subspaces

Motor response representations in the S1 and C1 tasks occupy the same neural subspace. This shared motor code was found across all recorded brain regions (LPFC, FEF, PAR, aIT, STR).

Sequential Transformation

Tasks engage shared subspaces sequentially: sensory information in the shared colour subspace is transformed into the appropriate motor subspace in a task-specific manner.

Task Belief Modulation

Monkeys maintain an internal belief about the current task. This belief dynamically scales the engagement of shared sensory subspaces — amplifying relevant and suppressing irrelevant representations.

Gain Modulation / Compression

Task-relevant stimulus dimensions are amplified while irrelevant dimensions are compressed. The Compression Index (CPI) correlates with the strength of task belief in LPFC.

Compositionality

The C1 task is composed from components shared with C2 (colour categorization) and S1 (axis 1 response). This compositional structure enables flexible behavior and potential for rapid learning.

Task Design Overview

Three compositionally related categorization tasks sharing sensory and motor components

Compositional Task Design S1: Shape → Axis 1 Categorize SHAPE Bunny → UL Tee → LR UL LR Axis 1 C1: Colour → Axis 1 Categorize COLOUR Red → LR Green → UL UL LR Axis 1 C2: Colour → Axis 2 Categorize COLOUR Red → UR Green → LL UR LL Axis 2 Shared: Colour Categorization Shared: Colour Categorization