Existence of a discounted Bellman solution for a finite MDP
ProvedBanditAlgorithm.mdp_discounted_bellman_solutionLet be a finite MDP with states, actions, transition rows and rewards , and let be a discount factor. Then there are a value function with and a map such that
Equivalently for every state , and is a greedy — hence optimal — deterministic memoryless policy for the -discounted criterion. The conclusion is stated as the conjunction of the inequality and the attaining action rather than through a maximum, so that it can be used without carrying a nonemptiness proof for the action set.
The proof is the Banach fixed point theorem. The discounted Bellman optimality operator
is a -contraction of in the supremum norm: for each fixed action the map moves by at most because is a probability vector, and a maximum over a finite set of uniformly close functions is uniformly close. The space is complete, so has a fixed point ; the inequality is the definition of the maximum and the equality is its attainment. The bounds come from evaluating the fixed point equation at a maximising and at a minimising state: gives , and gives .
This is the computational core of value iteration, and the starting point of the vanishing-discount proof of the average-reward optimality equation (Theorem 38.2 of Lattimore and Szepesvári, whose proof is left to their Exercise 38.10): the rescaled pairs solve the average-reward Bellman inequality, and a limit along solves the optimality equation.
import Definitions.Def_FiniteMDPLearning import Mathlib.Topology.MetricSpace.Contracting open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.mdp_discounted_bellman_solution {S A : ℕ} (hS : 0 < S) (hA : 0 < A)
(M : FiniteMDP S A) (γ : ℝ) (hγ0 : 0 ≤ γ) (hγ1 : γ < 1) :
∃ (V : Fin S → ℝ) (f : Fin S → Fin A),
(∀ s, V s ∈ Set.Icc (0 : ℝ) (1 / (1 - γ))) ∧
(∀ s a, M.r s a + γ * ∑ s', (M.P s a s' : ℝ) * V s' ≤ V s) ∧
(∀ s, V s = M.r s (f s) + γ * ∑ s', (M.P s (f s) s' : ℝ) * V s') := by
sorry