Abstract:Active inference is a theory of perception, learning, and decision making that can be applied to neuroscience, robotics, psychology, and machine learning. Recently, intensive research has been taking place to scale up this framework using Monte Carlo tree search and deep learning. The goal of this activity is to solve more complicated tasks using deep active inference. First, we review the existing literature and then progressively build a deep active inference agent as follows: we (1) implement a variational autoencoder (VAE), (2) implement a deep hidden Markov model (HMM), and (3) implement a deep critical hidden Markov model (CHMM). For the CHMM, we implemented two versions, one minimizing expected free energy, CHMM[EFE] and one maximizing rewards, CHMM[reward]. Then we experimented with three different action selection strategies: the ε-greedy algorithm as well as softmax and best action selection. According to our experiments, the models able to solve the dSprites environment are the ones that maximize rewards. On further inspection, we found that the CHMM minimizing expected free energy almost always picks the same action, which makes it unable to solve the dSprites environment. In contrast, the CHMM maximizing reward keeps on selecting all the actions, enabling it to successfully solve the task. The only difference between those two CHMMs is the epistemic value, which aims to make the outputs of the transition and encoder networks as close as possible. Thus, the CHMM minimizing expected free energy repeatedly picks a single action and becomes an expert at predicting the future when selecting this action. This effectively makes the KL divergence between the output of the transition and encoder networks small. Additionally, when selecting the action down the average reward is zero, while for all the other actions, the expected reward will be negative. Therefore, if the CHMM has to stick to a single action to keep the KL divergence small, then the action down is the most rewarding. We also show in simulation that the epistemic value used in deep active inference can behave degenerately and in certain circumstances effectively lose, rather than gain, information. As the agent minimizing EFE is not able to explore its environment, the appropriate formulation of the epistemic value in deep active inference remains an open question.

Learning in Hybrid Active Inference Models

Hybrid Recurrent Models Support Emergent Descriptions for Hierarchical Planning and Control

Deep hybrid models: infer and plan in a dynamic world

Planning to Learn: A Novel Algorithm for Active Learning during Model-Based Planning

On Predictive Planning and Counterfactual Learning in Active Inference

Hybrid Active Inference

Dynamic planning in hierarchical active inference

Learning Generative State Space Models for Active Inference

Demonstrating the Continual Learning Capabilities and Practical Application of Discrete-Time Active Inference

Learning and embodied decisions in active inference

Active Inference and Reinforcement Learning: A unified inference on continuous state and action spaces under partial observability

On efficient computation in active inference

Active inference on discrete state-spaces: a synthesis

Continuous-Discrete Reinforcement Learning for Hybrid Control in Robotics

Learning Discrete State Abstractions With Deep Variational Inference

Learning for Long-Horizon Planning via Neuro-Symbolic Abductive Imitation

Abstract Value Iteration for Hierarchical Reinforcement Learning

Deconstructing Deep Active Inference: A Contrarian Information Gatherer

Deconstructing deep active inference