Abstract:Models of many real-life applications, such as queuing models of communication networks or computing systems, have a countably infinite state-space. Algorithmic and learning procedures that have been developed to produce optimal policies mainly focus on finite state settings, and do not directly apply to these models. To overcome this lacuna, in this work we study the problem of optimal control of a family of discrete-time countable state-space Markov Decision Processes (MDPs) governed by an unknown parameter $\theta\in\Theta$, and defined on a countably-infinite state space $\mathcal X=\mathbb{Z}_+^d$, with finite action space $\mathcal A$, and an unbounded cost function. We take a Bayesian perspective with the random unknown parameter $\boldsymbol{\theta}^*$ generated via a given fixed prior distribution on $\Theta$. To optimally control the unknown MDP, we propose an algorithm based on Thompson sampling with dynamically-sized episodes: at the beginning of each episode, the posterior distribution formed via Bayes' rule is used to produce a parameter estimate, which then decides the policy applied during the episode. To ensure the stability of the Markov chain obtained by following the policy chosen for each parameter, we impose ergodicity assumptions. From this condition and using the solution of the average cost Bellman equation, we establish an $\tilde O(dh^d\sqrt{|\mathcal A|T})$ upper bound on the Bayesian regret of our algorithm, where $T$ is the time-horizon. Finally, to elucidate the applicability of our algorithm, we consider two different queuing models with unknown dynamics, and show that our algorithm can be applied to develop approximately optimal control algorithms.

Combinations and Mixtures of Optimal Policies in Unichain Markov Decision Processes are Optimal

Asymptotically Optimal Policies for Weakly Coupled Markov Decision Processes

Average Optimality in Markov Decision Processes with Unbounded Rewards

Finding Optimal Memoryless Policies of POMDPs under the Expected Average Reward Criterion

Bayesian Learning of Optimal Policies in Markov Decision Processes with Countably Infinite State-Space

The Finiteness of the Reward Function and the Optimal Value Function in Markov Decision Processes

Sufficiency of Markov Policies for Continuous-Time Jump Markov Decision Processes

A Rollout Algorithm For Multichain Markov Decision Processes With Average Cost

Maximizing the probability of visiting a set infinitely often for a Markov decision process with Borel state and action spaces

Solution to the risk-sensitive average cost optimality equation in a class of Markov decision processes with finite state space

Optimal Policies for Quantum Markov Decision Processes

Optimal Time-Abstract Schedulers for CTMDPs and Markov Games

Optimal Stationary Policies for a Class of Countable Markov Control Processes

Finding Optimal Observation-Based Policies for Constrained POMDPs under the Expected Average Reward Criterion

Optimal Sample Complexity for Average Reward Markov Decision Processes

Nearly Minimax Optimal Reinforcement Learning for Linear Mixture Markov Decision Processes

Mean Field Markov Decision Processes

Average-Cost MDPs with Infinite State and Action Sets: New Sufficient Conditions for Optimality Inequalities and Equations

Risk-Sensitive Optimal Control in Communicating Average Markov Decision Chains

Partial and Conditional Expectations in Markov Decision Processes with Integer Weights

Characterization of the optimal average cost in Markov decision chains driven by a risk-seeking controller