The representational geometry of cognitive maps under dynamic cognitive control

Recent work has shown that the brain abstracts non-spatial relationships between entities or task states into representations called cognitive maps. Here, we investigated how cognitive control enables flexible top-down selection of goal-relevant information from multidimensional cognitive maps retrieved from memory. We examined the relationship between cognitive control and representational geometry by conducting parallel analyses of fMRI data and recurrent neural network (RNN) models trained to perform the same task. We found both stable map-like representations in a medial temporal lobe and orbitofrontal cortical network that reflected both task-relevant and irrelevant dimensions and dynamic, orthogonal representations of only relevant task dimensions in a frontoparietal network. These representational motifs also emerged with distinct temporal profiles over the course of training in the RNN, with map-like representations appearing first. We further show that increasing control demands due to incongruence (conflicting responses) between current task-relevant and irrelevant dimensions impact the geometry of subjective representations, and the degree of this effect further accounts for individual differences in cognitive control. Taken together, our findings show how complementary representational geometries balance stability and behavioral flexibility, and reveal an intricate bidirectional relationship between cognitive control and cognitive map geometry.