Emergence of belief-like representations through reinforcement learning

Abstract

To behave adaptively, animals must learn to predict future reward, or value. To do this, animals are thought to learn reward predictions using reinforcement learning. However, in contrast to classical models, animals must learn to estimate value using only incomplete state information. Previous work suggests that animals estimate value in partially observable tasks by first forming “beliefs”—optimal Bayesian estimates of the hidden states in the task. Although this is one way to solve the problem of partial observability, it is not the only way, nor is it the most computationally scalable solution in complex, real-world environments. Here we show that a recurrent neural network (RNN) can learn to estimate value directly from observations, generating reward prediction errors that resemble those observed experimentally, without any explicit objective of estimating beliefs. We integrate statistical, functional, and dynamical systems perspectives on beliefs to show that the RNN’s learned representation encodes belief information, but only when the RNN’s capacity is sufficiently large. These results illustrate how animals can estimate value in tasks without explicitly estimating beliefs, yielding a representation useful for systems with limited capacity.

Document Details

Document Type
Pub Defense Publication
Publication Date
Sep 11, 2023
Source ID
10.1371/journal.pcbi.1011067

Entities

People

  • Jay A Hennig
  • Naoshige Uchida
  • Samuel J Gershman
  • Sandra A. Romero Pinto
  • Scott W. Linderman
  • Takahiro Yamaguchi

Organizations

  • Air Force Research Laboratory
  • National Institutes of Health

Tags

Fields of Study

  • Computer science

Readers

  • Artificial Intelligence
  • Brain and Cognitive Science; Experimental Psychology; Cognitive Neuroscience
  • Computational Modeling and Simulation

Technology Areas

  • AI & ML
  • AI & ML - Bayesian Inference
  • AI & ML - Machine Learning Algorithms
  • AI & ML - Neural Networks