English

MDP Geometry, Normalization and Reward Balancing Solvers

Machine Learning 2025-03-06 v4 Optimization and Control

Abstract

We present a new geometric interpretation of Markov Decision Processes (MDPs) with a natural normalization procedure that allows us to adjust the value function at each state without altering the advantage of any action with respect to any policy. This advantage-preserving transformation of the MDP motivates a class of algorithms which we call Reward Balancing, which solve MDPs by iterating through these transformations, until an approximately optimal policy can be trivially found. We provide a convergence analysis of several algorithms in this class, in particular showing that for MDPs for unknown transition probabilities we can improve upon state-of-the-art sample complexity results.

Keywords

Cite

@article{arxiv.2407.06712,
  title  = {MDP Geometry, Normalization and Reward Balancing Solvers},
  author = {Arsenii Mustafin and Aleksei Pakharev and Alex Olshevsky and Ioannis Ch. Paschalidis},
  journal= {arXiv preprint arXiv:2407.06712},
  year   = {2025}
}

Comments

AISTATS 2025 camera-ready version

R2 v1 2026-06-28T17:34:06.998Z