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.

On Predictive Planning and Counterfactual Learning in Active Inference

On Predictive planning and counterfactual learning in active inference

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

On efficient computation in active inference

Dynamic planning in hierarchical active inference

Learning in Hybrid Active Inference Models

Deep hybrid models: infer and plan in a dynamic world

Counterfactual Reasoning about Intent for Interactive Navigation in Dynamic Environments

Deconstructing deep active inference

Deconstructing Deep Active Inference: A Contrarian Information Gatherer

Learning Generative State Space Models for Active Inference

Learning and embodied decisions in active inference

Predicting Future Actions of Reinforcement Learning Agents

Online Pareto-Optimal Decision-Making for Complex Tasks using Active Inference

An intelligent tutor for planning in large partially observable environments

Active Inference, Belief Propagation, and the Bethe Approximation

Learning adaptive planning representations with natural language guidance

Competitive Multi-agent Deep Reinforcement Learning with Counterfactual Thinking

Active Inference as a Model of Agency

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