Partially observable Markov decision process

From HandWiki
Short description: Generalization of a Markov decision process

A partially observable Markov decision process (POMDP) is a generalization of a Markov decision process (MDP). A POMDP models an agent decision process in which it is assumed that the system dynamics are determined by an MDP, but the agent cannot directly observe the underlying state. Instead, it must maintain a sensor model (the probability distribution of different observations given the underlying state) and the underlying MDP. Unlike the policy function in MDP which maps the underlying states to the actions, POMDP's policy is a mapping from the history of observations (or belief states) to the actions.

The POMDP framework is general enough to model a variety of real-world sequential decision processes. Applications include robot navigation problems, machine maintenance, and planning under uncertainty in general. The general framework of Markov decision processes with imperfect information was described by Karl Johan Åström in 1965[1] in the case of a discrete state space, and it was further studied in the operations research community where the acronym POMDP was coined. It was later adapted for problems in artificial intelligence and automated planning by Leslie P. Kaelbling and Michael L. Littman.[2]

An exact solution to a POMDP yields the optimal action for each possible belief over the world states. The optimal action maximizes the expected reward (or minimizes the cost) of the agent over a possibly infinite horizon. The sequence of optimal actions is known as the optimal policy of the agent for interacting with its environment.

Definition

Formal definition

A discrete-time POMDP models the relationship between an agent and its environment. Formally, a POMDP is a 7-tuple [math]\displaystyle{ (S,A,T,R,\Omega,O,\gamma) }[/math], where

  • [math]\displaystyle{ S }[/math] is a set of states,
  • [math]\displaystyle{ A }[/math] is a set of actions,
  • [math]\displaystyle{ T }[/math] is a set of conditional transition probabilities between states,
  • [math]\displaystyle{ R: S \times A \to \mathbb{R} }[/math] is the reward function.
  • [math]\displaystyle{ \Omega }[/math] is a set of observations,
  • [math]\displaystyle{ O }[/math] is a set of conditional observation probabilities, and
  • [math]\displaystyle{ \gamma \in [0, 1) }[/math] is the discount factor.

At each time period, the environment is in some state [math]\displaystyle{ s \in S }[/math]. The agent takes an action [math]\displaystyle{ a \in A }[/math], which causes the environment to transition to state [math]\displaystyle{ s' }[/math] with probability [math]\displaystyle{ T(s'\mid s,a) }[/math]. At the same time, the agent receives an observation [math]\displaystyle{ o \in \Omega }[/math] which depends on the new state of the environment, [math]\displaystyle{ s' }[/math], and on the just taken action, [math]\displaystyle{ a }[/math], with probability [math]\displaystyle{ O(o \mid s',a) }[/math] (or sometimes [math]\displaystyle{ O(o \mid s') }[/math] depending on the sensor model). Finally, the agent receives a reward [math]\displaystyle{ r }[/math] equal to [math]\displaystyle{ R(s, a) }[/math]. Then the process repeats. The goal is for the agent to choose actions at each time step that maximize its expected future discounted reward: [math]\displaystyle{ E \left[ \sum_{t=0}^\infty \gamma^t r_t \right] }[/math], where [math]\displaystyle{ r_t }[/math] is the reward earned at time [math]\displaystyle{ t }[/math]. The discount factor [math]\displaystyle{ \gamma }[/math] determines how much immediate rewards are favored over more distant rewards. When [math]\displaystyle{ \gamma=0 }[/math] the agent only cares about which action will yield the largest expected immediate reward; when [math]\displaystyle{ \gamma \rightarrow 1 }[/math] the agent cares about maximizing the expected sum of future rewards.

Discussion

Because the agent does not directly observe the environment's state, the agent must make decisions under uncertainty of the true environment state. However, by interacting with the environment and receiving observations, the agent may update its belief in the true state by updating the probability distribution of the current state. A consequence of this property is that the optimal behavior may often include (information gathering) actions that are taken purely because they improve the agent's estimate of the current state, thereby allowing it to make better decisions in the future.

It is instructive to compare the above definition with the definition of a Markov decision process. An MDP does not include the observation set, because the agent always knows with certainty the environment's current state. Alternatively, an MDP can be reformulated as a POMDP by setting the observation set to be equal to the set of states and defining the observation conditional probabilities to deterministically select the observation that corresponds to the true state.

Belief update

After having taken the action [math]\displaystyle{ a }[/math] and observing [math]\displaystyle{ o }[/math], an agent needs to update its belief in the state the environment may (or not) be in. Since the state is Markovian (by assumption), maintaining a belief over the states solely requires knowledge of the previous belief state, the action taken, and the current observation. The operation is denoted [math]\displaystyle{ b' = \tau(b,a,o) }[/math]. Below we describe how this belief update is computed.

After reaching [math]\displaystyle{ s' }[/math], the agent observes [math]\displaystyle{ o \in \Omega }[/math] with probability [math]\displaystyle{ O(o\mid s',a) }[/math]. Let [math]\displaystyle{ b }[/math] be a probability distribution over the state space [math]\displaystyle{ S }[/math]. [math]\displaystyle{ b(s) }[/math] denotes the probability that the environment is in state [math]\displaystyle{ s }[/math]. Given [math]\displaystyle{ b(s) }[/math], then after taking action [math]\displaystyle{ a }[/math] and observing [math]\displaystyle{ o }[/math],

[math]\displaystyle{ b'(s') = \eta O(o\mid s',a) \sum_{s\in S} T(s'\mid s,a)b(s) }[/math]

where [math]\displaystyle{ \eta=1/\Pr(o\mid b,a) }[/math] is a normalizing constant with [math]\displaystyle{ \Pr(o\mid b,a) = \sum_{s'\in S}O(o\mid s',a)\sum_{s\in S}T(s'\mid s,a)b(s) }[/math].

Belief MDP

A Markovian belief state allows a POMDP to be formulated as a Markov decision process where every belief is a state. The resulting belief MDP will thus be defined on a continuous state space (even if the "originating" POMDP has a finite number of states: there are infinite belief states (in [math]\displaystyle{ B }[/math]) because there are an infinite number of probability distributions over the states (of [math]\displaystyle{ S }[/math])).[2]

Formally, the belief MDP is defined as a tuple [math]\displaystyle{ (B,A,\tau,r,\gamma) }[/math] where

  • [math]\displaystyle{ B }[/math] is the set of belief states over the POMDP states,
  • [math]\displaystyle{ A }[/math] is the same finite set of action as for the original POMDP,
  • [math]\displaystyle{ \tau }[/math] is the belief state transition function,
  • [math]\displaystyle{ r:B \times A \to \mathbb{R} }[/math] is the reward function on belief states,
  • [math]\displaystyle{ \gamma }[/math] is the discount factor equal to the [math]\displaystyle{ \gamma }[/math] in the original POMDP.

Of these, [math]\displaystyle{ \tau }[/math] and [math]\displaystyle{ r }[/math] need to be derived from the original POMDP. [math]\displaystyle{ \tau }[/math] is

[math]\displaystyle{ \tau(b,a,b') = \sum_{o\in \Omega} \Pr(b'|b,a,o) \Pr(o | a, b), }[/math]

where [math]\displaystyle{ \Pr(o | a,b) }[/math] is the value derived in the previous section and

[math]\displaystyle{ Pr(b'|b,a,o) = \begin{cases} 1 &\text{if the belief update with arguments } b,a,o \text{ returns } b' \\ 0 &\text{otherwise } \end{cases}. }[/math]

The belief MDP reward function ([math]\displaystyle{ r }[/math]) is the expected reward from the POMDP reward function over the belief state distribution:

[math]\displaystyle{ r(b,a) = \sum_{s\in S} b(s) R(s,a) }[/math].

The belief MDP is not partially observable anymore, since at any given time the agent knows its belief, and by extension the state of the belief MDP.

Policy and value function

Unlike the "originating" POMDP (where each action is available from only one state), in the corresponding Belief MDP all belief states allow all actions, since you (almost) always have some probability of believing you are in any (originating) state. As such, [math]\displaystyle{ \pi }[/math] specifies an action [math]\displaystyle{ a=\pi(b) }[/math] for any belief [math]\displaystyle{ b }[/math].

Here it is assumed the objective is to maximize the expected total discounted reward over an infinite horizon. When [math]\displaystyle{ R }[/math] defines a cost, the objective becomes the minimization of the expected cost.

The expected reward for policy [math]\displaystyle{ \pi }[/math] starting from belief [math]\displaystyle{ b_0 }[/math] is defined as

[math]\displaystyle{ V^\pi(b_0) = \sum_{t=0}^\infty \gamma^t r(b_t, a_t) = \sum_{t=0}^\infty \gamma^t E\Bigl[ R(s_t,a_t) \mid b_0, \pi \Bigr] }[/math]

where [math]\displaystyle{ \gamma\lt 1 }[/math] is the discount factor. The optimal policy [math]\displaystyle{ \pi^* }[/math] is obtained by optimizing the long-term reward.

[math]\displaystyle{ \pi^* = \underset{\pi}{\mbox{argmax}}\ V^\pi(b_0) }[/math]

where [math]\displaystyle{ b_0 }[/math] is the initial belief.

The optimal policy, denoted by [math]\displaystyle{ \pi^* }[/math], yields the highest expected reward value for each belief state, compactly represented by the optimal value function [math]\displaystyle{ V^* }[/math]. This value function is solution to the Bellman optimality equation:

[math]\displaystyle{ V^*(b) = \max_{a\in A}\Bigl[ r(b,a) + \gamma\sum_{o\in \Omega} \Pr(o\mid b,a) V^*(\tau(b,a,o)) \Bigr] }[/math]

For finite-horizon POMDPs, the optimal value function is piecewise-linear and convex.[3] It can be represented as a finite set of vectors. In the infinite-horizon formulation, a finite vector set can approximate [math]\displaystyle{ V^* }[/math] arbitrarily closely, whose shape remains convex. Value iteration applies dynamic programming update to gradually improve on the value until convergence to an [math]\displaystyle{ \epsilon }[/math]-optimal value function, and preserves its piecewise linearity and convexity.[4] By improving the value, the policy is implicitly improved. Another dynamic programming technique called policy iteration explicitly represents and improves the policy instead.[5][6]

Approximate POMDP solutions

In practice, POMDPs are often computationally intractable to solve exactly. This intractability is often due to the curse of dimensionality or the curse of history (the fact that optimal policies may depend on the entire history of actions and observations). To address these issues, computer scientists have developed methods that approximate solutions for POMDPs. These solutions typically attempt to approximate the problem or solution with a limited number of parameters,[7] plan only over a small part of the belief space online, or summarize the action-observation history compactly.

Grid-based algorithms[8] comprise one approximate solution technique. In this approach, the value function is computed for a set of points in the belief space, and interpolation is used to determine the optimal action to take for other belief states that are encountered which are not in the set of grid points. More recent work makes use of sampling techniques, generalization techniques and exploitation of problem structure, and has extended POMDP solving into large domains with millions of states.[9][10] For example, adaptive grids and point-based methods sample random reachable belief points to constrain the planning to relevant areas in the belief space.[11][12] Dimensionality reduction using PCA has also been explored.[13]

Online planning algorithms approach large POMDPs by constructing a new policy for the current belief each time a new observation is received. Such a policy only needs to consider future beliefs reachable from the current belief, which is often only a very small part of the full belief space. This family includes variants of Monte Carlo tree search[14] and heuristic search.[15] Similar to MDPs, it is possible to construct online algorithms that find arbitrarily near-optimal policies and have no direct computational complexity dependence on the size of the state and observation spaces.[16]

Another line of approximate solution techniques for solving POMDPs relies on using (a subset of) the history of previous observations, actions and rewards up to the current time step as a pseudo-state. Usual techniques for solving MDPs based on these pseudo-states can then be used (e.g. Q-learning). Ideally the pseudo-states should contain the most important information from the whole history (to reduce bias) while being as compressed as possible (to reduce overfitting).[17]

POMDP theory

Planning in POMDP is undecidable in general. However, some settings have been identified to be decidable (see Table 2 in,[18] reproduced below). Different objectives have been considered. Büchi objectives are defined by Büchi automata. Reachability is an example of a Büchi condition (for instance, reaching a good state in which all robots are home). coBüchi objectives correspond to traces that do not satisfy a given Büchi condition (for instance, not reaching a bad state in which some robot died). Parity objectives are defined via parity games; they enable to define complex objectives such that reaching a good state every 10 timesteps. The objective can be satisfied:

  • almost-surely, that is the probability to satisfy the objective is 1;
  • positive, that is the probability to satisfy the objective is strictly greater than 0;
  • quantitative, that is the probability to satisfy the objective is greater than a given threshold.

We also consider the finite memory case in which the agent is a finite-state machine, and the general case in which the agent has an infinite memory.

Objectives Almost-sure (infinite memory) Almost-sure (finite memory) Positive (inf. mem.) Positive (finite mem.) Quantitative (inf. mem) Quantitative (finite mem.)
Büchi EXPTIME-complete EXPTIME-complete undecidable EXPTIME-complete[18] undecidable undecidable
coBüchi undecidable EXPTIME-complete[18] EXPTIME-complete EXPTIME-complete undecidable undecidable
parity undecidable EXPTIME-complete[18] undecidable EXPTIME-complete[18] undecidable undecidable

Applications

POMDPs can be used to model many kinds of real-world problems. Notable applications include the use of a POMDP in management of patients with ischemic heart disease,[19] assistive technology for persons with dementia,[9][10] the conservation of the critically endangered and difficult to detect Sumatran tigers[20] and aircraft collision avoidance.[21]

One application is a teaching case, a crying baby problem, where a parent needs to sequentially decide whether to feed the baby based on the observation of whether the baby is crying or not, which is an imperfect representation of the actual baby's state of hunger.[22][23]

References

  1. Åström, K.J. (1965). "Optimal control of Markov processes with incomplete state information". Journal of Mathematical Analysis and Applications 10: 174–205. doi:10.1016/0022-247X(65)90154-X. https://lup.lub.lu.se/record/8867084. 
  2. 2.0 2.1 Kaelbling, L.P., Littman, M.L., Cassandra, A.R. (1998). "Planning and acting in partially observable stochastic domains". Artificial Intelligence 101 (1–2): 99–134. doi:10.1016/S0004-3702(98)00023-X. 
  3. Sondik, E.J. (1971). The optimal control of partially observable Markov processes (PhD thesis). Stanford University. Archived from the original on October 17, 2019.
  4. Smallwood, R.D., Sondik, E.J. (1973). "The optimal control of partially observable Markov decision processes over a finite horizon". Operations Research 21 (5): 1071–88. doi:10.1287/opre.21.5.1071. 
  5. Sondik, E.J. (1978). "The optimal control of partially observable Markov processes over the infinite horizon: discounted cost". Operations Research 26 (2): 282–304. doi:10.1287/opre.26.2.282. 
  6. Hansen, E. (1998). "Solving POMDPs by searching in policy space". 
  7. Hauskrecht, M. (2000). "Value function approximations for partially observable Markov decision processes". Journal of Artificial Intelligence Research 13: 33–94. doi:10.1613/jair.678. 
  8. Lovejoy, W. (1991). "Computationally feasible bounds for partially observed Markov decision processes". Operations Research 39: 162–175. doi:10.1287/opre.39.1.162. 
  9. 9.0 9.1 Jesse Hoey; Axel von Bertoldi; Pascal Poupart; Alex Mihailidis (2007). "Assisting Persons with Dementia during Handwashing Using a Partially Observable Markov Decision Process". doi:10.2390/biecoll-icvs2007-89. 
  10. 10.0 10.1 Jesse Hoey; Pascal Poupart; Axel von Bertoldi; Tammy Craig; Craig Boutilier; Alex Mihailidis. (2010). "Automated Handwashing Assistance For Persons With Dementia Using Video and a Partially Observable Markov Decision Process". Computer Vision and Image Understanding (CVIU) 114 (5): 503–519. doi:10.1016/j.cviu.2009.06.008. 
  11. Pineau, J., Gordon, G., Thrun, S. (August 2003). "Point-based value iteration: An anytime algorithm for POMDPs". pp. 1025–32. http://www.fore.robot.cc/papers/Pineau03a.pdf. 
  12. Hauskrecht, M. (1997). "Incremental methods for computing bounds in partially observable Markov decision processes". pp. 734–739. 
  13. Roy, Nicholas; Gordon, Geoffrey (2003). "Exponential Family PCA for Belief Compression in POMDPs". Advances in Neural Information Processing Systems. http://papers.nips.cc/paper/2319-exponential-family-pca-for-belief-compression-in-pomdps.pdf. 
  14. David Silver and Joel Veness (2010). "Monte-Carlo planning in large POMDPs". Advances in neural information processing systems. 
  15. Nan Ye, Adhiraj Somani, David Hsu, and Wee Sun Lee (2017). "DESPOT: Online POMDP Planning with Regularization". Journal of Artificial Intelligence Research 58. doi:10.1613/jair.5328. 
  16. Michael H. Lim, Tyler J. Becker, Mykel J. Kochenderfer, Claire J. Tomlin, and Zachary N. Sunberg (2023). "Optimality Guarantees for Particle Belief Approximation of POMDPs". Journal of Artificial Intelligence Research 77. doi:10.1613/jair.1.14525. 
  17. Francois-Lavet, V., Rabusseau, G., Pineau, J., Ernst, D., Fonteneau, R. (2019). "On overfitting and asymptotic bias in batch reinforcement learning with partial observability". 65. 1–30. https://www.jair.org/index.php/jair/article/view/11478. 
  18. 18.0 18.1 18.2 18.3 18.4 Chatterjee, Krishnendu; Chmelík, Martin; Tracol, Mathieu (2016-08-01). "What is decidable about partially observable Markov decision processes with ω-regular objectives". Journal of Computer and System Sciences 82 (5): 878–911. doi:10.1016/j.jcss.2016.02.009. ISSN 0022-0000. 
  19. Hauskrecht, M., Fraser, H. (2000). "Planning treatment of ischemic heart disease with partially observable Markov decision processes". Artificial Intelligence in Medicine 18 (3): 221–244. doi:10.1016/S0933-3657(99)00042-1. PMID 10675716. 
  20. Chadès, I., McDonald-Madden, E., McCarthy, M.A., Wintle, B., Linkie, M., Possingham, H.P. (16 September 2008). "When to stop managing or surveying cryptic threatened species". Proc. Natl. Acad. Sci. U.S.A. 105 (37): 13936–40. doi:10.1073/pnas.0805265105. PMID 18779594. Bibcode2008PNAS..10513936C. 
  21. Kochenderfer, Mykel J. (2015). "Optimized Airborne Collision Avoidance". Decision Making Under Uncertainty. The MIT Press. https://ieeexplore.ieee.org/document/7288641. 
  22. Kochenderfer, Mykel J.; Wheeler, Tim A.; Wray, Kyle H. (2022). Algorithms for decision making. Cambridge, Massachusetts; London, England: MIT Press. p. 678. ISBN 9780262047012. https://algorithmsbook.com/. 
  23. Moss, Robert J. (Sep 24, 2021). "WATCH: POMDPs: Decision Making Under Uncertainty POMDPs.jl. Crying baby problem" (in en) (video). The Julia Programming Language. https://www.youtube.com/watch?v=KDFzObtE6cs. 

External links