Posterior weighted reinforcement learning with state uncertainty.
Scientific Abstract
Reinforcement learning models generally assume that a stimulus is presented that allows a learner to unambiguously identify the state of nature, and the reward received is drawn from a distribution that depends on that state. However, in any natural environment, the stimulus is noisy. When there is state uncertainty, it is no longer immediately obvious how to perform reinforcement learning, since the observed reward cannot be unambiguously allocated to a state of the environment. This letter addresses the problem of incorporating state uncertainty in reinforcement learning models. We show that simply ignoring the uncertainty and allocating the reward to the most likely state of the environment results in incorrect value estimates. Furthermore, using only the information that is available before observing the reward also results in incorrect estimates. We therefore introduce a new technique, posterior weighted reinforcement learning, in which the estimates of state probabilities are updated according to the observed rewards (e.g., if a learner observes a reward usually associated with a particular state, this state becomes more likely). We show analytically that this modified algorithm can converge to correct reward estimates and confirm this with numerical experiments. The algorithm is shown to be a variant of the expectation-maximization algorithm, allowing rigorous convergence analyses to be carried out. A possible neural implementation of the algorithm in the cortico-basal-ganglia-thalamic network is presented, and experimental predictions of our model are discussed.
Similar content
Preprint
MRC CoRE in RND Output
ePC: Fast and Deep Predictive Coding in Digital Simulation
Paper
MRC CoRE in RND Output
ePC: Fast and Deep Predictive Coding in Digital Simulation
2026. 43rd International Conference on Machine Learning
Paper
MRC CoRE in RND Output
On the Infinite Width and Depth Limits of Predictive Coding Networks
2026, Proceedings of the 43rd International Conference on Machine Learning
Preprint
MRC CoRE in RND Output
Temporally distinct reward and action prediction error signals during value learning and habit formation
Posterior weighted reinforcement learning with state uncertainty.
Scientific Abstract
Reinforcement learning models generally assume that a stimulus is presented that allows a learner to unambiguously identify the state of nature, and the reward received is drawn from a distribution that depends on that state. However, in any natural environment, the stimulus is noisy. When there is state uncertainty, it is no longer immediately obvious how to perform reinforcement learning, since the observed reward cannot be unambiguously allocated to a state of the environment. This letter addresses the problem of incorporating state uncertainty in reinforcement learning models. We show that simply ignoring the uncertainty and allocating the reward to the most likely state of the environment results in incorrect value estimates. Furthermore, using only the information that is available before observing the reward also results in incorrect estimates. We therefore introduce a new technique, posterior weighted reinforcement learning, in which the estimates of state probabilities are updated according to the observed rewards (e.g., if a learner observes a reward usually associated with a particular state, this state becomes more likely). We show analytically that this modified algorithm can converge to correct reward estimates and confirm this with numerical experiments. The algorithm is shown to be a variant of the expectation-maximization algorithm, allowing rigorous convergence analyses to be carried out. A possible neural implementation of the algorithm in the cortico-basal-ganglia-thalamic network is presented, and experimental predictions of our model are discussed.
Citation
2010.Neural Comput, 22(5):1149-79.
Downloads
Similar content
Preprint
MRC CoRE in RND Output
ePC: Fast and Deep Predictive Coding in Digital Simulation
Paper
MRC CoRE in RND Output
ePC: Fast and Deep Predictive Coding in Digital Simulation
2026. 43rd International Conference on Machine Learning
Paper
MRC CoRE in RND Output
On the Infinite Width and Depth Limits of Predictive Coding Networks
2026, Proceedings of the 43rd International Conference on Machine Learning
Preprint
MRC CoRE in RND Output