Reinforcement learning in multidimensional environments relies on attention mechanisms

Yael Niv, Reka Daniel, Andra Geana, Samuel J. Gershman, Yuan Chang Leong, Angela Radulescu, Robert C. Wilson

Research output: Contribution to journalArticle

115 Scopus citations

Abstract

In recent years, ideas from the computational field of reinforcement learning have revolutionized the study of learning in the brain, famously providing new, precise theories of how dopamine affects learning in the basal ganglia. However, reinforcement learning algorithms are notorious for not scaling well to multidimensional environments, as is required for real-world learning. We hypothesized that the brain naturally reduces the dimensionality of real-world problems to only those dimensions that are relevant to predicting reward, and conducted an experiment to assess by what algorithms and with what neural mechanisms this “representation learning” process is realized in humans. Our results suggest that a bilateral attentional control network comprising the intraparietal sulcus, precuneus, and dorsolateral prefrontal cortex is involved in selecting what dimensions are relevant to the task at hand, effectively updating the task representation through trial and error. In this way, cortical attention mechanisms interact with learning in the basal ganglia to solve the “curse of dimensionality” in reinforcement learning.

Original languageEnglish (US)
Pages (from-to)8145-8157
Number of pages13
JournalJournal of Neuroscience
Volume35
Issue number21
DOIs
StatePublished - May 27 2015

All Science Journal Classification (ASJC) codes

  • Neuroscience(all)

Keywords

  • Attention
  • Frontoparietal network
  • Model comparison
  • Reinforcement learning
  • Representation learning
  • fMRI

Fingerprint Dive into the research topics of 'Reinforcement learning in multidimensional environments relies on attention mechanisms'. Together they form a unique fingerprint.

  • Cite this

    Niv, Y., Daniel, R., Geana, A., Gershman, S. J., Leong, Y. C., Radulescu, A., & Wilson, R. C. (2015). Reinforcement learning in multidimensional environments relies on attention mechanisms. Journal of Neuroscience, 35(21), 8145-8157. https://doi.org/10.1523/JNEUROSCI.2978-14.2015