A new study on meta reinforcement learning algorithms helps us understand how the human brain learns to adapt to complexity and uncertainty when learning and making decisions. A research team, led by Professor Sang Wan Lee at KAIST jointly with John O’Doherty at Caltech, succeeded in discovering both a computational and neural mechanism for human meta reinforcement learning, opening up the possibility of porting key elements of human intelligence into artificial intelligence algorithms. This study provides a glimpse into how it might ultimately use computational models to reverse engineer human reinforcement learning.
This work was published on Dec 16, 2019 in the journal Nature Communications. The title of the paper is “Task complexity interacts with state-space uncertainty in the arbitration between model-based and model-free learning.”
Human reinforcement learning is an inherently complex and dynamic process, involving goal setting, strategy choice, action selection, strategy modification, cognitive resource allocation etc. This a very challenging problem for humans to solve owing to the rapidly changing and multifaced environment in which humans have to operate. To make matters worse, humans often need to often rapidly make important decisions even before getting the opportunity to collect a lot of information, unlike the case when using deep learning methods to model learning and decision-making in artificial intelligence applications.
In order to solve this problem, the research team used a technique called 'reinforcement learning theory-based experiment design' to optimize the three variables of the two-stage Markov decision task - goal, task complexity, and task uncertainty. This experimental design technique allowed the team not only to control confounding factors, but also to create a situation similar to that which occurs in actual human problem solving.
Secondly, the team used a technique called ‘model-based neuroimaging analysis.’ Based on the acquired behavior and fMRI data, more than 100 different types of meta reinforcement learning algorithms were pitted against each other to find a computational model that can explain both behavioral and neural data. Thirdly, for the sake of a more rigorous verification, the team applied an analytical method called ‘parameter recovery analysis,’ which involves high-precision behavioral profiling of both human subjects and computational models.
In this way, the team was able to accurately identify a computational model of meta reinforcement learning, ensuring not only that the model’s apparent behavior is similar to that of humans, but also that the model solves the problem in the same way as humans do.
The team found that people tended to increase planning-based reinforcement learning (called model-based control), in response to increasing task complexity. However, they resorted to a simpler, more resource efficient strategy called model-free control, when both uncertainty and task complexity were high. This suggests that both the task uncertainty and the task complexity interact during the meta control of reinforcement learning. Computational fMRI analyses revealed that task complexity interacts with neural representations of the reliability of the learning strategies in the inferior prefrontal cortex.
These findings significantly advance understanding of the nature of the computations being implemented in the inferior prefrontal cortex during meta reinforcement learning as well as providing insight into the more general question of how the brain resolves uncertainty and complexity in a dynamically changing environment. Identifying the key computational variables that drive prefrontal meta reinforcement learning, can also inform understanding of how this process might be vulnerable to break down in certain psychiatric disorders such as depression and OCD. Furthermore, gaining a computational understanding of how this process can sometimes lead to increased model-free control, can provide insights into how under some situations task performance might break down under conditions of high cognitive load.
Professor Lee said, “This study will be of enormous interest to researchers in both the artificial intelligence and human/computer interaction fields since this holds significant potential for applying core insights gleaned into how human intelligence works with AI algorithms.”
This work was funded by the National Institute on Drug Abuse, the National Research Foundation of Korea, the Ministry of Science and ICT, Samsung Research Funding Center of Samsung Electronics.
< external_image >
Figure 1 (modified from the figures of the original paper doi:10.1038/s41467-019-13632-1). Computations implemented in the inferior prefrontal cortex during meta reinforcement learning. (A) Computational model of human prefrontal meta reinforcement learning (left) and the brain areas whose neural activity patterns are explained by the latent variables of the model. (B) Examples of behavioral profiles. Shown on the left is choice bias for different goal types and on the right is choice optimality for task complexity and uncertainty. (C) Parameter recoverability analysis. Compared are the effect of task uncertainty (left) and task complexity (right) on choice optimality.
“KAIST Space Institute will present a new paradigm for space research and education, foster creative talents, and become a space research center to lead the advancement of national space initiatives” (KAIST President Kwang-Hyung Lee) < Vision Declaration Ceremony Photo during Space Research Institute Opening Ceremony > KAIST (President Kwang-Hyung Lee opened the ‘Space Institute’, which brings together all of KAIST’s space technology capabilities to re
2024-09-30Stretchable displays, praised for their spatial efficiency, design flexibility, and human-like flexibility, are seen as the next generation of display technology. A team of Korean researchers has developed a stretchable display that can expand by 25% while maintaining clear image quality without distortion. It can also stretch and contract up to 5,000 times at 15% expansion without any performance degradation, making it the first deformation-free stretchable display with a negative Poisson's rat
2024-09-20< Photos from KAIST-POSTECH Science War (photographed by Student Junhyeok Park of KAIST Freshman Course) > The future leaders of science at KAIST and POSTECH (President Seong Keun Kim) held their annual science and sporting event at POSTECH for two days from September 20th to 21st. The 'KAIST-POSTECH Science War (hereafter KA-PO War)' is a festival consisting of science and sports games and various side events to promote exchange and cooperation between the two universities. It is als
2024-09-19< (From left) Professor Jimin Park of the Department of Chemical and Biomolecular Engineering and Dr. Inho Kim, a graduate of the Department of Materials Science and Engineering > KAIST (represented by President Kwang-Hyung Lee) announced on the 13th of September that Professor Jimin Park from KAIST’s Department of Chemical and Biomolecular Engineering and Dr. Inho Kim, a graduate from the Department of Materials Science and Engineering (currently a postdoctoral researcher at Ca
2024-09-13< Photo. Scholars gatheres for NRF Information Session at Chung Keun Mo Hall > KAIST’s International Office, headed by Vice President Soyoung Kim, successfully organized the ‘NRF Information Session for International Scholars’ on September 11, 2024, in collaboration with the National Research Foundation of Korea (NRF). The event was held at KAIST’s main campus to enourage the international scholar’s active participation in research projects and suppor
2024-09-13