A policy gradient approach for Finite Horizon Constrained Markov Decision Processes

TL;DR

This paper introduces the first policy gradient algorithm tailored for finite horizon constrained Markov Decision Processes, enabling the derivation of non-stationary, stage-dependent policies with proven convergence.

Contribution

It develops a novel constrained policy gradient method specifically for finite horizon problems, addressing the gap in existing infinite horizon-focused algorithms.

Findings

Algorithm converges to a constrained optimal policy.

Performs better than existing algorithms in experiments.

Handles large or continuous state and action spaces effectively.

Abstract

The infinite horizon setting is widely adopted for problems of reinforcement learning (RL). These invariably result in stationary policies that are optimal. In many situations, finite horizon control problems are of interest and for such problems, the optimal policies are time-varying in general. Another setting that has become popular in recent times is of Constrained Reinforcement Learning, where the agent maximizes its rewards while it also aims to satisfy some given constraint criteria. However, this setting has only been studied in the context of infinite horizon MDPs where stationary policies are optimal. We present an algorithm for constrained RL in the Finite Horizon Setting where the horizon terminates after a fixed (finite) time. We use function approximation in our algorithm which is essential when the state and action spaces are large or continuous and use the policy gradient method to find the optimal policy. The optimal policy that we obtain depends on the stage and so is non-stationary in general. To the best of our knowledge, our paper presents the first policy gradient algorithm for the finite horizon setting with constraints. We show the convergence of our algorithm to a constrained optimal policy. We also compare and analyze the performance of our algorithm through experiments and show that our algorithm performs better than some other well known algorithms.