ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
654
Citations
0
Influential Citations
Frontiers in Human Neuroscience
Venue
2011
Year
Computational learning models are critical for understanding mechanisms of adaptive behavior. However, the two major current frameworks, reinforcement learning (RL) and Bayesian learning, both have certain limitations. For example, many Bayesian models are agnostic of inter-individual variability and involve complicated integrals, making online learning difficult. Here, we introduce a generic hierarchical Bayesian framework for individual learning under multiple forms of uncertainty (e.g., environmental volatility and perceptual uncertainty). The model assumes Gaussian random walks of states at all but the first level, with the step size determined by the next highest level. The coupling between levels is controlled by parameters that shape the influence of uncertainty on learning in a subject-specific fashion. Using variational Bayes under a mean-field approximation and a novel approximation to the posterior energy function, we derive trial-by-trial update equations which (i) are analytical and extremely efficient, enabling real-time learning, (ii) have a natural interpretation in terms of RL, and (iii) contain parameters representing processes which play a key role in current theories of learning, e.g., precision-weighting of prediction error. These parameters allow for the expression of individual differences in learning and may relate to specific neuromodulatory mechanisms in the brain. Our model is very general: it can deal with both discrete and continuous states and equally accounts for deterministic and probabilistic relations between environmental events and perceptual states (i.e., situations with and without perceptual uncertainty). These properties are illustrated by simulations and analyses of empirical time series. Overall, our framework provides a novel foundation for understanding normal and pathological learning that contextualizes RL within a generic Bayesian scheme and thus connects it to principles of optimality from probability theory.
This paper addresses a fundamental gap in computational models of learning: the need for a framework that can account for individual differences in learning under uncertainty while remaining computationally tractable for online learning. Traditional reinforcement learning (RL) models often lack a principled treatment of uncertainty, while Bayesian models, though theoretically elegant, frequently involve complicated integrals and are agnostic to inter-individual variability. Mathys (2011) bridges this divide by introducing a hierarchical Bayesian framework that captures multiple forms of uncertainty—such as environmental volatility and perceptual uncertainty—and yields efficient trial-by-trial update equations.
The significance of this work lies in its potential to unify RL and Bayesian learning under a single normative scheme. By deriving update equations that are analytical and extremely efficient, the model enables real-time learning and is suitable for analyzing behavioral and neural time series. Moreover, the parameters in the model have natural interpretations in terms of precision-weighting of prediction error, a concept central to current theories of learning and possibly linked to neuromodulatory mechanisms. This makes the framework particularly attractive for computational neuroscience, where it can be used to explain both normal and pathological learning.
Furthermore, the model's generality—handling both discrete and continuous states and accommodating deterministic and probabilistic relations—positions it as a versatile tool for a wide range of experimental paradigms. By contextualizing RL within a generic Bayesian scheme, the paper connects reinforcement learning to principles of optimality from probability theory, offering a deeper theoretical foundation for understanding adaptive behavior.
The paper presents simulations and analyses of empirical time series to illustrate the model's properties. While specific quantitative metrics are not detailed in the abstract, the key result is the derivation of efficient update equations that enable real-time learning. The simulations demonstrate the model's ability to handle various forms of uncertainty and to capture individual differences. The empirical analyses show the model's applicability to real data, though the abstract does not provide concrete performance comparisons with other models.
The broader impact of this work is twofold. First, it provides a computational foundation for understanding learning under uncertainty that is both principled and practical, making it a valuable tool for researchers in psychology and neuroscience. Second, by linking Bayesian inference with reinforcement learning, it offers a unifying framework that can guide the development of new algorithms in machine learning, particularly in areas requiring online learning and adaptation to changing environments. The emphasis on individual differences and precision-weighting of prediction error also has implications for understanding psychiatric disorders characterized by aberrant learning, such as schizophrenia and addiction.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba