T. Parr and K. J. Friston, “Generalised free energy and active inference,” Biol Cybern, vol. 113, no. 5, pp. 495–513, Dec. 2019, doi: 10.1007/s00422-019-00805-w.
> **Abstract**. Active inference is an approach to understanding behaviour that rests upon the idea that the brain uses an internal generative model to predict incoming sensory data. The fit between this model and data may be improved in two ways. The brain could optimise probabilistic beliefs about the variables in the generative model (i.e. perceptual inference). Alternatively, by acting on the world, it could change the sensory data, such that they are more consistent with the model. This implies a common objective function (variational free energy) for action and perception that scores the fit between an internal model and the world. We compare two free energy functionals for active inference in the framework of Markov decision processes. One of these is a functional of beliefs (i.e. probability distributions) about states and policies, but a function of observations, while the second is a functional of beliefs about all three. In the former (expected free energy), prior beliefs about outcomes are not part of the generative model (because they are absorbed into the prior over policies). Conversely, in the second (generalised free energy), priors over outcomes become an explicit component of the generative model. When using the free energy function, which is blind to future observations, we equip the generative model with a prior over policies that ensure preferred (i.e. priors over) outcomes are realised. In other words, if we expect to encounter a particular kind of outcome, this lends plausibility to those policies for which this outcome is a consequence. In addition, this formulation ensures that selected policies minimise uncertainty about future outcomes by minimising the free energy expected in the future. When using the free energy functional—that effectively treats future observations as hidden states—we show that policies are inferred or selected that realise prior preferences by minimising the free energy of future expectations. Interestingly, the form of posterior beliefs about policies (and associated belief updating) turns out to be identical under both formulations, but the quantities used to compute them are not.
# Introduction Over the past years, we have tried to establish active inference (a corollary of the free energy principle) as a relatively straightforward and principled explanation for action, perception and cognition. Active inference can be summarised as self-evidencing (Hohwy 2016), in the sense that action and perception can be cast as maximising Bayesian model evidence, under generative models of the world. When this maximisation uses approximate Bayesian inference, this is equivalent to minimising variational free energy (Friston et al. 2006)—a form of bounded rational behaviour that minimises a variational bound on model evidence. Recently, we have migrated the basic idea from models that generate continuous sensations (like velocity and luminance contrast) (Brown and Friston 2012) to discrete state-space models, specifically Markov decision processes (Friston et al. 2017a). These models represent the world in terms of discrete states, like I am on this page and reading this word (Friston et al. 2017d). Discrete state-space models can be inferred using belief propagation (Yedidia et al. 2005) or variational message passing (Dauwels 2007; Winn 2004) schemes that have a degree of neuronal plausibility (Friston et al. 2017c). The resulting planning as inference scheme (Attias 2003; Baker et al. 2009; Botvinick and Toussaint 2012; Verma and Rao 2006) has a pleasingly broad explanatory scope, accounting for a range of phenomena in cognitive neuroscience, active vision and motor control (see Table 1). In this paper, we revisit the role of (expected) free energy in active inference and offer an alternative, simpler and more general formulation. This formulation does not substantially change the message passing or belief updating; however, it provides an interesting perspective on planning as inference and the way that we may perceive the future.
Table 1 Applications of active inference for Markov decision processes Full size table In current descriptions of active inference, the basic argument goes as follows: active inference is based upon the maximisation of model evidence or minimisation of variational free energy in two complementary ways. First, one can update one’s beliefs about latent or hidden states of the world to make them consistent with observed evidence—or one can actively sample the world to make observations consistent with beliefs about states of the world. The important thing here is that both action and perception are in game of minimising the same quantity, namely variational free energy. A key aspect of this formulation is that action (i.e. behaviour) is absorbed into inference, which means that agents have beliefs about what they are doing—and will do. This calls for prior beliefs about action or policies (i.e. sequences of actions). So where did these prior beliefs come from?
The answer obtains from a reductio ad absurdum argument: if action realises prior beliefs and minimises free energy, then the only tenable prior beliefs are that action will minimise free energy. If this were not the case, we reach the following absurd conclusion. If a free energy minimising creature did not have the prior belief that it selects policies that minimise (expected) free energy, it would infer (and therefore pursue) policies that were not free energy minimising. As such, it would not be a free energy minimising creature, which is a contradiction. This leads to the prior belief that I will select policies that minimise the free energy expected under that policy. The endpoint of this argument is that action or policy selection becomes a form of Bayesian model selection, where the evidence for a particular policy becomes the free energy expected in the future. This expected free energy is a slightly unusual objective function because it scores the evidence for plausible policies based on outcomes that have yet to be observed. This means that the expected free energy becomes the variational free energy expected under (posterior predictive) beliefs about outcomes. These priors are usually informed by prior beliefs about outcomes that play the role of prior preferences or utility functions in reinforcement learning and economics.
In summary, beliefs about states of the world and policies are continuously updated to minimise variational free energy, where posterior beliefs about policies (that prescribe action) are based upon expected free energy (that may or may not include prior preferences over future outcomes). This is the current story and leads to interesting issues that rest on the fact that expected free energy can be decomposed into epistemic and pragmatic parts (Friston et al. 2015). This decomposition provides a principled explanation for the epistemics of planning and inference that underwrite the exploitation and exploration dilemma, novelty, salience and so on. However, there is another way of telling this story that leads to a conceptually different sort of interpretation.
In what follows, we show that the same Bayesian policy (model) selection obtains from minimising variational free energy when future outcomes are treated as hidden or latent states of the world. In other words, we can regard active inference as minimising a generalised free energy under generative models that entertain the consequences of (policy-dependent) hidden states of the world in the future. This simple generalisation induces posterior beliefs over future outcomes that now play the role of latent or hidden states. In this setting, the future is treated in exactly the same way as the hidden or unobservable states of the world generating observations in the past. On this view, one gets the expected free energy for free, because the variational free energy involves an expectation under posterior beliefs over future outcomes. In turn, this means that beliefs about states and policies can be simply and uniformly treated as minimising the same (generalised) free energy, without having to invoke any free energy minimising priors over policies.
Technically, this leads to the same form of belief updating and (Bayesian) policy selection but provides a different perspective on the free energy principle per se. This perspective says that self-evidencing and active inference both have one underlying imperative, namely to minimise generalised free energy or uncertainty. When this uncertainty is evaluated under models that generate outcomes in the future, future outcomes become hidden states that are only revealed by the passage of time. In this context, outcomes in the past become observations in standard variational inference, while outcomes in the future become posterior beliefs about latent observations that have yet to disclose themselves. In this way, the generalised free energy can be seen as comprising variational free energy contributions from the past and future. The current paper provides the formal basis for the above arguments. In brief, we will see that both the expected and generalised free energy formulations lead to the same update equations. However, there is a subtle difference. In the expected free energy formalism, prior preferences or beliefs about outcomes are used to specify the prior over policies. In the generalised formulation, prior beliefs about outcomes in the future inform posterior beliefs about the hidden states that cause them. Because of the implicit forward and backward message passing in the belief propagation scheme obtained at the free energy minimum (Yedidia et al. 2005), these prior beliefs or preferences act to distort expected trajectories (into the future) towards preferences in an optimistic way (Sharot et al. 2012). Intuitively, the expected free energy contribution to generalised free energy evaluates the (complexity) cost of this distortion, thereby favouring policies that lead naturally to preferred outcomes—without violating beliefs about state transitions and the (likelihood) mapping between states and outcomes. The implicit coupling between beliefs about the future and current actions means that, in one sense, the future can cause the past.
Framing probabilistic reasoning in terms of inferential message passing formalises several prominent concepts in the study of human decision making. The idea that prior beliefs distort beliefs about future and that this optimism about the future propagates backwards in time to influence behaviour in an adaptive way (McKay and Dennett 2010; Sharot 2011), is highly consistent with an influence of beliefs about the future over beliefs about the present. Simplistically, the idea behind these accounts is that adaptive behaviour relies upon the (possibly false) belief that future events will accord with our preferences. It is only by believing that we will realise these goals that we act in a manner consistent with their realisation. Intuitively, without the belief that we will end up eating dinner, there would be no reason to shop for ingredients. The passing of messages from past to future resonates with the notion that working memory is vital for predicting the future and planning actions accordingly (Gilhooly 2005; Hikosaka et al. 2000), and underwrites research on episodic future thinking and counterfactual reasoning (Schacter et al. 2015). Appealing to bidirectional inferential message passing has enabled us to reproduce a range of behavioural and electrophysiological phenomena through simulation (summarised in Table 1).
This paper comprises three sections. In the first, we outline the approach we have used to date (i.e. minimising the variational free energy under prior beliefs that policies with a low expected free energy are more probable). In the second, we introduce a generalisation of the variational free energy that incorporates beliefs about future outcomes. The third section compares these two approaches conceptually and through illustrative simulations.
Active inference and variational free energy
The free energy principle is motivated by the defining characteristic of living creatures, namely that they persist in the face of a changing world. In other words, their states occupy a small proportion of all possible states with a high probability. From the perspective of statistical physics, this means that they show a form of self-organised, non-equilibrium steady-state that maintains a low entropy probability distribution over their states. In information theory, self-information or surprise (a.k.a. negative log model evidence) averaged over time is entropy. More generally, entropy is defined in terms of an ensemble average. However, under that assumption that a system has achieved its (non-equilibrium) steady state, the ensemble and time average are equivalent (under mild weakly mixing assumptions). This means, at any given time, all biological systems are compelled to minimise their surprise. While this may seem like a very bold statement, we do not intend to trivialise the many constraints that dictate behaviour. The point here is that when all these constraints are written into a generative model as prior beliefs, they all contribute to the same cost function: surprise. This reframes the problem of expressing the constraints biological systems must satisfy as a problem of specifying the right set of priors. Although the computation of surprise is often intractable, an approximation is simple to calculate. This is variational free energy (Beal 2003; Dayan et al. 1995; Friston 2003) which depends upon specifying a generative model of how data are caused. This generative model comprises a series of conditional probability distributions. For a Markov decision process, it assumes a series of states (s) that evolve through time. At each time step, the probability of transitioning from one state to the next depends upon a policy (π). Neither states nor policies are directly accessible to the creature in question. However, each state probabilistically generates an observable outcome (o). As Jensen’s inequality demonstrates, free energy is an upper bound on surprise.
𝐹⏟Freeenergy=−𝐸𝑄(𝑠̃,𝜋)[ln𝑃(𝑜̃,𝑠̃,𝜋)𝑄(𝑠̃,𝜋)]≥−ln𝐸𝑄(𝑠̃,𝜋)[𝑃(𝑜̃,𝑠̃,𝜋)𝑄(𝑠̃,𝜋)]Jensen′sinequality=−ln𝑃(𝑜̃)Surprise F ⏟ Free energy = − E Q ( s ~ , π ) [ ln P ( o ~ , s ~ , π ) Q ( s ~ , π ) ] ≥ − ln E Q ( s ~ , π ) [ P ( o ~ , s ~ , π ) Q ( s ~ , π ) ] ⏟ Jensen ′ s inequality = − ln P ( o ~ ) ⏟
Surprise In the equation above, indicates a probability distribution over outcomes 𝑜̃=(𝑜1,𝑜2,…,𝑜𝑇) o ~ = ( o 1 , o 2 , … , o T ) that are generated by hidden states of the world 𝑠̃=(𝑠1,𝑠2,…,𝑠𝑇) s ~ = ( s 1 , s 2 , … , s T )
and policies, which define the generative model. The generative model is thus expressed as a joint probability distribution over outcomes (i.e. consequences) and their causes (i.e. hidden states of the world and policies available to the agent). Marginalising (i.e. summing or integrating) over the states and policies gives the evidence (a.k.a., marginal likelihood). The log of this marginal likelihood is negative surprise. is a probability distribution over unobservable (hidden) states and policies—that becomes an approximate posterior distribution as free energy is minimised. The minimisation of free energy over time ensures entropy does not increase, thereby enabling biological systems to resist the second law of thermodynamics and their implicit dissipation or decay.
Note that the generative model is not a model of the biological system itself, but an implicit model of how the environment generates its sensory data. The dynamics of inference and behaviour that we are interested in here emerge from minimising free energy under an appropriate choice of generative model. For readers with a physics background, and analogy would be that the free energy plays the role of a Lagrangian whose ‘potential energy’ component is given by the generative model. Just as a Lagrangian is used to recover the equations of motion for a physical system, we use the free energy to recover the belief updates that determine a biological system’s behaviour.
In the following, we begin by describing the form of the generative model we have used to date. We will then address the form of the approximate posterior distribution. To make inference tractable, this generally involves a mean-field approximation that factorises the approximate posterior distribution into independent factors or marginal distributions.
The generative models used in this paper are subtly different for each free energy functional, but the variables themselves are the same. These are policies (𝜋 π ) and state trajectories (𝑠̃ s ~ ), all of which are latent (unknown random) variables that have to be inferred. States evolve as a discrete Markov chain, where the transition probabilities are functions of the policy. Likelihood distributions probabilistically map hidden states to observations (𝑜̃ o ~ ). Figure (left) shows these dependencies as a graphical Bayesian network. This type of generative model has been used extensively in simulations of active inference (FitzGerald et al. , ; Friston et al. , , , ; Schwartenbeck et al. ), see Table .
Fig. 1 figure1 Markov decision process. This shows the basic structure of the discrete state-space generative model used in this paper, assuming the current time is t = τ. The factor graph on the left is the generative model we have used in previous work. Importantly, the prior belief about observations only enters this graph through the expected free energy, 𝐺 G (see main text), which enters the prior over policies. Policies index alternative trajectories, or sequences, of actions. In this sense, they are not time dependent, as each policy determines a sequence of actions for all time-points. Conversely, the actions (u) are time dependent. U is an array that specifies an action for each time step (rows) and each policy (columns). The selected action therefore depends upon the most likely policy and the action that policy implies for that time step. Action selection is technically not part of the generative model, as it relies upon the posterior distribution Q (please see main text for details), obtained by inverting the model. This is an important, aspect of active inference, as it underwrites the way in which the system performing inference may change the process generating its observed data. The grey region of this graph indicates that the observation at the next time step is not yet available, so cannot yet be incorporated into the graph. The right factor graph is the new version of the generative model considered in this paper. This generative model does not require an expected free energy, and the prior over outcomes enters the model directly as a constraint on outcomes. This also shows a time dependence, as future outcomes are treated as unobserved latent variables (indicated by an unfilled circle). Observed variables are shown as filled circles in both graphs and unobserved variables as unfilled circles. Factors of the generative model (i.e. conditional probability distributions and prior probabilities) are shown as squares. These squares are connected to those circles containing variables that participate in the same factor. Please refer to the main text and Table 2 for a description of the variables. In the panel on the right, the definitions are given for each of the factors in blue squares. Here, Cat refers to the categorical distribution (color figure online)
Full size image
The role of the generative model is simply to define the free energy functional which, as we will see in Sects. –, gives rise to the belief update rules that we will employ for our simulations. However, it is helpful to imagine how we might generate data from such a model. We outline this process with the model on the left of Fig. in mind. We could start at the first time step and sample a state from the categorical prior over initial states. The parameters of this prior () are simply a vector of probabilities for each alternative state. From this, we can now sample from the likelihood. This is formulated as a matrix (), whose columns correspond to a state and whose rows are the alternative outcomes that may be generated. To generate an outcome, we would select the column of this matrix corresponding to the state we sampled and sample an outcome from this column-vector of probabilities. It is this outcome that would be available to a synthetic creature.
Taking a discrete time step into the future, we can sample a new state from the column of a transition matrix () associated with the state at the previous time. Crucially, the transition probabilities are conditioned upon the selected action. This means we have a separate -matrix for each action. Action selection depends upon the policy, with each policy and time point associated with an action. For the model on the left of Fig. , this means we calculate the expected free energy () for each policy, which depends upon a vector of prior probabilities for outcomes under these policies (). Combining these with a prior bias term ()—as set out in more detail in Sect. —we can construct a prior over policies. Sampling from this and selecting the action that corresponds to this policy, at this time, specify the -matrix from which to sample the state for the current time step. We could then sample the outcome for this time from the relevant column of the -matrix. This process can be repeated for a series of discrete time steps, generating a new outcome for each time. A similar approach could be taken to generate data from the model on the right of Fig. . However, note that the likelihood here comprises both and , and the policy prior only includes (i.e. the expected free energy does not explicitly feature in this model). The procedure outlined above provides an intuition into the beliefs a creature has about how its sensory data are generated by acting on hidden states in the environment.
It is worth noting that the free energy is a of the distributions in the generative model and of the approximate posterior beliefs, but a of observations. Continuing with this free energy, we now consider the mean-field approximation in current implementations of active inference, and its consequences for the variational free energy. In the next few sections, we unpack the variational free energy, and its role in active inference based on Markov decision processes. The argument that follows is a little involved, but we summarise the key steps here, such that the agenda of each of the following sections is clear. In Sect. , we specify the form of the variational distribution we employ, and the free energy that results from this. In Sect. , we unpack the terms in the free energy as they pertain to the generative model. This depends upon having a prior belief about policies. Section attempts to identify this prior, through finding the optimal posterior and extrapolating backwards in time. This highlights a shortcoming of this approach that is resolved in Sect. . In addition to providing a more appropriate prior for policy selection, Sect. sets out the role of free energy in simulating behaviour. In brief, this involves finding the variational distribution over policies that minimises free energy. As free energy is a function of sensory observations, this means we need to update these distributions following each new observation. Section follows the same approach to find the free energy minima for beliefs about states, giving the fixed points to which these distributions must be updated following each new observation.
Definition of the mean-field variational free energy To define the variational free energy for the above generative model, we first need to specify the form of the approximate posterior distribution, 𝑄 Q . We do this via a mean-field approximation that treats the (policy dependent) state at each time step as approximately independent of the state at any other time step. We treat the distribution over the policy as a separate factor, which implies a set of (policy) models, 𝜋 π , over hidden variables 𝑠𝜏 s τ : 𝑄(𝑠̃,𝜋)=𝑄(𝜋)∏𝜏𝑄(𝑠𝜏|𝜋) Q ( s ~ , π ) = Q ( π ) ∏ τ Q ( s τ | π )
Mean-field approximations originated in statistical physics, where they can be used to approximate Helmholtz free energy through appealing to an average with respect to a ‘reference’ Hamiltonian. This reference can be simply defined for a system with non-interacting components (or degrees of freedom). In virtue of the assumption that the system’s degrees of freedom do not interact, their Hamiltonian (scaled negative log probability) may be expressed as a sum of contributions from each component. Exponentiating this sum, the probability density can be expressed as a product of marginal probabilities for each degree of freedom. The same idea has been employed extensively in statistical inference and machine learning, where a mean-field approximation refers to the use of a variational distribution comprising a product of marginals (Winn and Bishop ; Yedidia et al. ). The ‘mean field’ is the expected value of each (log) factor of the generative model (), which include the interactions, under the fully factorised distribution (). The advantage to using a mean-field approximation is the computational tractability that comes from being able to separately optimise each marginal distribution. We can now substitute this factorised distribution into our definition for the variational free energy above:
𝐹=𝐸𝑄(𝜋)[𝐹𝜋]+𝐷𝐾𝐿[𝑄(𝜋)||𝑃(𝜋)]𝐹𝜋=−𝐸𝑄(𝑠̃|𝜋)[ln𝑃(𝑜̃,𝑠̃|𝜋)−∑𝜏ln𝑄(𝑠𝜏|𝜋)] F= E Q ( π ) [ F π ]+ D K L [Q(π) | | P(π)] F π =− E Q ( s ~ | π ) [lnP( o ~ , s ~ | π)− ∑ τ ln Q ( s τ | π ) ]
In this form, the variational free energy is expressed in terms of policy-dependent terms (second equality) that bound the (negative log) evidence for each policy and a complexity cost or KL divergenceFootnote 1 () that scores the departure of the posterior beliefs over policies from the corresponding prior beliefs.
Past and future
There is an important difference in how past and future outcomes are treated by the variational free energy. Note that—as a function of outcomes—the components of the free energy that depend on outcomes can only be evaluated for the past and present. Hidden states, on the other hand, enter the expression as about states. In other words, the free energy is a functional of distributions over states, rather than a function, as in the case of outcomes. This means that free energy evaluation takes account of future states. We can express this explicitly by writing the variational free energy, at time , as a sum over all time steps, factorising the generative distribution according to the conditional independencies expressed in Fig. :
𝐹𝜋=∑𝜏𝐹𝜋𝜏𝐹𝜋𝜏=−𝐸𝑄(𝑠𝜏|𝜋)𝑄(𝑠𝜏−1|𝜋)[[𝜏≤𝑡]⋅ln𝑃(𝑜𝜏|𝑠𝜏)+ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)−ln𝑄(𝑠𝜏|𝜋)] F π = ∑ τ F π τ F π τ =− E Q ( s τ | π ) Q ( s τ − 1 | π ) [ [ τ ≤ t ] ⋅ ln P ( o τ | s τ ) + ln P ( s τ | s τ − 1 , π ) − ln Q ( s τ | π ) ]
In the above, the Iverson (square) brackets return 1 if the expression is true, and 0 otherwise. It is this condition that differentiates contributions from the past from the future. This equation is obtained in a straightforward way by factorising the generative model (joint distribution) in the second line of Eq. , in line with the generative model depicted in Fig. . Because the generative model does not include future observations as random variables (given that these data have yet to be collected), there are no accompanying likelihood factors. This reflects the fact that the only data that contribute to the free energy are those we currently have access to. Given the dependence of the right-hand side on the current time () the free energy should, strictly speaking, be written as a function of and . As we are interested here in online inference, we will assume an implicit conditioning upon for all free energies throughout this paper. The Iverson brackets above allow us to decompose the sum into past and future components:
𝐹𝜋=∑𝜏≤𝑡𝐹𝜋𝜏+∑𝜏>𝑡𝐸𝑄(𝑠𝜏−1|𝜋)[𝐷KL[𝑄(𝑠𝜏|𝜋)||𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]]Complexity F π = ∑ τ ≤ t F π τ + ∑ τ > t E Q ( s τ − 1 | π ) [ D KL [ Q ( s τ | π ) | | P ( s τ | s τ − 1 , π ) ] ] ⏟
Complexity (5) In this decomposition, the contribution of beliefs about future states reduces to a complexity cost. This is the KL divergence between approximate posterior beliefs about states in the future and prior beliefs. The latter are based upon the (policy-specific) transition probabilities in the generative model.
Policy posteriors and priors
Using the full variational free energy (over all policies) from Eq. 3, we can evaluate posterior beliefs about policies. The variational derivative of the free energy with respect to these beliefs is (where we omit constants, and where 𝜎(⋅) σ ( ⋅ ) is a softmax function—i.e. a normalised exponential function): 𝛿𝐹𝛿𝑄(𝜋)=𝐹𝜋−ln𝑃(𝜋)+ln𝑄(𝜋)𝛿𝐹𝛿𝑄(𝜋)=0⇔𝑄(𝜋)=𝜎(ln𝑃(𝜋)−𝐹𝜋) δ F δ Q ( π ) = F π −lnP(π)+lnQ(π) δ F δ Q ( π ) =0⇔Q(π)=σ(lnP(π)− F π ) (6)
The second line derives from the first through rearranging, exponentiating both sides of the equation, and normalising to ensure the approximate posterior sums to one. This, together with Eq. 5, implies the belief prior to any observations (i.e. at 𝑡=0 t = 0 ), which is given by: 𝑄𝑜(𝜋)=𝜎(ln𝑃(𝜋)−∑𝜏𝐸𝑄(𝑠𝜏−1|𝜋)[𝐷𝐾𝐿[𝑄(𝑠𝜏|𝜋)||𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]]) Q o ( π ) = σ ( ln P ( π ) − ∑ τ E Q ( s τ − 1 | π ) [ D K L [ Q ( s τ | π ) | | P ( s τ | s τ − 1 , π ) ] ] ) (7)
This is an unsatisfying result, in which it fails to accommodate our prior knowledge that outcomes will become available in the future. In other words, the posterior at each time step is calculated under a different model (see Fig. 2). Fig. 2 figure2 Temporal progression of Markov decision process. The upper graphs shows the structure of the generative model implied using the variational free energy, equipped with a prior that the expected free energy will be minimised by policy selection. Observations are added to the model as they occur. The lower graphs show the structure of the generative model that explicitly represents future outcomes, and minimises a generalised free energy through policy selection. As observations are made, the outcome variables collapse to delta functions. These graphics are intended to highlight two alternative conceptions of a generative model employed in an online setting. The key problem here is how to deal with missing (future) outcomes. These could be omitted until such a time as they become available. Alternatively, they could be treated as hidden variables about which we can hold beliefs. Please note that this graphic illustrates different ways of formulating the generative model used to calculate belief updates. It does not show belief updates, behaviour or any other free energy minimising process. These will be detailed in subsequent sections and figures. However, the reason for making this distinction is important for how we formulate the free energy. The key distinction between the free energies compared in this paper is which of the two perspectives on future outcomes we choose to adopt
Full size image
Expected free energy
To finesse this shortcoming, we can assume agents select the policy that they expect will lead to the lowest free energy (summed over time). This is motivated by the reductio ad absurdum in the introduction and is expressed mathematically as: 𝑄𝑜(𝜋)≜𝜎(ln𝑃(𝜋)−𝐺𝜋) Q o ( π ) ≜ σ ( ln P ( π ) − G π ) (8) This replaces the expression for 𝑄𝑜(𝜋) Q o ( π ) given in Sect. 2.3. (We retain the notation Qo(π) for the prior here to distinguish this from the fixed form prior P(π), which does not depend on the beliefs about states.) Gπ is the expected free energy, conditioned on a policy. It is defined as: 𝐺𝜋=∑𝜏>𝑡𝐺𝜋𝜏𝐺𝜋𝜏=−𝐸𝑄̃(𝑜𝜏,𝑠𝜏|𝜋)[ln𝑃(𝑜𝜏,𝑠𝜏)−ln𝑄(𝑠𝜏|𝜋)] G π = ∑ τ > t G π τ G π τ =− E Q ~ ( o τ , s τ | π ) [lnP( o τ , s τ )−lnQ( s τ | π)] (9)
There is an apparent problem with this quantity: The first term within the expectation is a function of outcomes that have yet to be observed. To take this into account, we have defined an (approximate) joint distribution over states and outcomes: 𝑄̃(𝑜𝜏,𝑠𝜏|𝜋)=𝑃(𝑜𝜏|𝑠𝜏)𝑄(𝑠𝜏|𝜋) Q ~ ( o τ , s τ | π ) = P ( o τ | s τ ) Q ( s τ | π ) , and take the expectation with respect to this. This means that we can express a (posterior predictive) belief about the observations in the future based on (posterior predictive) beliefs about hidden states. One can obtain a useful form of the expected free energy by rearranging the above: if we factorise the generative model, we obtain: 𝐺𝜋𝜏=−𝐸𝑄̃(𝑜𝜏,𝑠𝜏|𝜋)[ln𝑃(𝑠𝜏|𝑜𝜏)−ln𝑄(𝑠𝜏|𝜋)Epistemicvalue+ln𝑃(𝑜𝜏)⏟Extrinsicvalue] G π τ = − E Q ~ ( o τ , s τ | π ) [ ln P ( s τ | o τ ) − ln Q ( s τ | π ) ⏟ Epistemic value + ln P ( o τ ) ⏟ Extrinsic value ] (10)
This form shows that policies that have a low expected free energy are those that resolve uncertainty, and that fulfil prior preferences about outcomes. It is the first of these terms that endorses the metaphor of the brain as a scientist, performing experiments (i.e. actions with sensory consequences) to verify or refute hypotheses about the world (Friston et al. 2012b; Gregory 1980).
The second term speaks to the notion of a ‘crooked scientist’ (Bruineberg et al. 2016), who designs experiments to confirm prior beliefs, i.e. preferred outcomes. This preference is the same as the evidence (a.k.a., marginal likelihood) associated with a given model. This means policies are selected such that the most probable outcomes under that policy match the most probable outcomes under prior preferences (defined in terms of a marginal likelihood). Treating 𝑄(𝑠𝜏|𝜋) Q ( s τ | π ) as a prior, and 𝑃(𝑠𝜏|𝑜𝜏) P ( s τ | o τ ) as a posterior, we can directly substitute these into Bayes’ rule, which says that their ratio is equal to the ratio of the corresponding likelihood (𝑄(𝑜𝜏|𝑠𝜏,𝜋)≈𝑃(𝑜𝜏|𝑠𝜏) Q ( o τ | s τ , π ) ≈ P ( o τ | s τ ) ) and marginal likelihood (𝑄(𝑜𝜏|𝜋) Q ( o τ | π ) ): 𝑃(𝑠𝜏|𝑜𝜏)𝑄(𝑠𝜏|𝜋)=𝑄(𝑜𝜏|𝑠𝜏,𝜋)𝑄(𝑜𝜏|𝜋) P ( s τ | o τ ) Q ( s τ | π ) = Q ( o τ | s τ , π ) Q ( o τ | π ) (11) Due to the symmetry of Bayes’ rule, another perspective on this is that 𝑃(𝑠𝜏|𝑜𝜏) P ( s τ | o τ ) is a likelihood that generates states from observations. This view treats the right-hand side of the above as the ratio between a posterior and a prior. Using this relationship, we can express expected free energy in terms of risk and ambiguity: 𝐺𝜋𝜏=𝐷𝐾𝐿[𝑄(𝑜𝜏|𝜋)||𝑃(𝑜𝜏)]Risk+𝐸𝑄(𝑠𝜏|𝜋)[𝐻[ln𝑃(𝑜𝜏|𝑠𝜏)]]Ambiguity G π τ = D K L [ Q ( o τ | π ) | | P ( o τ ) ] ⏟ Risk + E Q ( s τ | π ) [ H [ ln P ( o τ | s τ ) ] ] ⏟
Ambiguity In this equation, H is the Shannon entropy (i.e. negative expected log probability). This means that the prior belief about outcomes enters the generative model through the KL divergence between outcomes expected under any policy and prior preferences. This form also illustrates the correspondence between the expected free energy and the quantities ‘risk’ and ‘ambiguity’ from behavioural economics (Ellsberg 1961; Ghirardato and Marinacci 2002). Risk quantifies the expected cost of a policy as a divergence from preferred outcomes and is sometimes referred to as Bayesian risk or regret (Huggins and Tenenbaum 2015), which underlies KL control and related Bayesian control rules (Kappen et al. 2012; Ortega and Braun 2010; Todorov 2008) and special cases that include Thompson sampling (Lloyd and Leslie 2013; Strens 2000). Ambiguous states are those that have an uncertain mapping to observations. The greater these quantities, the less likely it is that the associated policy will be chosen.
Having identified a suitable prior belief for policies 𝑄𝑜(𝜋) Q o ( π ) , we can calculate the fixed point of the free energy with respect to the variational posterior over policies and use this to update the posterior after each time step: 𝛿𝐹𝛿𝑄(𝜋)=0⇔𝑄(𝜋)=𝜎(ln𝑄0(𝜋)−𝐹𝜋(𝜋))=𝜎(ln𝑃(𝜋)−𝐺(𝜋)−𝐹𝜋) δ F δ Q ( π ) =0⇔Q(π)=σ(ln Q 0 (π)− F π (π)) =σ(lnP(π)−G(π)− F π ) (12) This highlights the way in which the expected free energy influences policy selection. Distributions over policies are updated at each time step to a fixed point that depends upon the expected free energy. The expected free energy is a functional of posterior beliefs about states. Section 2.5 sets out how these may be optimised in relation to sensory outcomes. Hidden state updates To complete our description of active inference, we derive the belief update equations for the hidden states: 𝛿𝐹𝜋𝛿𝑄(𝑠𝜏|𝜋)=−ln𝑃(𝑜𝜏|𝑠𝜏)−𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]−𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)]+ln𝑄(𝑠𝜏|𝜋)𝛿𝐹𝜋𝛿𝑄(𝑠𝜏|𝜋)=0⇔𝑄(𝑠𝜏|𝜋)=𝜎(ln𝑃(𝑜𝜏|𝑠𝜏)+𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]+𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)]) δ F π δ Q ( s τ | π ) =−lnP( o τ | s τ )− E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)]− E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)]+lnQ( s τ | π) δ F π δ Q ( s τ | π ) =0⇔Q( s τ | π)=σ(lnP( o τ | s τ )+ E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)]+ E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)]) (13) This result says that, to minimise free energy, we update beliefs about states under policies at each time step such that they are equal to a softmax function of a sum of expected log probabilities. These are the terms in the generative model that depend upon the state about which we optimise beliefs. Technically, these are the state’s Markov blanket (Pearl 1998). These comprise the constraints based upon beliefs about the previous state, the next state and the sensory outcome generated by the current state. The expectations here are simple to calculate, in virtue of the categorical distributions used to define the model and variational posterior. Practically, this means that the sufficient statistics of these are vectors (or matrices, for conditional distributions), where each element is the probability of each alternative value the state can take. (For conditional distributions, these are matrices where each column is a different value for the variable in the conditioning set.) Table 2 sets out the notation used for these sufficient statistics. Crucially, the linear algebraic expression of these statistics means expectations reduce to matrix–vector multiplications or dot products as set out in Fig. 3. Note that as we progress through time, new outcomes become available. As the free energy minima depend upon available outcomes, this means we need to update the variational posteriors following each new outcome. Table 2 Variables in update equations Full size table Fig. 3 figure3 Belief update equations. The blue panels show the update equations using the standard variational approach. The pink panels show the update equations when the generalised free energy is used. The equations in this figure show the fixed points for the sufficient statistics of each variational distribution. These are calculated as in the main text by finding the minima of each of the free energy functionals. As such, updating the variational distributions (left-hand side of each equation) to their fixed points (right-hand side of each equation) following each new observation minimises the corresponding free energy. The dotted outline indicates the correspondence between the generalised free energy and the sum of the variational and expected free energies, and therefore the equivalence of the form of the posteriors over policies. However, it should be remembered that the variables within these equations are not identical, as the update equations demonstrate. See Table 2 for the definitions of the variables as they appear here. The equations used here are discrete updates. A more biologically plausible (gradient ascent) scheme is used in the simulations. These simply replace the updates with differential equations that have stationary points corresponding to the variational solutions above. Because the belief updates specified in Fig. 3 take each belief distribution to its free energy minimum, the belief updates and corresponding policy choices necessarily minimise free energy. In the update equations shown here, oτ is treated as a binary vector with one in the element corresponding to the observed data, and zero for all other elements. This ensures consistency with the linear algebraic expression of the update equations (color figure online) Full size image Summary In the above, we have provided an overview of our approach to date. This uses a variational free energy functional to derive belief updates, while policy selection is performed based on an expected free energy. The resulting update equations are shown in Fig. 3 (blue panels). This formulation has been very successful in explaining a range of cognitive functions, as summarised in Table 1. In the following, we present an alternative line of reasoning. As indicated in Fig. 2, there is more than one way to think about the data assimilation and evidence accumulation implicit in this formulation. So far, we have considered the addition of new observations as time progresses. We now consider the case in which (future) outcomes are represented throughout time. This means that future or latent outcomes have the potential to influence beliefs about past states. Active inference and generalised free energy We define the generalised free energy as =𝐸𝑄(𝜋)[𝜋]+𝐷𝐾𝐿[𝑄(𝜋)||𝑃(𝜋)]𝜋=∑𝜏𝜋𝜏𝜋𝜏=−𝐸𝑄(𝑜𝜏,𝑠𝜏|𝜋)[ln𝑃(𝑜𝜏,𝑠𝜏|𝑠𝜏−1,𝜋)Energy−ln𝑄(𝑜𝜏|𝜋)−ln𝑄(𝑠𝜏|𝜋)Entropy] F = E Q ( π ) [ F π ]+ D K L [Q(π) | | P(π)] F π = ∑ τ F π τ F π τ =− E Q ( o τ , s τ | π ) [ ln P ( o τ , s τ | s τ − 1 , π ) ⏟ Energy − ln Q ( o τ | π ) − ln Q ( s τ | π ) ⏟ Entropy ] (14) where, as above, the expectation is with respect to 𝑄(𝑜𝜏,𝑠𝜏|𝜋)=𝑄(𝑜𝜏|𝑠𝜏)𝑄(𝑠𝜏|𝜋) Q ( o τ , s τ | π ) = Q ( o τ | s τ ) Q ( s τ | π ) . However, we now distinguish the past and the future through the following: 𝑄(𝑜𝜏|𝑠𝜏)={𝑃(𝑜𝜏|𝑠𝜏)𝛿(𝑜𝜏,𝑜∗𝜏):𝜏>𝑡:𝜏≤𝑡 Q ( o τ | s τ ) = { P ( o τ | s τ ) : τ > t δ ( o τ , o τ ∗ ) : τ ≤ t (15) The δ here is a Kronecker delta function (a discrete version of a Dirac delta) that is one when the arguments are equal, and zero otherwise. The starred (*) argument indicates the data we have actually observed. In the generalised free energy, the marginals of the joint distribution over outcomes and states define the entropy but the expectation is over the joint distribution. It is important to note that 𝑄(𝑜𝜏,𝑠𝜏|𝜋)≠𝑄(𝑜𝜏|𝜋)𝑄(𝑠𝜏|𝜋) Q ( o τ , s τ | π ) ≠ Q ( o τ | π ) Q ( s τ | π ) . It is this inequality that underlies the epistemic components of generalised free energy. Interestingly, if we assumed conditional independence between outcomes and hidden states, 𝑄(𝑜𝜏,𝑠𝜏|𝜋)=𝑄(𝑜𝜏|𝜋)𝑄(𝑠𝜏|𝜋) Q ( o τ , s τ | π ) = Q ( o τ | π ) Q ( s τ | π ) , the resulting belief update equations would correspond exactly to a variational message passing algorithm (Dauwels 2007) applied to a model with missing data. When the expectation is taken with respect to the approximate posteriors, the marginalisation implicit in this definition ensures that −𝐸𝑄(𝑜𝜏,𝑠𝜏|𝜋)[ln𝑄(𝑜𝜏|𝜋)]=∑𝑜𝜏,𝑠𝜏𝑄(𝑜𝜏,𝑠𝜏|𝜋)ln𝑄(𝑜𝜏|𝜋)=−∑𝑜𝜏𝑄(𝑜𝜏|𝜋)ln𝑄(𝑜𝜏|𝜋)=𝐻[𝑄(𝑜𝜏|𝜋)] − E Q ( o τ , s τ | π ) [lnQ( o τ | π)] = ∑ o τ , s τ Q ( o τ , s τ | π ) ln Q ( o τ | π ) =− ∑ o τ Q ( o τ | π ) ln Q ( o τ | π ) =H[Q( o τ | π)] (16) If we write out the generative model in full and substitute this (omitting constants) into Eq. 14, we can use the same implicit marginalisation to write: 𝜋𝜏=−𝐸𝑄(𝑜𝜏|𝑠𝜏)𝑄(𝑠𝜏|𝜋)[ln𝑃(𝑜𝜏|𝑠𝜏)]−𝐸𝑄(𝑠𝜏|𝜋)𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]+𝐸𝑄(𝑜𝜏|𝜋)[ln𝑄(𝑜𝜏|𝜋)]+𝐸𝑄(𝑠𝜏|𝜋)[ln𝑄(𝑠𝜏|𝜋)]−𝐸𝑄(𝑜𝜏|𝜋)[ln𝑃(𝑜𝜏)]𝑄(𝑜𝜏|𝜋)=𝐸𝑄(𝑠𝜏|𝜋)[𝑄(𝑜𝜏|𝑠𝜏)] F π τ =− E Q ( o τ | s τ ) Q ( s τ | π ) [lnP( o τ | s τ )]− E Q ( s τ | π ) Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)] + E Q ( o τ | π ) [lnQ ( o τ | π ) ]+ E Q ( s τ | π ) [lnQ ( s τ | π ) ]− E Q ( o τ | π ) [lnP( o τ )] Q ( o τ | π ) = E Q ( s τ | π ) [Q( o τ | s τ )] (17) The implicit generative model now incorporates a prior over observations. This means that the generative model is replaced with that shown on the right of Fig. 1: 𝑃(𝑜̃,𝑠̃,𝜋|𝑚)=𝑃(𝑜̃|𝑠̃,𝑚)𝑃(𝑠̃|𝜋)𝑃(𝜋)𝑃(𝑜̃|𝑠̃,𝑚)=1𝑍𝑃(𝑜̃|𝑠̃)𝑃(𝑜̃|𝑚)𝑍=∑𝑜̃𝑃(𝑜̃|𝑠̃)𝑃(𝑜̃|𝑚) P ( o ~ , s ~ , π | m ) =P( o ~ | s ~ ,m)P( s ~ | π)P(π) P ( o ~ | s ~ , m ) = 1 Z P( o ~ | s ~ )P( o ~ | m) Z= ∑ o ~ P ( o ~ | s ~ ) P ( o ~ | m ) (18) Here, we have defined the distribution over states and observations in terms of two independent factors, a likelihood and a prior over observations, i.e. preferred observations conditioned on the model. For simplicity, we will omit the explicit conditioning on 𝑚 m , so that 𝑃(𝑜̃|𝑚)=𝑃(𝑜̃) P ( o ~ | m ) = P ( o ~ ) . This quantity plays exactly the same role as that of the preferences in the formulation described in the previous section. However, while it has the same influence over policy selection, it can no longer be interpreted as model evidence. Instead, it is a policy-independent prior that contributes to the evidence. For past states, this distribution is flat. Crucially, this means the generalised free energy reduces to the variational free energy for outcomes that had been observed in the past. Separating out contributions from the past and the future, we are left with the following: 𝜋=∑𝜏≤𝑡𝐹𝜋𝜏+∑𝜏>𝑡𝜋𝜏 F π = ∑ τ ≤ t F π τ + ∑ τ > t G π τ (19) Unlike 𝐺 G (the expected free energy), G is the free energy of the expected future. We can rearrange Eq. 17 (for future states) in several ways that offer some intuition for the properties of the generalised free energy. 𝜋𝜏=𝐷𝐾𝐿[𝑄(𝑠𝜏|𝜋)||𝐸𝑄(𝑠𝜏−1|𝜋)[𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]]Complexity+𝐷𝐾𝐿[𝑄(𝑜𝜏|𝜋)||𝑃(𝑜𝜏)]Risk+𝐸𝑄(𝑠𝜏|𝜋)[𝐻[𝑃(𝑜𝜏|𝑠𝜏)]]Ambiguity=𝐷𝐾𝐿[𝑄(𝑠𝜏|𝜋)||𝐸𝑄(𝑠𝜏−1|𝜋)[𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]]Complexity−𝐷𝐾𝐿[𝑄(𝑜𝜏,𝑠𝜏|𝜋)||𝑄(𝑠𝜏|𝜋)𝑄(𝑜𝜏|𝜋)]Epistemicvalue(Mutualinformation)−𝐸𝑄(𝑜𝜏|𝜋)[ln𝑃(𝑜𝜏)]Extrinsicvalue G π τ = D K L [ Q ( s τ | π ) | | E Q ( s τ − 1 | π ) [ P ( s τ | s τ − 1 , π ) ] ] ⏟ Complexity + D K L [ Q ( o τ | π ) | | P ( o τ ) ] ⏟ Risk + E Q ( s τ | π ) [ H [ P ( o τ | s τ ) ] ] ⏟ Ambiguity = D K L [ Q ( s τ | π ) | | E Q ( s τ − 1 | π ) [ P ( s τ | s τ − 1 , π ) ] ] ⏟ Complexity − D K L [ Q ( o τ , s τ | π ) | | Q ( s τ | π ) Q ( o τ | π ) ] ⏟ Epistemic value ( Mutual information ) − E Q ( o τ | π ) [ ln P ( o τ ) ] ⏟ Extrinsic value (20) To obtain the mutual information term, we have used the relationship ln𝑃(𝑜𝜏|𝑠𝜏)=ln𝑄(𝑜𝜏|𝑠𝜏)=ln𝑄(𝑜𝜏,𝑠𝜏|𝜋)−ln𝑄(𝑠𝜏|𝜋) ln P ( o τ | s τ ) = ln Q ( o τ | s τ ) = ln Q ( o τ , s τ | π ) − ln Q ( s τ | π ) . The imperative to maximise the mutual information (Barlow 1961, 1974; Linsker 1990; Optican and Richmond 1987) can be interpreted as an epistemic drive (Denzler and Brown 2002). This is because policies that (are believed to) result in observations that are highly informative about the hidden states are associated with a lower generalised free energy. As a KL divergence is always greater than or equal to zero, the second equality indicates that the free energy of the expected future is an upper bound on expected surprise. To find the belief update equations for the policies, we take the variational derivative of the generalised free energy with respect to the posterior over policies and set the result to zero in the usual way: 𝛿𝛿𝑄(𝜋)𝛿𝛿𝑄(𝜋)=𝜋−ln𝑃(𝜋)+ln𝑄(𝜋)=0⇔𝑄(𝜋)=𝜎(ln𝑃(𝜋)−𝜋) δ F δ Q ( π ) = F π −lnP(π)+lnQ(π) δ F δ Q ( π ) =0⇔Q(π)=σ(lnP(π)− F π ) (21) At time 𝜏=0 τ = 0 , no observations have been made, and the distribution above becomes a prior. When this is the case, 𝜋=𝜋 F π = G π , so the prior over policies is: 𝑄𝑜(𝜋)=𝜎(ln𝑃(𝜋)−𝜏=0𝜋)=𝜎(ln𝑃(𝜋)−(𝜋)) Q o ( π ) = σ ( ln P ( π ) − F π τ = 0 ) = σ ( ln P ( π ) − G ( π ) ) If we take the variational derivative of Eq. 17 with respect to the hidden states: 𝛿𝜋𝛿𝑄(𝑠𝜏|𝜋)𝛿𝜋𝛿𝑄(𝑠𝜏|𝜋)=ln𝑄(𝑠𝜏|𝜋)−𝐸𝑃(𝑜𝜏|𝑠𝜏)[ln𝑃(𝑜𝜏|𝑠𝜏)]−𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]−𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)]+𝐸𝑃(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)−ln𝑃(𝑜𝜏)]=0⇔𝑄(𝑠𝜏|𝜋)=𝜎(𝐸𝑃(𝑜𝜏|𝑠𝜏)[ln𝑃(𝑜𝜏|𝑠𝜏)]+𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]+𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)]−𝐸𝑃(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)−ln𝑃(𝑜𝜏)]) δ F π δ Q ( s τ | π ) =lnQ( s τ | π)− E P ( o τ | s τ ) [lnP( o τ | s τ )] − E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)]− E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)] + E P ( o τ | s τ ) [lnQ ( o τ | π ) −lnP( o τ )] δ F π δ Q ( s τ | π ) =0⇔Q ( s τ | π ) =σ( E P ( o τ | s τ ) [lnP( o τ | s τ )]+ E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)] + E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)] − E P ( o τ | s τ ) [lnQ ( o τ | π ) −lnP( o τ )]) (22) The derivative of 𝐸𝑄(𝑜𝜏|𝜋)[ln𝑄(𝑜𝜏|𝜋)] E Q ( o τ | π ) [ ln Q ( o τ | π ) ] is a little complicated, so this is presented step by step in “Appendix B”. The hidden state update has a different interpretation in the past compared to the future: ∀𝜏≤𝑡:𝑄(𝑠𝜏|𝜋)=𝜎(ln𝑃(𝑜𝜏|𝑠𝜏)+𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]+𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)])∀𝜏>𝑡:𝑄(𝑠𝜏|𝜋)=𝜎(−𝐻[𝑃(𝑜𝜏|𝑠𝜏)]+𝐸𝑄(𝑠𝜏−1|𝜋)[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]+𝐸𝑄(𝑠𝜏+1|𝜋)[ln𝑃(𝑠𝜏+1|𝑠𝜏,𝜋)]−𝐸𝑃(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)−ln𝑃(𝑜𝜏)]) ∀τ≤t : Q ( s τ | π ) =σ(lnP( o τ | s τ )+ E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)]+ E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)]) ∀τ>t : Q ( s τ | π ) =σ(−H[P( o τ | s τ )]+ E Q ( s τ − 1 | π ) [lnP( s τ | s τ − 1 ,π)]+ E Q ( s τ + 1 | π ) [lnP( s τ + 1 | s τ ,π)] − E P ( o τ | s τ ) [lnQ ( o τ | π ) −lnP( o τ )]) (23) The final term for future beliefs implies that future states are considered more probable if they are expected to be similar to those that generate preferred outcomes. In other words, there is an optimistic distortion of beliefs about the trajectory into the future. Summary We have introduced a generalised free energy functional that is expressed as a functional of beliefs about data. The variational free energy can be seen as a special case of this generalised functional, when beliefs about outcomes collapse to delta functions. When we derive update equations (Fig. 3, pink panels) under this functional, the updates look very similar to those based on the variational free energy approach. An important difference between the two approaches is that we have now included the prior probability of outcomes in the generative model. This has no influence over beliefs about the past, but distorts beliefs about the future in an optimistic fashion. This formulation generalises not only the standard active inference formalism, but also active data selection or sensing approaches in machine learning (MacKay 1992) and computational neuroscience (Yang et al. 2016b). See “Appendix A” for a discussion of the relationship between these. Comparison of active inference under expected and generalised free energy The generalised free energy has the appeal that belief updating and policy selection both minimise the same objective function. In contrast, formulations of active inference to date have required two different quantities (the variational free energy and the expected free energy, respectively) to derive these processes. Although the form of belief updating is the same, the belief updates resulting from the use of a generalised free energy are different in subtle ways. In this section, we will explore these differences and show how generalised active inference reproduces the behaviours illustrated in our earlier papers. The notable differences between the updates are found in the policy prior, the treatment of outcomes and the future hidden state updates. The prior over policies is very similar in both formulations. The expected and generalised free energy (at 𝜏=0 τ = 0 ) differ only in that there is an additional complexity term in the latter. This has a negligible influence on behaviour, as the first action is performed after observations have been made at the first time step. At this point, the posterior belief about policies is identical, as the variational free energy supplies the missing complexity term. Although the priors are different, in both form and motivation, the posterior beliefs turn out to be computed identically. Any difference in these can be attributed to the quantities used to calculate them, namely the outcomes and the hidden states. Outcomes in the generalised formulation are represented explicitly as beliefs. This means that the prior over outcomes is incorporated explicitly in the generative model. There are two important consequences of this. The first is that the posterior beliefs about future outcomes (i.e. the probability of future outcomes given those already observed) can be derived in a parsimonious way, without the need to define additional prior distributions. The second is that hidden state beliefs in the future are biased towards these preferred outcomes. A prior belief about an outcome at a particular time point thus distorts the trajectory of hidden states at each time point reaching back to the present. In addition to this, beliefs about hidden states in the future acquire an ‘ambiguity’ term. This means that states associated with an imprecise mapping to sensory outcomes are believed less likely to be inferred. In summary, not only are belief trajectories drawn in optimistic directions, they also tend towards states that offer informative observations. To make the abstract considerations above a little more concrete, we have employed an established generative model that has previously been used to demonstrate epistemic (i.e. information seeking) behaviours under active inference (Friston et al. 2015). This is a T-maze task (Fig. 4), in which an agent decides between (temporally deep) policies. Temporal depth here refers to the depth of the planning horizon. A temporally deep policy is one that considers sequences of actions, as opposed to only the next action. In one arm, there is an unconditionedFootnote 2 (rewarding) stimulus. In another, there is no stimulus, and this condition is considered aversive. In the final arm, there is always an instructional or conditioned stimulus that indicates the arm that contains the reward. There are two possible contexts for the maze. The first is that where the unconditioned stimulus is in the left arm and the second where it is in the right arm. The starting location and the location of the conditioned stimulus are neither aversive nor rewarding. Under each of the schemes illustrated here, the degree to which a stimulus is rewarding is expressed in terms of the prior preference (i.e. C). In other words, we can think of reward as the log probability of a given observation. The more probable an outcome is considered to be, the more attractive it appears to be. This is because policies that do not lead to these outcomes violate prior beliefs and are unlikely to be selected a posteriori. Please see “Appendix A” (term 4) for an interpretation of this that appeals to expected utility theory and risk aversion. There is an important distinction here between schemes based upon Bellman optimality and the scheme on offer here. This is that active inference depends upon probabilistic beliefs and does not assume direct access to knowledge about states of the world. Practically, this means that the agent has no direct access to the hidden states, but must infer them based upon the (observable) outcomes. The importance of this is that the information gain associated with an exploratory behaviour can be quantified by the change in beliefs (or uncertainty reduction) that this behaviour facilitates. Fig. 4 figure4 T-maze simulation. The left part of this figure shows the structure of the generative model used to illustrate the behavioural consequences of each set of update equations. We have previously used this generative model to address exploration and exploitation in two-step tasks; further details of which can be found in Friston et al. (2015). In brief, an agent can find itself in one of four different locations and can move among these locations. Locations 2 and 3 are absorbing states, so the agent is not able to leave these locations once they have been visited. The initial location is always 1. Policies define the possible sequences of movements the agent can take throughout the trial. For all ten available policies, after the second action, the agent stays where it is. There are two possible contexts: the unconditioned stimulus (US) may be in the left or right arm of the maze. The context and location together give rise to observable outcomes. The first of these is the location, which is obtained through an identity mapping from the hidden state representing location. The second outcome is the cue that is observed. In location 1, a conditioned stimulus (CS) is observed, but there is a 50% chance of observing blue or green, regardless of the context, so this is uninformative (and ambiguous). Location 4 deterministically generates a CS based on the context, so visiting this location resolves uncertainty about the location of the US. The US observation is probabilistically dependent on the context. It is observed with a 90% chance in the left arm in context 1 and a 90% chance in the right arm in context 2. The right part of this figure compares an agent that minimises its variational free energy (under the prior belief that it will select policies with a low expected free energy) with an agent that minimises its generalised free energy. The upper plots show the posterior beliefs about policies, where darker shades indicate more probable policies. Below these, the posterior beliefs about states (location and context) are shown, with blue dots superimposed to show the true states used to generate the data. The lower plots show the prior beliefs about outcomes (i.e. preferences), and the true outcomes (blue dots) the agent encountered. Note that a US is preferred to either CS, both of which are preferable to no stimulus (NS). Outcomes are observed at each time step, depending upon actions selected at the previous step. The time steps shown here align with the sequence of events during a trial, such that a new outcome is available at each step. Actions induce transitions from one time step to the next (color figure online) Full size image As Fig. 4 shows, regardless of the active inference scheme we use, the agent first samples the unrewarding, but epistemically valuable, uncertainty resolving cue location. This entails moving from the initial location in the centre of the maze, where the agent is uncertain about the context, to the location with the conditioned stimulus. To have made the decision to make this move, the agent updated its beliefs about states of the world (sπτ) in relation to the outcomes (o1) available in the central location using the fixed-point solutions shown in the ‘hidden states’ panels of Fig. 3. It does so for beliefs about every time point from the start to the end of the (four step) planning horizon. As these belief updates were derived by finding the free energy minima, this means these belief updates necessarily minimise free energy. Once beliefs have been optimised, they may be used to compute the expected free energy (or the corresponding part of the generalised free energy) as in the ‘free energies’ panel of Fig. 3. These are then used to update beliefs about policies as in the ‘policy selection’ panel. In computing these free energies, we required a posterior predictive belief about outcomes, which can be obtained using the likelihood probabilities to project beliefs about states to beliefs about outcomes (‘outcomes’ panel of Fig. 3). Given that the context unambiguously determines the conditioned stimulus and that our agent is initially uncertain about the context, the greatest information gain (and therefore smallest expected or generalised free energy) is associated with policies that sample this cue location. On reaching the conditioned stimulus and observing the green conditioned stimulus (o2), the agent again updates beliefs about states to their new fixed point. Here, the free energy minimum corresponds to the belief that the second context (with the unconditioned stimulus in the right arm) is in play. Having resolved uncertainty about the context of the maze, the agent proceeds to maximise its extrinsic reward by moving to the reward location and finding the unconditioned stimulus (o3). This is consistent with the smaller expected and generalised free energies associated with policies that realise prior beliefs about outcomes (C in the ‘free energies’ panels of Fig. 3). Although the most striking feature of these simulation results is their similarity, there are some interesting differences worth considering. These are primarily revealed by the beliefs about hidden states over time. Under each of the schemes presented here, for a hypothetical rat performing this task, there exist a set of (neuronal) units that encode beliefs about each possible state. For each state, there are units representing the configuration of that state in the past and future, in addition to the present. The activity in these units is shown in Fig. 5. The differences here are more dramatic than in the subsequent behaviours illustrated in Fig. 4. At the first time step (column 1), both agents infer that they will visit location 4 at the next time, resolving uncertainty about the context of the maze. From this future point onwards, however, the beliefs diverge. This can be seen clearly in the lower rows of column 1: the beliefs about the future at the first time step. The agent who employs expected free energy believes they will stay in the uncertainty resolving arm of the maze, while the generalised agent believes they will end up in one of the (potentially) rewarding arms. Despite a shared proximal belief trajectory, the distal elements of the two agents’ paths are pulled in opposite directions. As each future time point approaches, the beliefs about that time begin to converge—as observations become available. Fig. 5 figure5 Optimistic distortions of future beliefs. These raster plots represent the (Bayesian model average of the) approximate posterior beliefs about states (specifically, those pertaining to location). At each time step t, there is a set of units encoding beliefs about every other time step τ in the past and future. The evolution of these beliefs is reflected the evidence accumulation or belief updating of approximate posterior expectations, with lighter shades indicating more probable states Full size image Taken together, Figs. 4 and 5 illustrate an interesting feature of the generalised formulation. Although subtle, at t =1, beliefs about location at τ = 2 are different, as shown in Fig. 5. Specifically, locations 2 and 3 appear slightly more probable, at the expense of location 4. This illustrates that beliefs about the proximal future are distorted by beliefs about future outcomes. Similarly, at t =2, the generalised scheme considers it more likely that it will transition to location 4 relative to the variational scheme. Referring back to Fig. 4, we see that this corresponds to an increased posterior probability for policy 10 at this time step. Here, beliefs about future states and outcomes have influenced beliefs about the plausibility of different behavioural options at the present. In this case, the agent believes that it will experience observations associated with states 2 and 3 in the distal future (τ = 4). This enhances the probability of being in states in the more proximal future that are consistent with transitions into states 2 or 3. As these are absorbing states (the probability of staying in those states, once occupied, is one), these states are highly consistent with a transition to themselves. This induces a belief that states 2 and 3 are more probable at time τ = 3. Note that, as there are other plausible states that could have transitioned into 2 and 3 at τ = 4, the probability of states 2 and 3 at τ = 3 is less than at τ = 4. The same reasoning explains the higher probability of 2 and 3 at τ = 2 (relative to the standard scheme), but with a lower probability relative to occupying these states at later times. If instead the agent believed there was a very low probability of ending up at the goal location, this would induce beliefs that those states that lead to these locations with high probability were themselves unlikely. Another way of putting this is that if I had strong beliefs about where I were to end up, I could infer where I might have been immediately before this. This will depend upon the relative probabilities of going from plausible penultimate locations to the goal location. By propagating these back to the present, I will infer that the most probable trajectory is the one that leads to this goal, and will act to fulfil my beliefs about this trajectory. In the absence of, possibly false, beliefs about where I would end up, I would not end up acting to fulfil these beliefs. Conclusion The generalised free energy introduced in this paper provides a new perspective on active inference. It unifies the imperatives to minimise variational free energy with respect to data, and expected free energy through model selection, under a single objective function. Like the expected free energy, this generalised free energy can be decomposed in several ways, giving rise to familiar information theoretic measures and objective functions in Bayesian reinforcement learning. Generalised free energy minimisation replicates the epistemic and reward seeking behaviours induced in earlier active inference schemes, but prior preferences now induce an optimistic distortion of belief trajectories into the future. This allows beliefs about outcomes in the distal future to influence beliefs about states in the proximal future and present. That these beliefs then drive policy selection suggests that, under the generalised free energy formulation, (beliefs about) the future can indeed cause the past. Software note Although the generative model changes from application to application, the belief updates described in this paper are generic and can be implemented using standard routines (here spm_MDP_VB_X.m). These routines are available as MATLAB code in the SPM academic software: http://www.fil.ion.ucl.ac.uk/spm/. Simulations of the sort reported above can be reproduced (and customised) via a graphical user interface by typing in ≫ DEM and selecting the ‘+’ next to the ‘Habit learning’ button. Notes 1. The KL divergence (also known as relative entropy or information gain) is defined as follows: 𝐷KL[𝑄(𝑥)||𝑃(𝑥)]≜𝐸𝑄(𝑥)[ln𝑄(𝑥)−ln𝑃(𝑥)] D KL [ Q ( x ) | | P ( x ) ] ≜ E Q ( x ) [ ln Q ( x ) − ln P ( x ) ] . 2. The terms conditioned stimulus and unconditioned stimulus are used in the sense of classical (Pavlovian) conditioning paradigms. References Attias H (2003) Planning by probabilistic inference. In: Proceedings of the 9th international workshop on artificial intelligence and statistics Baker CL, Saxe R, Tenenbaum JB (2009) Action understanding as inverse planning. Cognition 113:329–349. https://doi.org/10.1016/j.cognition.2009.07.005 Article PubMed Google Scholar Barlow H (1961) Possible principles underlying the transformations of sensory messages. In: Rosenblith W (ed) Sensory communication. MIT Press, Cambridge, pp 217–234 Google Scholar Barlow HB (1974) Inductive inference, coding, perception, and language. Perception 3:123–134 CAS Article Google Scholar Beal MJ (2003) Variational algorithms for approximate Bayesian inference. University of London, London Google Scholar Benrimoh D, Parr T, Vincent P, Adams RA, Friston K (2018) Active inference and auditory hallucinations computational psychiatry 2:183–204. https://doi.org/10.1162/cpsy_a_00022 Article PubMed Google Scholar Botvinick M, Toussaint M (2012) Planning as inference. Trends Cognit Sci 16:485–488 Article Google Scholar Brown H, Friston KJ (2012) Free-energy and illusions: the Cornsweet effect. Front Psychol 3:43. https://doi.org/10.3389/fpsyg.2012.00043 Article PubMed PubMed Central Google Scholar Bruineberg J, Kiverstein J, Rietveld E (2016) The anticipating brain is not a scientist: the free-energy principle from an ecological-enactive perspective. Synthese. https://doi.org/10.1007/s11229-016-1239-1 Article PubMed PubMed Central Google Scholar Bruineberg J, Rietveld E, Parr T, van Maanen L, Friston KJ (2018) Free-energy minimization in joint agent-environment systems: a niche construction perspective. J Theor Biol 455:161–178. https://doi.org/10.1016/j.jtbi.2018.07.002 Article PubMed PubMed Central Google Scholar Daunizeau J, Preuschoff K, Friston K, Stephan K (2011) Optimizing experimental design for comparing models of brain function. PLOS Comput Biol 7:e1002280. https://doi.org/10.1371/journal.pcbi.1002280 CAS Article PubMed PubMed Central Google Scholar Dauwels J (2007) On variational message passing on factor graphs. In: IEEE international symposium on information theory, ISIT 2007. IEEE, pp 2546–2550 Dayan P, Hinton GE, Neal RM, Zemel RS (1995) The Helmholtz machine. Neural Comput 7:889–904 CAS Article Google Scholar Denzler J, Brown CM (2002) Information theoretic sensor data selection for active object recognition and state estimation. IEEE Trans Pattern Anal Mach Intell 24:145–157. https://doi.org/10.1109/34.982896 Article Google Scholar El-Gamal MA (1991) The role of priors in active bayesian learning in the sequential statistical decision framework. In: Grandy WT, Schick LH (eds) Maximum entropy and Bayesian methods: Laramie, Wyoming, 1990. Springer Netherlands, Dordrecht, pp 33–38. https://doi.org/10.1007/978-94-011-3460-6_3 Chapter Google Scholar Ellsberg D (1961) Risk, ambiguity, and the savage axioms. Q J Econ 75:643–669. https://doi.org/10.2307/1884324 Article Google Scholar FitzGerald T, Dolan R, Friston K (2014) Model averaging, optimal inference, and habit formation. Front Hum Neurosci. https://doi.org/10.3389/fnhum.2014.00457 Article PubMed PubMed Central Google Scholar FitzGerald TH, Dolan RJ, Friston K (2015a) Dopamine, reward learning, and active inference. Front Comput Neurosci 9:136. https://doi.org/10.3389/fncom.2015.00136 Article PubMed PubMed Central Google Scholar FitzGerald TH, Moran RJ, Friston KJ, Dolan RJ (2015b) Precision and neuronal dynamics in the human posterior parietal cortex during evidence accumulation. Neuroimage 107:219–228. https://doi.org/10.1016/j.neuroimage.2014.12.015 Article PubMed PubMed Central Google Scholar FitzGerald TH, Schwartenbeck P, Moutoussis M, Dolan RJ, Friston K (2015c) Active inference, evidence accumulation, and the urn task. Neural Comput 27:306–328. https://doi.org/10.1162/neco_a_00699 Article PubMed Google Scholar Friston K (2003) Learning and inference in the brain. Neural Netw 16:1325–1352. https://doi.org/10.1016/j.neunet.2003.06.005 Article PubMed Google Scholar Friston K, Buzsaki G (2016) The functional anatomy of time: what and when in the brain. Trends Cognit Sci. https://doi.org/10.1016/j.tics.2016.05.001 Article Google Scholar Friston K, Kilner J, Harrison L (2006) A free energy principle for the brain. J Physiol-Paris 100:70–87. https://doi.org/10.1016/j.jphysparis.2006.10.001 Article PubMed Google Scholar Friston K, Adams R, Montague R (2012a) What is value—accumulated reward or evidence? Front Neurorobotics 6:11. https://doi.org/10.3389/fnbot.2012.00011 Article Google Scholar Friston K, Adams RA, Perrinet L, Breakspear M (2012b) Perceptions as hypotheses: saccades as experiments. Front Psychol 3:151. https://doi.org/10.3389/fpsyg.2012.00151 Article PubMed PubMed Central Google Scholar Friston K, Samothrakis S, Montague R (2012c) Active inference and agency: optimal control without cost functions. Biol Cybernet 106:523–541. https://doi.org/10.1007/s00422-012-0512-8 Article Google Scholar Friston K, Schwartenbeck P, FitzGerald T, Moutoussis M, Behrens T, Dolan RJ (2014) The anatomy of choice: dopamine and decision-making. Philos Trans R Soc B Biol Sci 369:20130481. https://doi.org/10.1098/rstb.2013.0481 Article Google Scholar Friston K, Rigoli F, Ognibene D, Mathys C, Fitzgerald T, Pezzulo G (2015) Active inference and epistemic value. Cognit Neurosci 6:187–214. https://doi.org/10.1080/17588928.2015.1020053 Article Google Scholar Friston K, FitzGerald T, Rigoli F, Schwartenbeck P, O’Doherty J, Pezzulo G (2016) Active inference and learning. Neurosci Biobehav Rev 68:862–879. https://doi.org/10.1016/j.neubiorev.2016.06.022 Article PubMed PubMed Central Google Scholar Friston K, FitzGerald T, Rigoli F, Schwartenbeck P, Pezzulo G (2017a) Active inference: a process theory. Neural Comput 29:1–49. https://doi.org/10.1162/NECO_a_00912 Article PubMed Google Scholar Friston KJ, Lin M, Frith CD, Pezzulo G, Hobson JA, Ondobaka S (2017b) Active inference, curiosity and insight. Neural Comput 29(10):2633–2683 Article Google Scholar Friston KJ, Parr T, de Vries B (2017c) The graphical brain: belief propagation and active inference. Netw Neurosci 1:381–414. https://doi.org/10.1162/NETN_a_00018 Article PubMed PubMed Central Google Scholar Friston KJ, Rosch R, Parr T, Price C, Bowman H (2017d) Deep temporal models and active inference. Neurosci Biobehav Rev 77:388–402. https://doi.org/10.1016/j.neubiorev.2017.04.009 Article PubMed PubMed Central Google Scholar Ghirardato P, Marinacci M (2002) Ambiguity made precise: a comparative foundation. J Econ Theory 102:251–289. https://doi.org/10.1006/jeth.2001.2815 Article Google Scholar Gilhooly (2005) Working memory and planning, 1st edn. In: Morris R, Ward G (eds) The cognitive psychology of planning. Psychology Press, London, 256 p. https://doi.org/10.4324/9780203493564 Book Google Scholar Gregory RL (1980) Perceptions as hypotheses. Philos Trans R Soc Lond B Biol Sci 290:181 CAS Article Google Scholar Hikosaka O, Takikawa Y, Kawagoe R (2000) Role of the basal ganglia in the control of purposive saccadic eye movements. Physiol Rev 80:953 CAS Article Google Scholar Hohwy J (2016) The self-evidencing brain. Noûs 50:259–285. https://doi.org/10.1111/nous.12062 Article Google Scholar Huggins JH, Tenenbaum JB (2015) Risk and regret of hierarchical Bayesian learners. Paper presented at the Proceedings of the 32nd international conference on international conference on machine learning—Volume 37, Lille, France Hwa R (2004) Sample selection for statistical parsing. Comput Linguist 30:253–276 Article Google Scholar Kaplan R, Friston KJ (2018) Planning and navigation as active inference. Biol Cybernet. https://doi.org/10.1007/s00422-018-0753-2 Article Google Scholar Kappen HJ, Gomez Y, Opper M (2012) Optimal control as a graphical model inference problem. Mach Learn 87:159–182 Article Google Scholar Krause A (2008) Optimizing sensing: theory and applications. Carnegie Mellon University, Pittsburgh Google Scholar Lewis DD, Gale WA (1994) A sequential algorithm for training text classifiers. In: Proceedings of the 17th annual international ACM SIGIR conference on Research and development in information retrieval. Springer-Verlag New York, Inc., pp 3–12 Lindley DV (1956) On a measure of the information provided by an experiment. Ann Math Stat 27:986–1005. https://doi.org/10.1214/aoms/1177728069 Article Google Scholar Linsker R (1990) Perceptual neural organization: some approaches based on network models and information theory. Annu Rev Neurosci 13:257–281 CAS Article Google Scholar Lloyd K, Leslie DS (2013) Context-dependent decision-making: a simple Bayesian model. J R Soc Interface 10:1. https://doi.org/10.1098/rsif.2013.0069 Article Google Scholar MacKay DJC (1992) Information-based objective functions for active data selection. Neural Comput 4:590–604. https://doi.org/10.1162/neco.1992.4.4.590 Article Google Scholar McKay RT, Dennett DC (2010) The evolution of misbelief. Behav Brain Sci 32:493–510. https://doi.org/10.1017/S0140525X09990975 Article Google Scholar Mirza MB, Adams RA, Mathys CD, Friston KJ (2016) Scene construction, visual foraging, and active inference. Front Comput Neurosci. https://doi.org/10.3389/fncom.2016.00056 Article PubMed PubMed Central Google Scholar Moiseiwitsch BL (2013) Variational principles. Dover Publications, Mineola Google Scholar Moutoussis M, Trujillo-Barreto NJ, El-Deredy W, Dolan RJ, Friston KJ (2014) A formal model of interpersonal inference. Front Hum Neurosci 8:160. https://doi.org/10.3389/fnhum.2014.00160 Article PubMed PubMed Central Google Scholar Optican L, Richmond BJ (1987) Temporal encoding of two-dimensional patterns by single units in primate inferior cortex. II Information theoretic analysis. J Neurophysiol 57:132–146 Article Google Scholar Ortega PA, Braun DA (2010) A minimum relative entropy principle for learning and acting. J Artif Int Res 38:475–511 Google Scholar Parr T, Friston KJ (2017a) The computational anatomy of visual neglect. Cereb Cortex. https://doi.org/10.1093/cercor/bhx316 Article PubMed Central Google Scholar Parr T, Friston KJ (2017b) Uncertainty, epistemics and active inference. J R Soc Interface 14:20170376 Article Google Scholar Parr T, Friston KJ (2017c) Working memory, attention, and salience in active inference. Sci Rep 7:14678. https://doi.org/10.1038/s41598-017-15249-0 CAS Article PubMed PubMed Central Google Scholar Parr T, Friston KJ (2018) The discrete and continuous brain: from decisions to movement—and back again. Neural Comput 30:1–10 Article Google Scholar Parr T, Friston KJ (2019) The computational pharmacology of oculomotion. Psychopharmacology. https://doi.org/10.1007/s00213-019-05240-0 Article PubMed PubMed Central Google Scholar Parr T, Benrimoh D, Vincent P, Friston K (2018a) Precision and false perceptual inference. Front Integr Neurosci. https://doi.org/10.3389/fnint.2018.00039 Article PubMed PubMed Central Google Scholar Parr T, Rees G, Friston KJ (2018b) Computational neuropsychology and Bayesian inference. Front Hum Neurosci. https://doi.org/10.3389/fnhum.2018.00061 Article PubMed PubMed Central Google Scholar Parr T, Rikhye RV, Halassa MM, Friston KJ (2019) Prefrontal computation as active inference. Cereb Cortex. https://doi.org/10.1093/cercor/bhz118 Article PubMed Google Scholar Pearl J (1998) Graphical models for probabilistic and causal reasoning. In: Smets P (ed) Quantified representation of uncertainty and imprecision. Springer Netherlands, Dordrecht, pp 367–389. https://doi.org/10.1007/978-94-017-1735-9_12 Chapter Google Scholar Pearl J (2014) Probabilistic reasoning in intelligent systems: networks of plausible inference. Elsevier, Amsterdam Google Scholar Prosser A, Friston KJ, Bakker N, Parr T (2018) A Bayesian account of psychopathy: a model of lacks remorse and self-aggrandizing. Comput Psychiatry. https://doi.org/10.1162/cpsy_a_00016 Article Google Scholar Rasmussen CE, Ghahramani Z (2001) Occam's razor, advances in neural information processing systems 13. In: Leen TK, Dietterich TG, Tresp V (eds) Proceedings from the conference, neural information processing systems. https://papers.nips.cc/book/advances-in-neural-information-processing-systems-13-2000 Sales AC, Friston KJ, Jones MW, Pickering AE, Moran RJ (2018) Locus Coeruleus tracking of prediction errors optimises cognitive flexibility: an Active Inference model bioRxiv:340620 Schacter DL, Benoit RG, De Brigard F, Szpunar KK (2015) Episodic future thinking and episodic counterfactual thinking: intersections between memory and decisions. Neurobiol Learn Mem 117:14–21. https://doi.org/10.1016/j.nlm.2013.12.008 Article PubMed
Schwartenbeck P, FitzGerald TH, Mathys C, Dolan R, Friston K (2015a) The dopaminergic midbrain encodes the expected certainty about desired outcomes. Cereb Cortex 25:3434–3445. https://doi.org/10.1093/cercor/bhu159
Schwartenbeck P, FitzGerald TH, Mathys C, Dolan R, Kronbichler M, Friston K (2015b) Evidence for surprise minimization over value maximization in choice behavior. Sci Rep 5:16575. https://doi.org/10.10/srep16575
Article PubMed PubMed Central Google Scholar Schwartenbeck P, FitzGerald TH, Mathys C, Dolan R, Wurst F, Kronbichler M, Friston K (2015c) Optimal inference with suboptimal models: addiction and active Bayesian inference. Med Hypotheses 84:109–117. https://doi.org/10.1016/j.mehy.2014.12.007 Article PubMed PubMed Central Google Scholar Settles B (2010) Active learning literature survey, vol 52. University of Wisconsin, Madison, p 11 Google Scholar Sharot T (2011) The optimism bias. Curr Biol 21:R941–R945. https://doi.org/10.1016/j.cub.2011.10.030 CAS Article PubMed Google Scholar Sharot T, Guitart-Masip M, Korn Christoph W, Chowdhury R, Dolan Raymond J (2012) How dopamine enhances an optimism bias in humans. Curr Biol 22:1477–1481. https://doi.org/10.1016/j.cub.2012.05.053 CAS Article PubMed PubMed Central Google Scholar Shewry MC, Wynn HP (1987) Maximum entropy sampling. J Appl Stat 14:165–170. https://doi.org/10.1080/02664768700000020 Article Google Scholar Strens MJA (2000) A Bayesian framework for reinforcement learning. Paper presented at the proceedings of the seventeenth international conference on machine learning Sutton RS, Barto AG (1998) Reinforcement learning: an introduction, vol 1. MIT Press, Cambridge Google Scholar Todorov E (2008) General duality between optimal control and estimation. In: IEEE conference on decision and control Verma D, Rao RP (2006) Planning and acting in uncertain environments using probabilistic inference. In: 2006 IEEE/RSJ international conference on intelligent robots and systems. IEEE, pp 2382–2387 Vincent P, Parr T, Benrimoh D, Friston KJ (2019) With an eye on uncertainty: Modelling pupillary responses to environmental volatility. PLOS Comput Biol 15:e1007126. https://doi.org/10.1371/journal.pcbi.1007126 Article PubMed PubMed Central Google Scholar Wald A (1947) An essentially complete class of admissible decision functions. Ann Math Stat 4:549–555. https://doi.org/10.1214/aoms/1177730345 Article Google Scholar Winn JM (2004) Variational message passing and its applications. Citeseer Winn J, Bishop CM (2005) Variational message passing. J Mach Learn Res 6:661–694 Google Scholar Yang SC-H, Lengyel M, Wolpert DM (2016a) Active sensing in the categorization of visual patterns. eLife 5:e12215. https://doi.org/10.7554/elife.12215 Article PubMed PubMed Central Google Scholar Yang SC-H, Wolpert DM, Lengyel M (2016b) Theoretical perspectives on active sensing. Curr Opin Behav Sci 11:100–108. https://doi.org/10.1016/j.cobeha.2016.06.009 CAS Article Google Scholar Yedidia JS, Freeman WT, Weiss Y (2005) Constructing free-energy approximations and generalized belief propagation algorithms. IEEE Trans Inf Theory 51:2282–2312 Article Google Scholar Download references Acknowledgements TP is supported by the Rosetrees Trust (Award Number 173346). KJF is a Wellcome Principal Research Fellow (Ref: 088130/Z/09/Z). The authors thank Dimitrije Markovic for his insightful comments on the manuscript. Author information Affiliations Wellcome Centre for Human Neuroimaging, Institute of Neurology, University College London, 12 Queen Square, London, WC1N 3BG, UK Thomas Parr & Karl J. Friston Corresponding author Correspondence to Thomas Parr. Ethics declarations Conflict of interest The authors declare that they have no conflict of interest. Additional information Publisher's Note Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations. Communicated by Benjamin Lindner. Appendices Appendix A: Active data selection Active data selection has been a topic of interest in both neuroscience and machine learning for a number of years (Krause 2008). Several different approaches have been taken to define the best data to sample (Settles 2010), and the optimal experiments to perform to do this (Daunizeau et al. 2011). This appendix addresses the relationship between the future components of the expected free energy and established methods. Writing in full, the (negative) free energy of the expected future is −𝐺𝜋𝜏=𝐻[𝑄(𝑠𝜏|𝜋)]1+𝐸𝑄[ln𝑃(𝑠𝜏|𝑠𝜏−1,𝜋)]2+𝐻[𝑄(𝑜𝜏|𝜋)]3+𝐸𝑄[ln𝑃(𝑜𝜏)]4−𝐸𝑄[𝐻[𝑃(𝑜𝜏|𝑠𝜏)]]5 − G π τ = H [ Q ( s τ | π ) ] ⏟ 1 + E Q [ ln P ( s τ | s τ − 1 , π ) ] ⏟ 2 + H [ Q ( o τ | π ) ] ⏟ 3 + E Q [ ln P ( o τ ) ] ⏟ 4 − E Q [ H [ P ( o τ | s τ ) ] ] ⏟ 5 Under active inference, the above functional is maximised. If we were to use only term 3, this maximisation reduces to ‘uncertainty sampling’ (Hwa 2004; Lewis and Gale 1994; Shewry and Wynn 1987). This involves (as the name suggests) selecting the data points about which uncertainty is highest. A problem with this approach is that it may favour the sampling of ambiguous (uninformative) data. This means that location 1 in our simulation would be very attractive, as there is always a 50% chance of observing each (uninformative) cue. A more sophisticated objective function includes both 3 and 5 (Denzler and Brown 2002; Lindley 1956; MacKay 1992; Yang et al. 2016a). This means that uncertain data points are more likely to be sampled, but only if there is an unambiguous mapping between the latent variable of interest and the data. This renders location 4 more attractive, as it initially associated with uncertain observations but also a precise likelihood mapping. If we were to use just term 5, location 4 would continue to be attractive even after being observed. The contribution of term 3 is to implement a form of ‘inhibition of return’. The relative influences of these terms are unpacked in greater detail in (Parr and Friston 2017b). Term 4 is a homologue of expected utility (reward) in reinforcement learning (Sutton and Barto 1998) and is an important quantity in sequential statistical decision theory (El-Gamal 1991; Wald 1947). On its own, this would not lead to any information seeking behaviour. Terms 1 and 2 together contribute to an ‘Occam factor’ (Rasmussen and Ghahramani 2001), a component of some previously used objective functions (MacKay 1992). We have assumed here that active learning models are myopic, but this is not necessarily the case. On inclusion of an explicit transition model, these models implicitly acquire terms 1 and 2 in the evaluation of posterior beliefs.
Many of these schemes rely upon exact, as opposed to approximate, Bayesian inference. The former can be seen as a special case of the latter, in which the distributions Q are equal to the true posteriors. In some cases, this will violate the mean-field assumption used here, and such cases will factorise in a slightly more complicated way. However, it is still possible to represent the terms above in exactly the same way, as long as we introduce an additional corrective (mutual information) term to account for pairwise interactions in the marginal posterior distributions. This is the approach taken in the Bethe free energy that underwrites exact inference procedures such as belief propagation (Pearl 2014; Yedidia et al. 2005). All of these quantities are emergent properties of a system that minimises its expected free energy. In the schemes mentioned above, the quantities were pragmatically selected to sample data efficiently. Here, they can be seen as special cases of the free energy functional used to define the active inference or sensing that underwrites perception (Friston et al. 2012b; Gregory 1980). Appendix B: Variational derivative of expected marginal Below are the steps taken to obtain the variational derivative of an expected marginal. This is needed for the hidden state update equations under the generalised free energy. For those unfamiliar with variational calculus, “Appendix C” provides a brief introduction. 𝛿𝛿𝑄(𝑠𝜏|𝜋)𝐸𝑄(𝑜𝜏,𝑠𝜏|𝜋)[ln𝑄(𝑜𝜏|𝜋)]=𝐸𝑄(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)]+𝐸𝑄(𝑜𝜏,𝑠′𝜏|𝜋)[𝛿𝛿𝑄(𝑠𝜏|𝜋)ln𝐸𝑄(𝑠𝜏|𝜋)[𝑄(𝑜𝜏|𝑠𝜏)]]=𝐸𝑄(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)]+𝐸𝑄(𝑜𝜏,𝑠′𝜏|𝜋)[𝑄(𝑜𝜏|𝑠𝜏)𝐸𝑄(𝑠𝜏|𝜋)[𝑄(𝑜𝜏|𝑠𝜏)]]=𝐸𝑄(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)]+∑𝑜𝜏[∑𝑠′𝜏𝑄(𝑜𝜏|𝑠′𝜏)𝑄(𝑠′𝜏|𝜋)∑𝑠𝜏𝑄(𝑜𝜏|𝑠𝜏)𝑄(𝑠𝜏|𝜋)𝑄(𝑜𝜏|𝑠𝜏)]=𝐸𝑄(𝑜𝜏|𝑠𝜏)[ln𝑄(𝑜𝜏|𝜋)]+1 δ δ Q ( s τ | π ) E Q ( o τ , s τ | π ) [lnQ( o τ | π)] = E Q ( o τ | s τ ) [lnQ( o τ | π)]+ E Q ( o τ , s τ ′ | π ) [ δ δ Q ( s τ | π ) ln E Q ( s τ | π ) [ Q ( o τ | s τ ) ] ] = E Q ( o τ | s τ ) [lnQ( o τ | π)]+ E Q ( o τ , s τ ′ | π ) [ Q ( o τ | s τ ) E Q ( s τ | π ) [ Q ( o τ | s τ ) ] ] = E Q ( o τ | s τ ) [lnQ( o τ | π)]+ ∑ o τ [ ∑ s τ ′ Q ( o τ | s τ ′ ) Q ( s τ ′ | π ) ∑ s τ Q ( o τ | s τ ) Q ( s τ | π ) Q ( o τ | s τ ) ] = E Q ( o τ | s τ ) [lnQ( o τ | π)]+1 In the update equations, we can omit the constant 1. Appendix C: A primer on variational calculus This appendix offers a brief introduction to variational calculus, with a focus on the notion of a functional (or variational) derivative. Variational calculus deals with the problem of finding a function (f) that extremises a functional (a function of a function). Formally, this means we try to find: 𝑓̃(𝑥)=argmin𝑓𝑆[𝑓(𝑥)]𝑆[𝑓(𝑥)]≜∑𝑥(𝑓(𝑥),𝑥) f ~ (x)= arg min f S[f(x)] S[f(x)]≜ ∑ x L ( f ( x ) , x ) We can solve this problem by parameterising our function in terms of a second (arbitrary) function (g) and a scalar (u): 𝑓(𝑥,𝑢)∂∂𝑢𝑆[𝑓(𝑥,𝑢)]≜𝑓̃(𝑥)+𝑢𝑔(𝑥)=∑𝑥∂∂𝑢(𝑓(𝑥,𝑢),𝑥)=∑𝑥∂𝑓(𝑥,𝑢)∂𝑢∂∂𝑓(𝑓(𝑥,𝑢),𝑥)=∑𝑥𝑔(𝑥)∂∂𝑓(𝑓(𝑥,𝑢),𝑥) f(x,u) ≜ f ~ (x)+ug(x) ∂ ∂ u S[f(x,u)] = ∑ x ∂ ∂ u L ( f ( x , u ) , x ) = ∑ x ∂ f ( x , u ) ∂ u ∂ ∂ f L ( f ( x , u ) , x ) = ∑ x g ( x ) ∂ ∂ f L ( f ( x , u ) , x ) When u is zero, 𝑓=𝑓̃ f = f ~ and the functional is minimised. This implies: ∂∂𝑢𝑆[𝑓(𝑥,𝑢)]∣∣∣𝑢=0=0=∑𝑥𝑔(𝑥)∂∂𝑓(𝑓(𝑥),𝑥) ∂ ∂ u S [ f ( x , u ) ] | u = 0 = 0 = ∑ x g ( x ) ∂ ∂ f L ( f ( x ) , x ) As g may be any arbitrary function, the only way that the expression above will always hold is if the partial derivative of L with respect to f is zero for all x. This tells us (using δ to indicate a variational derivative with respect to a function): 𝑓̃(𝑥)=argmin𝑓𝑆[𝑓(𝑥)]⇔𝛿𝑆𝛿𝑓∣∣∣𝑓=𝑓̃=0⇔∂∂𝑓∣∣∣𝑓=𝑓̃=0 f ~ ( x ) = arg min f S [ f ( x ) ] ⇔ δ S δ f | f = f ~ = 0 ⇔ ∂ L ∂ f | f = f ~ = 0 Note that had we assumed that L was also a function of the gradient of f, we would instead have recovered the Euler–Lagrange equation used in analytical mechanics, for which L is referred to as a Lagrangian. In the main text of this paper, the functional (S) is typically a free energy, the function (f) is an approximate posterior distribution, and x are hidden states. For example: (𝑄(𝑠̃|𝜋),𝑠̃)=𝑄(𝑠̃|𝜋)(ln𝑄(𝑠̃|𝜋)−ln𝑃(𝑜̃,𝑠̃|𝜋))𝑆=𝐹𝜋=∑𝑠̃(𝑄(𝑠̃|𝜋),𝑠̃)𝑃(𝑠̃|𝜋,𝑜̃)≈argmin𝑄(𝑠̃|𝜋)𝐹𝜋
This says that, to minimise a free energy functional with respect to a probability distribution, we need to find the point at which the partial derivative of the term inside the sum, with respect to this distribution, is zero. We hope that this appendix is sufficient for readers not familiar with this style of mathematics to gain some intuition for the variational derivatives used throughout this paper. For interested readers, a more comprehensive introduction to this field can be found in (Moiseiwitsch 2013). For applications in the context of variational inference, please see (Beal 2003). Rights and permissions Open Access This article is distributed under the terms of the Creative Commons Attribution 4.0 International License (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. Reprints and Permissions About this article Verify currency and authenticity via CrossMark Cite this article Parr, T., Friston, K.J. Generalised free energy and active inference. Biol Cybern 113, 495–513 (2019). https://doi.org/10.1007/s00422-019-00805-w Download citation Received 04 July 2017 Accepted 13 September 2019 Published 27 September 2019 Issue Date December 2019 DOI https://doi.org/10.1007/s00422-019-00805-w Share this article Anyone you share the following link with will be able to read this content: Get shareable link Provided by the Springer Nature SharedIt content-sharing initiative Keywords Bayesian Active inference Free energy Data selection Epistemic value Intrinsic motivation Download PDF