Markov Decision Processes III: The Average Reward Optimality Equation for Unichain ModelsTextbook
Motivation
When a system is controlled indefinitely and decisions are frequent — a router admitting packets, a queue accepting jobs, a machine being maintained — discounting future rewards is often unjustified, and what matters is the long-run average reward per period. Puterman's Chapter 8 (doi:10.1002/9780470316887) develops the theory of this criterion, and its central object is a single equation, the average reward optimality equation , whose unknowns are a scalar gain and a bias function . For unichain models, in which every stationary policy generates a Markov chain with one recurrent class, this equation determines the optimal gain and an optimal stationary policy. The results go back to Howard (Dynamic Programming and Markov Processes, MIT Press, 1960) for the recurrent case and to Blackwell (Discrete dynamic programming, Annals of Mathematical Statistics 33, 1962, doi:10.1214/aoms/1177704593) and Derman for the general finite case; Puterman's Section 8.4 proves them through the discounted theory of mission II, by letting the discount factor tend to one.
Setting
The model is stationary (Assumption 8.0.1): a finite set of states, for each a finite nonempty set of actions, a reward and transition probabilities , none depending on the decision epoch. A policy may randomize and may depend on the whole history; the deterministic stationary policy applies the decision rule at every epoch. Its transition matrix is .
For a policy , is the expected reward over epochs. Since the limit of need not exist (Example 8.1.1), the chapter works with the lim sup and lim inf average rewards and , and with . A policy is average optimal when for all and , the strongest of the three criteria of Section 8.1.2.
The optimality residual is , and the optimality equation is . A decision rule is -improving when it attains at every state. A transition matrix is unichain when it consists of a single recurrent class plus a possibly empty set of transient states, and the MDP is unichain when is unichain for every deterministic decision rule.
Formalization targets
Goal — Theorem 8.4.5 (printed p. 361)
For a finite unichain model: (a) some deterministic stationary policy is average optimal; (b) the optimality equation has a solution, and (d) its scalar satisfies for every ; (c) for every solution, every -improving decision rule gives an average optimal stationary policy.
Theorem 8.4.1 (printed p. 356)
If then ; if then ; if then .
Theorem 8.4.3 (printed p. 358)
In a finite unichain model has a solution, and every solution has the same .
Theorem 8.4.4 (printed p. 361)
If and is -improving, then is average optimal.
Significance
Theorem 8.4.1(c) is what the source calls "one of the most important results for average reward models": a solution of the optimality equation with constant pins down the optimal gain under every criterion at once, so that in finite unichain models the three optimality criteria of Section 8.1.2 coincide. Theorem 8.4.3 guarantees such a solution exists, and Theorem 8.4.4 reads an optimal policy off it. Together, Theorem 8.4.5 reduces the infinite-horizon average reward problem over all history-dependent randomized policies to a finite system of equations in , which is what policy iteration, value iteration and linear programming solve in Sections 8.5 to 8.8.
The results are classical and proved. Formalizing them fixes the chain-structure hypothesis in
a checkable form and pins down which criterion "average optimal" means, two places where the
literature is loose. The platform's MarkovDecisionProcesses series has the finite-horizon
(mission I) and discounted (mission II) models; this mission adds the undiscounted stationary
model, the gains, and the unichain classification, on which Chapter 9's multichain optimality
equations and Chapter 10's sensitive discount optimality can be built.
Difficulty
The obvious argument for Theorem 8.4.3 is to take the discounted optimal value of mission II and let . It fails as stated because blows up like ; what converges is the Laurent expansion of the value of a fixed stationary policy, Corollary 8.2.4, and that expansion needs the limiting matrix and the deviation matrix of a unichain chain. So the proof must first develop the Markov chain theory of Section 8.2 and Appendix A, choose a subsequence of discount factors along which one policy is discount optimal (possible because is finite), and only then pass to the limit in the discounted optimality equation.
Theorem 8.4.1 looks elementary and hides the analytic step: iterating along an arbitrary history-dependent policy and dividing by requires the telescoping term to vanish, which uses boundedness of , and requires the reduction from history-dependent randomized to Markov randomized policies (Theorem 8.1.2). For Theorem 8.4.4 the step is Corollary 8.2.7, that forces the gain of to be , which is the multiplication by that annihilates .
The traps are in the definitions. Recurrence and the unichain property must be stated so that the source's Example 8.4.3 comes out as the book says — the policy using has the absorbing state as its single recurrent class — and the optimality residual must use the lim sup / lim inf gains, since a definition through a limit that need not exist would be a junk value on the policies of Example 8.1.1.
Formalization scope
State and action spaces are Fintypes and admissible actions are nonempty Finsets, as in
missions I and II; the stationary model is a new structure because mission II's DiscountedMDP
bundles a discount factor, and carries the same data otherwise. Policies are history-dependent
and randomized, so "average optimal" has its full strength; a stationary policy is the
deterministic one built from a decision rule. Expected total reward is defined by the policy
evaluation recursion, as in the earlier missions, rather than through a measure on
trajectories.
Gains are Filter.limsup and Filter.liminf of on ; these are
the source's because the sequence is bounded by , and the suprema over the
nonempty family of policies are genuine real suprema for the same reason. The residual
is a Finset.sup' over the admissible actions. Recurrence is "every state reachable from
reaches " and unichain is "any two recurrent states communicate", the definitions of Appendix
A for finite chains, applied to for every admissible deterministic decision rule.
Restrictions relative to the printed text, all noted in the items: Theorem 8.4.1 is stated for finite where the source says countable, since the chapter's standing assumption and the model are finite; the gain of a stationary policy in (8.4.5) is written as its lim inf gain, which equals it; and the chain of (8.4.6) is stated through and , since presupposes existing limits. Nothing is trivialized: the existential in Theorem 8.4.5(a) has to produce a decision rule, and with a junk maximum is impossible since every is nonempty. Welcome contributions beyond the milestones: Theorem 8.1.2 (reduction to Markov policies), Corollary 8.2.7 (the gain of a stationary policy from the evaluation equations), and the equivalence of the three optimality criteria in finite models.
Selected references
- Martin L. Puterman, Markov Decision Processes: Discrete Stochastic Dynamic Programming, Wiley, 1994, Chapter 8. doi:10.1002/9780470316887
- Ronald A. Howard, Dynamic Programming and Markov Processes, MIT Press, 1960.
- David Blackwell, Discrete dynamic programming, Annals of Mathematical Statistics 33 (1962). doi:10.1214/aoms/1177704593
- Cyrus Derman, Finite State Markovian Decision Processes, Academic Press, 1970.
- Paul J. Schweitzer and Awi Federgruen, The functional equations of undiscounted Markov renewal programming, Mathematics of Operations Research 3 (1978). doi:10.1287/moor.3.4.308