Harnessing the flexibility of neural networks to predict dynamic theoretical parameters underlying human choice behavior

Yoav Ger*, Eliya Nachmani, Lior Wolf, Nitzan Shahar

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Reinforcement learning (RL) models are used extensively to study human behavior. These rely on normative models of behavior and stress interpretability over predictive capabilities. More recently, neural network models have emerged as a descriptive modeling paradigm that is capable of high predictive power yet with limited interpretability. Here, we seek to augment the expressiveness of theoretical RL models with the high flexibility and predictive power of neural networks. We introduce a novel framework, which we term theoretical-RNN (t-RNN), whereby a recurrent neural network is trained to predict trial-by-trial behavior and to infer theoretical RL parameters using artificial data of RL agents performing a two-armed bandit task. In three studies, we then examined the use of our approach to dynamically predict unseen behavior along with time-varying theoretical RL parameters. We first validate our approach using synthetic data with known RL parameters. Next, as a proof-of-concept, we applied our framework to two independent datasets of humans performing the same task. In the first dataset, we describe differences in theoretical RL parameters dynamic among clinical psychiatric vs. healthy controls. In the second dataset, we show that the exploration strategies of humans varied dynamically in response to task phase and difficulty. For all analyses, we found better performance in the prediction of actions for t-RNN compared to the stationary maximum-likelihood RL method. We discuss the use of neural networks to facilitate the estimation of latent RL parameters underlying choice behavior.

Original languageEnglish
Article numbere1011678
JournalPLoS Computational Biology
Volume20
Issue number1
DOIs
StatePublished - Jan 2024

Funding

FundersFunder number
Israel Science Foundation2536/20
Tel Aviv University2923/20

    Fingerprint

    Dive into the research topics of 'Harnessing the flexibility of neural networks to predict dynamic theoretical parameters underlying human choice behavior'. Together they form a unique fingerprint.

    Cite this