Non-Stationary Markov Decision Processes, a Worst-Case Approach using Model-Based Reinforcement Learning, Extended version

  • 2020-01-10 16:43:46
  • Erwan Lecarpentier, Emmanuel Rachelson
  • 0

Abstract

This work tackles the problem of robust zero-shot planning in non-stationarystochastic environments. We study Markov Decision Processes (MDPs) evolvingover time and consider Model-Based Reinforcement Learning algorithms in thissetting. We make two hypotheses: 1) the environment evolves continuously with abounded evolution rate; 2) a current model is known at each decision epoch butnot its evolution. Our contribution can be presented in four points. 1) wedefine a specific class of MDPs that we call Non-Stationary MDPs (NSMDPs). Weintroduce the notion of regular evolution by making an hypothesis ofLipschitz-Continuity on the transition and reward functions w.r.t. time; 2) weconsider a planning agent using the current model of the environment butunaware of its future evolution. This leads us to consider a worst-case methodwhere the environment is seen as an adversarial agent; 3) following thisapproach, we propose the Risk-Averse Tree-Search (RATS) algorithm, a zero-shotModel-Based method similar to Minimax search; 4) we illustrate the benefitsbrought by RATS empirically and compare its performance with referenceModel-Based algorithms.

 

Quick Read (beta)

Non-Stationary Markov Decision Processes
a Worst-Case Approach using Model-Based Reinforcement Learning

Erwan Lecarpentier
Université de Toulouse
ONERA - The French Aerospace Lab
[email protected]
&Emmanuel Rachelson
Université de Toulouse
ISAE-SUPAERO
[email protected]
Abstract

This work tackles the problem of robust planning in non-stationary stochastic environments. We study Markov Decision Processes (MDPs) evolving over time and consider Model-Based Reinforcement Learning algorithms in this setting. We make two hypotheses: 1) the environment evolves continuously with a bounded evolution rate; 2) a current model is known at each decision epoch but not its evolution. Our contribution can be presented in four points. 1) we define a specific class of MDPs that we call Non-Stationary MDPs (NSMDPs). We introduce the notion of regular evolution by making an hypothesis of Lipschitz-Continuity on the transition and reward functions w.r.t. time; 2) we consider a planning agent using the current model of the environment but unaware of its future evolution. This leads us to consider a worst-case method where the environment is seen as an adversarial agent; 3) following this approach, we propose the Risk-Averse Tree-Search (RATS) algorithm, a Model-Based method similar to minimax search; 4) we illustrate the benefits brought by RATS empirically and compare its performance with reference Model-Based algorithms.

 

Non-Stationary Markov Decision Processes
a Worst-Case Approach using Model-Based Reinforcement Learning


  Erwan Lecarpentier Université de Toulouse ONERA - The French Aerospace Lab erwan.le[email protected] Emmanuel Rachelson Université de Toulouse ISAE-SUPAERO [email protected]

\@float

noticebox[b]33rd Conference on Neural Information Processing Systems (NeurIPS 2019), Vancouver, Canada.\[email protected]

1 Introduction

One of the hot topics of modern Artificial Intelligence (AI) is the ability for an agent to adapt its behavior to changing tasks. In the literature, this problem is often linked to the setting of Lifelong Reinforcement Learning (LRL) (silver2013lifelong; abel2018state; abel2018policy) and learning in non-stationary environments (choi1999hidden; jaulmes2005learning; hadoux2015markovian). In LRL, the tasks presented to the agent change sequentially at discrete transition epochs (silver2013lifelong). Similarly, the non-stationary environments considered in the literature often evolve abruptly (hadoux2015markovian; hadoux2014sequential; doya2002multiple; da2006dealing; choi1999hidden; choi2000hidden; choi2001solving; campo1991state; wiering2001reinforcement). In this paper, we investigate environments continuously changing over time that we call Non-Stationary Markov Decision Processes (NSMDPs). In this setting, it is realistic to bound the evolution rate of the environment using a Lipschitz Continuity (LC) assumption.

Model-based Reinforcement Learning approaches (sutton1998reinforcement) benefit from the knowledge of a model allowing them to reach impressive performances, as demonstrated by the Monte Carlo Tree Search (MCTS) algorithm (silver2016mastering). In this matter, the necessity to have access to a model is a great concern of AI (asadi2018lipschitz; jaulmes2005learning; doya2002multiple; da2006dealing). In the context of NSMDPs, we assume that an agent is provided with a snapshot model when its action is computed. By this, we mean that it only has access to the current model of the environment but not its future evolution, as if it took a photograph but would be unable to predict how it is going to evolve. This hypothesis is realistic, because many environments have a tractable state while their future evolution is hard to predict (da2006dealing; wiering2001reinforcement). In order to solve LC-NSMDPs, we propose a method that considers the worst-case possible evolution of the model and performs planning w.r.t. this model. This is equivalent to considering Nature as an adversarial agent. The paper is organized as follows: first we describe the NSMDP setting and the regularity assumption (Section 2); then we outline related works (Section 3); follows the explanation of the worst-case approach proposed in this paper (Section 4); then we describe an algorithm reflecting this approach (Section 5); finally we illustrate its behavior empirically (Section 6).

2 Non-Stationary Markov Decision Processes

To define a Non-Stationary Markov Decision Process (NSMDP), we revert to the initial MDP model introduced by puterman2014markov, where the transition and reward functions depend on time.

Definition 1.

NSMDP. An NSMDP is an MDP whose transition and reward functions depend on the decision epoch. It is defined by a 5-tuple {S,T,A,(pt)tT,(rt)tT} where S is a state space; T{1,2,,N} is the set of decision epochs with N+; A is an action space; pt(ss,a) is the probability of reaching state s while performing action a at decision epoch t in state s; rt(s,a,s) is the scalar reward associated to the transition from s to s with action a at decision epoch t.

This definition can be viewed as that of a stationary MDP whose state space has been enhanced with time. While this addition is trivial in episodic tasks where an agent is given the opportunity to interact several times with the same MDP, it is different when the experience is unique. Indeed, no exploration is allowed along the temporal axis. Within a stationary, infinite-horizon MDP with a discounted criterion, it is proven that there exists a Markovian deterministic stationary policy (puterman2014markov). It is not the case within NSMDPs where the optimal policy is non-stationary in the most general case. Additionally, we define the expected reward received when taking action a at state s and decision epoch t as Rt(s,a)=𝔼spt(s,a)[rt(s,a,s)]. Without loss of generality, we assume the reward function to be bounded between -1 and 1. In this paper, we consider discrete time decision processes with constant transition durations, which imply deterministic decision times in Definition 1. This assumption is mild since many discrete time sequential decision problems follow that assumption. A non-stationary policy π is a sequence of decision rules πt which map states to actions (or distributions over actions). For a stochastic non-stationary policy πt(as), the value of a state s at decision epoch t within an infinite horizon NSMDP is defined, with γ[0,1) a discount factor, by:

Vtπ(s)=𝔼[i=tγi-tRi(si,ai)|st=s,aiπi(si),si+1pi(si,ai)],

The definition of the state-action value function Qtπ for π at decision epoch t is straightforward:

Qtπ(s,a)=Rt(s,a)+γ𝔼spt(s,a)[Vt+1π(s)].

Overall, we defined an NSMDP as an MDP where we stress out the distinction between state, time, and decision epoch due to the inability for an agent to explore the temporal axis at will. This distinction is particularly relevant for non-episodic tasks, i.e. when there is no possibility to re-experience the same MDP starting from a prior date.

The regularity hypothesis. Many real-world problems can be modeled as an NSMDP. For instance, the problem of path planning for a glider immersed in a non-stationary atmosphere (chung2015learning; lecarpentier2017empirical), or that of vehicle routing in dynamic traffic congestion. Realistically, we consider that the expected reward and transition functions do not evolve arbitrarily fast over time. Conversely, if such an assumption was not made, a chaotic evolution of the NSMDP would be allowed which is both unrealistic and hard to solve. Hence, we assume that changes occur slowly over time. Mathematically, we formalize this hypothesis by bounding the evolution rate of the transition and expected reward functions, using the notion of Lipschitz Continuity (LC).

Definition 2.

Lipschitz Continuity. Let (X,dX) and (Y,dY) be two metric spaces and f:XY, f is L-Lipschitz Continuous (L-LC) with LR+ iff dY(f(x),f(x^))LdX(x,x^),(x,x^)X2. L is called a Lipschitz constant of the function f.

We apply this hypothesis to the transition and reward functions of an NSMDP so that those functions are LC w.r.t. time. For the transition function, this leads to the consideration of a metric between probability density functions. For that purpose, we use the 1-Wasserstein distance (villani2008optimal).

Definition 3.

1-Wasserstein distance. Let (X,dX) be a Polish metric space, μ,ν any probability measures on X, Π(μ,ν) the set of joint distributions on X×X with marginals μ and ν. The 1-Wasserstein distance between μ and ν is W1(μ,ν)=infπΠ(μ,ν)X×XdX(x,y)dπ(x,y).

The choice of the Wasserstein distance is motivated by the fact that it quantifies the distance between two distributions in a physical manner, respectful of the topology of the measured space (dabney2018distributional; asadi2018lipschitz). First, it is sensitive to the difference between the supports of the distributions. Comparatively, the Kullback-Leibler divergence between distributions with disjoint supports is infinite. Secondly, if one consider two regions of the support where two distributions differ, the Wasserstein distance is sensitive to the distance between the elements of those regions. Comparatively, the total-variation metric is the same regardless of this distance.

Definition 4.

(Lp,Lr)-LC-NSMDP. An (Lp,Lr)-LC-NSMDP is an NSMDP whose transition and reward functions are respectively Lp-LC and Lr-LC w.r.t. time, i.e., (t,t^,s,s,a)T2×S2×A,

W1(pt(s,a),pt^(s,a))Lp|t-t^|𝑎𝑛𝑑|rt(s,a,s)-rt^(s,a,s)|Lr|t-t^|.

One should remark that the LC property should be defined with respect to actual decision times and not decision epoch indexes for the sake of realism. In the present case, both have the same value, and we choose to keep this convention for clarity. Our results however extend easily to the case where indexes and times do not coincide. From now on, we consider (Lp,Lr)-LC-NSMDPs, making Lipschitz Continuity our regularity property. Notice that R is defined as a convex combination of r by the probability measure p. As a result, the notion of Lipschitz Continuity of R is strongly related to that of r and p as showed by Property 1. All the proofs of the paper can be found in the Appendix.

Property 1.

Given an (Lp,Lr)-LC-NSMDP, the expected reward function Rt:s,aEspt(s,a){rt(s,a,s)} is LR-LC with LR=Lr+Lp.

This result shows R’s evolution rate is conditioned by the evolution rates of r and p. It allows to work either with the reward function r or its expectation R, benefiting from the same LC property.

3 Related work

iyengar2005robust introduced the framework of robust MDPs, where the transition function is allowed to evolve within a set of functions due to uncertainty. This differs from our work in two fundamental aspects: 1) we consider uncertainty in the reward model as well; 2) we use a stronger Lipschitz formulation on the set of possible transition and reward functions, this last point being motivated by its relevance to the non-stationary setting. szita2002varepsilon also consider the robust MDP setting and adopt a different constraint hypothesis on the set of possible functions than our LC assumption. They control the total variation distance of transition functions from subsequent decision epochs by a scalar value. Those slowly changing environments allow model-free RL algorithms such as Q-Learning to find near optimal policies. lim2013reinforcement consider learning in robust MDPs where the model evolves in an adversarial manner for a subset of 𝒮×𝒜. In that setting, they propose to learn to what extent the adversary can modify the model and to deduce a behavior close to the minimax policy. even2009online studied the case of non-stationary reward functions with fixed transition models. No assumption is made on the set of possible functions and they propose an algorithm achieving sub-linear regret w.r.t. the best stationary policy. dick2014online viewed a similar setting from the perspective of online linear optimization. csaji2008value studied the NSMDP setting with an assumption of reward and transition functions varying in a neighborhood of a reference reward-transition function pair. Finally, abbasi2013online address the adversarial NSMDP setting with a mixing assumption constraint instead of the LC assumption we make.

Non-stationary environments also have been studied through the framework of Hidden Mode MDPs (HM-MDP) introduced by choi1999hidden. This is a special class of Partially Observable MDPs (POMDPs) (kaelbling1998planning) where a hidden mode indexes a latent stationary MDP within which the agent evolves. Similarly to the context of LRL, the agent experiences a series of different MDPs over time. In this setting, choi1999hidden; choi2000hidden proposed methods to learn the different models of the latent stationary MDPs. doya2002multiple built a modular architecture switching between models and policies when a change is detected. Similarly, wiering2001reinforcement; da2006dealing; hadoux2014sequential proposed a method tracking the switching occurrence and re-planning if needed. Overall, as in LRL, the HM-MDP setting considers abrupt evolution of the transition and reward functions whereas we consider a continuous one. Other settings have been considered, as by jaulmes2005learning, who do not make particular hypothesis on the evolution of the NSMDP. They build a learning algorithm for POMDPs solving, weighting recently experienced transitions more than older ones to account for the time dependency.

To plan robustly within an NSMDP, our approach consists in exploiting the slow LC evolution of the environment. Utilizing Lipschitz continuity to infer bounds on a function is common in the RL, bandit and optimization communities (kleinberg2008multi; rachelson2010locality; pirotta2015policy; pazis2013pac; munos2014bandits). We implement this approach with a minimax-like algorithm (fudenberg1991game), where the environment is seen as an adversarial agent.

4 Worst-case approach

We consider finding an optimal policy within an LC-NSMDP under the non-episodic task hypothesis. The latter prevents us from learning from previous experience data since they become outdated with time and no information samples have been collected yet for future time steps. An alternative is to use model-based RL algorithms such as MCTS. For a current state s0, such algorithms focus on finding the optimal action a0* by using a generative model. This action is then undertaken and the operation repeated at the next state. However, using the true NSMDP model for this purpose is an unrealistic hypothesis, since this model is generally unknown. We assume the agent does not have access to the true NSMDP model; instead, we introduce the notion of snapshot model.Intuitively, the snapshot associated to time t0 is a temporal slice of the NSMDP at t0.

Definition 5.

Snapshot of an NSMDP. The snapshot of an NSMDP {S,T,A,(pt)tT,(rt)tT} at decision epoch t0, denoted by MDPt0, is the stationary MDP defined by the 4-tuple {S,A,pt0,rt0} where pt0(ss,a) and rt0(s,a,s) are the transition and reward functions of the NSMDP at t0.

Similarly to the NSMDP, this definition induces the existence of the snapshot expected reward Rt0 defined by Rt0:s,a𝔼spt0(s,a){rt0(s,a,s)}. Notice that the snapshot MDPt0 is stationary and coincides with the NSMDP only at t0. Particularly, one can generate a trajectory {s0,r0,,sk} within an NSMDP using the sequence of snapshots {MDPt0,,MDPt0+k-1} as a model. Overall, the hypothesis of using snapshot models amounts to considering a planning agent only able to get the current stationary model of the environment. In real-world problems, predictions often are uncertain or hard to perform e.g. in the thermal soaring problem of a glider.

We consider a generic planning agent at s0,t0, using MDPt0 as a model of the NSMDP. By planning, we mean conducting a look-ahead search within the possible trajectories starting from s0,t0 given a model of the environment. The search allows in turn to identify an optimal action w.r.t. the model. This action is then undertaken and the agent jumps to the next state where the operation is repeated. The consequence of planning with MDPt0 is that the estimated value of an s,t pair is the value of the optimal policy of MDPt0, written VMDPt0*(s). The true optimal value of s at t within the NSMDP does not match this estimate because of the non-stationarity. The intuition we develop is that, given the slow evolution rate of the environment, for a state s seen at a future decision epoch during the search, we can predict a scope into which the transition and reward functions at s lie.

Property 2.

Set of admissible snapshot models. Consider an (Lp,Lr)-LC-NSMDP, s,t,aS×T×A. The transition and expected reward functions (pt,Rt) of the snapshot MDPt respect

(pt,Rt)ΔtW1(pt-1(s,a),Lp)×||(Rt-1(s,a),LR)

where LR=Lp+Lr and Bd(c,r) denotes the ball of centre c, defined with metric d and radius r.

For a future prediction at s,t, we consider the question of using a better model than pt0,Rt0. The underlying evolution of the NSMDP being unknown, a desirable feature would be to use a model leading to a policy that is robust to every possible evolution. To that end, we propose to use the snapshots corresponding to the worst possible evolution scenario under the constraints of Property 2. We claim that such a practice is an efficient way to 1) ensure robust performance to all possible evolutions of the NSMDP and 2) avoid catastrophic terminal states. Practically, this boils down to using a different value estimate for s at t than VMDPt0*(s) which provided no robustness guarantees.

Given a policy π=(πt)t𝒯 and a decision epoch t, a worst-case NSMDP corresponds to a sequence of transition and reward models minimizing the expected value of applying π in any pair (s,t), while remaining within the bounds of Property 2. We write V¯tπ(s) this value for s at decision epoch t.

V¯tπ(s)min(pi,Ri)Δi,i𝒯𝔼[i=tγi-tRi(si,ai)|st=saiπi(si),si+1pi(si,ai)] (1)

Intuitively, the worst-case NSMDP is a model of a non-stationary environment leading to the poorest possible performance for π, while being an admissible evolution of MDPt. Let us define Q¯tπ(s,a) as the worst-case Q-value for the pair (s,a) at decision epoch t:

Q¯tπ(s,a)min(p,R)Δt𝔼sp[R(s,a)+γV¯t+1π(s)]. (2)

5 Risk-Averse Tree-Search algorithm

The algorithm. Tree search algorithms within MDPs have been well studied and cover two classes of search trees, namely closed loop (keller2013trial; kocsis2006bandit; browne2012survey) and open loop (bubeck2010open; lecarpentier2018open). Following (keller2013trial), we consider closed loop search trees, composed of decision nodes alternating with chance nodes. We adapt their formulation to take time into account, resulting in the following definitions. A decision node at depth t, denoted by νs,t, is labeled by a unique state / decision epoch pair (s,t). The edges leading to its children chance nodes correspond to the available actions at (s,t). A chance node, denoted by νs,t,a, is labeled by a state / decision epoch / action triplet (s,t,a). The edges leading to its children decision nodes correspond to the reachable state / decision epoch pairs (s,t) after performing a in (s,t) as illustrated by Figure 1.

Figure 1: Tree structure, dmax=2, 𝒜={a1,a2}.
ϵ RATS DP-snapshot DP-NSMDP
0 𝔼[r] -0.026 0.48 0.47
CVaR -0.81 -0.90 -0.9
0.5 𝔼[r] -0.032 -0.46 -0.077
CVaR -0.81 -0.90 -0.81
1 𝔼[r] 0.67 -0.78 0.66
CVaR 0.095 -0.90 -0.033
Figure 2: Expected return 𝔼[r] and CVaR at 5%.
Figure 3: Tree structure and results from the Non-Stationary bridge experiment.

We consider the problem of estimating the optimal action a0* at s0,t0 within a worst-case NSMDP, knowing MDPt0. This problem is twofold. It requires 1) to estimate the worst-case NSMDP given MDPt0 and 2) to explore the latter in order to identify a0*. We propose to tackle both problems with an algorithm inspired by the minimax algorithm (fudenberg1991game) where the max operator corresponds to the agent’s policy, seeking to maximize the return; and the min operator corresponds to the worst-case model, seeking to minimize the return. Estimating the worst-case NSMDP requires to estimate the sequence of subsequent snapshots minimizing Equation 2. The inter-dependence of those snapshots (Equation 1) makes the problem hard to solve (iyengar2005robust), particularly because of the combinatorial nature of the opponent’s action space. Instead, we propose to solve a relaxation of this problem, by considering snapshots only constrained by MDPt0. Making this approximation leaves a possibility to violate property 2 but allows for an efficient search within the developed tree and (as will be shown experimentally) leads to robust policies. For that purpose, we define the set of admissible snapshot models w.r.t. MDPt0 by Δt0tW1(pt0(s,a),Lp|t-t0|)×||(Rt0(s,a),LR|t-t0|). The relaxed analogues of Equations 1 and 2 for s,t,a𝒮×𝒯×𝒜 are defined as follows:

V^t0,tπ(s)min(pi,Ri)Δt0i,i𝒯𝔼[i=tγi-tRi(si,ai)|st=saiπi(si),si+1pi(si,ai)],
Q^t0,tπ(s,a)min(p,R)Δt0t𝔼sp[R(s,a)+γV^t0,t+1π(s)].

Their optimal counterparts, while seeking to find the optimal policy, verify the following equations:

V^t0,t*(s) =maxa𝒜Q^t0,t*(s,a), (3)
Q^t0,t*(s,a) =min(p,R)Δt0t𝔼sp[R(s,a)+γV^t0,t+1*(s)]. (4)

We now provide a method to calculate those quantities within the nodes of the tree search algorithm.
Max nodes. A decision node νs,t corresponds to a max node due to the greediness of the agent w.r.t. the subsequent values of the children. We aim at maximizing the return while retaining a risk-averse behavior. As a result, the value of νs,t follows Equation 3 and is defined as:

V(νs,t)=maxa𝒜V(νs,t,a). (5)

Min nodes. A chance node νs,t,a corresponds to a min node due to the use of a worst-case NSMDP as a model which minimizes the value of νs,t,a w.r.t. the reward and the subsequent values of its children. Writing the value of νs,t,a as the value of s,t,a, within the worst-case snapshot minimizing Equation 4, and using the children’s values as values for the next reachable states, leads to Equation 6.

V(νs,t,a)=min(p,R)Δt0tR(s,a)+γ𝔼spV(νs,t+1) (6)

Our approach considers the environment as an adversarial agent, as in an asymmetric two-player game, in order to search for a robust plan. The resulting algorithm, RATS for Risk-Averse Tree-Search, is described in Algorithm 6. Given an initial state / decision epoch pair, a minimax tree is built using the snapshot MDPt0 and the operators corresponding to Equations 5 and 6 in order to estimate the worst-case snapshots at each depth. The tree is built, the action leading to the best possible value from the root node is selected and a real transition is performed. The next state is then reached, the new snapshot model MDPt0+1 is acquired and the process re-starts. Notice the use of R(ν) and p(νν) in the pseudo-code: they are light notations respectively standing for Rt(s,a) corresponding to a chance node ννs,t,a and the probability pt(s|s,a) to jump to a decision node ννs,t+1 given a chance node ννs,t,a. {algorithm2e}[t] \DontPrintSemicolonRATS (s0, t0, maxDepth)  ν0 = rootNode(s0,t0)  Minimax(ν0)  ν*=argmaxν in ν0.childrenν.valuereturn ν*.action    Minimax (ν, maxDepth)  \Ifν is DecisionNode \Ifν.state is terminal or ν.depth = maxDepth return ν.value = heuristicValue(ν.state)  \Else return ν.value = maxνν.childrenMinimax(ν, maxDepth)  \Else return ν.value=min(p,R)Δt0tR(ν)+γνν.childrenp(νν)Minimax(ν,maxDepth)RATS algorithm The tree built by RATS is entirely developed until the maximum depth dmax. A heuristic function is used to evaluate the leaf nodes of the tree.

Analysis of RATS. We are interested in characterizing Algorithm 6 without function approximation and therefore will consider finite, countable, 𝒮×𝒜 sets. We now detail the computation of the min operator (Property 3), the computational complexity of RATS (Property 4) and the heuristic function.

Property 3.

Closed-form expression of the worst case snapshot of a chance node. Following Algorithm 6, a solution to Equation 6 is given by:

R^(s,a)=Rt0(s,a)-LR|t-t0|𝑎𝑛𝑑p^(s,a)=(1-λ)pt0(s,a)+λp𝑠𝑎𝑡(s,a)

with p𝑠𝑎𝑡(s,a)=(0,,0,1,0,,0) with 1 at position argminsV(νs,t+1), λ=1 if W1(p𝑠𝑎𝑡,p0)Lp|t-t0| and λ=Lp|t-t0|/W1(p𝑠𝑎𝑡,p0) otherwise.

Property 4.

Computational complexity. The total computation complexity of Algorithm 6 is O(B|S|1.5|A|(|S||A|)dmax) with B the number of time steps and dmax the maximum depth.

Heuristic function. As in vanilla minimax algorithms, Algorithm 6 bootstraps the values of the leaf nodes with a heuristic function if these leaves do not correspond to terminal states. Given such a leaf node νs,t, a heuristic aims at estimating the value of the optimal policy at (s,t) within the worst-case NSMDP, i.e. V^t0,t*(s). Let H(s,t) be such a heuristic function, we call heuristic error in (s,t) the difference between H(s,t) and V^t0,t*(s). Assuming that the heuristic error is uniformly bounded, the following property provides an upper bound on the propagated error due to the choice of H.

Property 5.

Upper bound on the propagated heuristic error within RATS. Consider an agent executing Algorithm 6 at s0,t0 with a heuristic function H. We note L the set of all leaf nodes. Suppose that the heuristic error is uniformly bounded, i.e. δ>0,νs,tL,|H(s)-V^t0,t*(s)|δ. Then we have for every decision and chance nodes νs,t and νs,t,a, at any depth d[0,dmax]:

|V(νs,t)-V^t0,t*(s)|γ(dmax-d)δ and |V(νs,t,a)-Q^t0,t*(s,a)|γ(dmax-d)δ.

This last result implies that with any heuristic function H inducing a uniform heuristic error, the propagated error at the root of the tree is guaranteed to be upper bounded by γdmaxδ. In particular, since the reward function is bounded by hypothesis, we have V^t0,t*(s)1/(1-γ). Thus, selecting for instance the zero function ensures a root node heuristic error of at most γdmax/(1-γ). In order to improve the precision of the algorithm, we propose to guide the heuristic by using a function reflecting better the value of state s at leaf node νs,t. The ideal function would of course be H(s)=V^t0,t*(s), reducing the heuristic error to zero, but this is intractable. Instead, we suggest to use the value of s within the snapshot MDPt using an evaluation policy π, i.e. H(s)=VMDPtπ(s). This snapshot is also not available, but Property 6 provides a range wherein this value lies.

Property 6.

Bounds on the snapshots values. Let sS, π a stationary policy, MDPt0 and MDPt two snapshot MDPs, t,t0T2 be. We note VMDPiπ(s) the value of s within MDPi following π. Then,

|V𝑀𝐷𝑃t0π(s)-V𝑀𝐷𝑃tπ(s)||t-t0|LR/(1-γ).

Since MDPt0 is available, VMDPt0π(s) can be estimated, e.g. via Monte-Carlo roll-outs. Let V^MDPt0π(s) denote such an estimate. Following Property 6, VMDPt0π(s)-|t-t0|LR/(1-γ)VMDPtπ(s). Hence, a worst-case heuristic on VMDPtπ(s) is H(s)=V^MDPt0π(s)-|t-t0|LR/(1-γ). The bounds provided by Property 5 decrease quickly with dmax, and given that dmax is large enough, RATS provides the optimal risk-averse maximizing the worst-case value for any evolution of the NSMDP.

6 Experiments

We compare the RATS algorithm with two policies 11 1 Code: https://github.com/SuReLI/rats-experiments – ML reproducibility checklist: Appendix Section LABEL:sec:reproducibility-checklist. . The first one, named DP-snapshot, uses Dynamic Programming to compute the optimal actions w.r.t. the snapshot models at each decision epoch. The second one, named DP-NSMDP, uses the real NSMDP as a model to provide its optimal action. The latter behaves as an omniscient agent and should be seen as an upper bound on the performance. We choose a particular grid-world domain coined “Non-Stationary bridge” illustrated in Appendix, Section LABEL:sec:nsbridge-figure. An agent starts at the state labeled S in the center and the goal is to reach one of the two terminal states labeled G where a reward of +1 is received. The gray cells represent holes that are terminal states where a reward of -1 is received. Reaching the goal on the right leads to the highest payoff since it is closest to the initial state and a discount factor γ=0.9 is applied. The actions are 𝒜={Up, Right, Down, Left}. The transition function is stochastic and non-stationary. At decision epoch t=0, any action deterministically yields the intuitive outcome. With time, when applying Left or Right, the probability to reach the positions usually stemming from Up and Down increases symmetrically until reaching 0.45. We set the Lipschitz constant Lp=1. Aside, we introduce a parameter ϵ[0,1] controlling the behavior of the environment. If ϵ=0, only the left-hand side bridge becomes slippery with time. It reflects a close to worst-case evolution for a policy aiming to the left-hand side goal. If ϵ=1, only the right-hand side bridge becomes slippery with time. It reflects a close to worst-case evolution for a policy aiming to the right-hand side goal. In between, the misstep probability is proportionally balanced between left and right. One should note that changing ϵ from 0 to 1 does not cover all the possible evolutions from MDPt0 but provides a concrete, graphical illustration of RATS’s behavior for various possible evolutions of the NSMDP.

We tested RATS with dmax=6 so that leaf nodes in the search tree are terminal states. Hence, the optimal risk-averse policy is applied and no heuristic approximation is made. Our goal is to demonstrate that planning in this worst-case NSMDP allows to minimize the loss given any possible evolution of the environment. To illustrate this, we report results reflecting different evolutions of the same NSMDP using the ϵ factor. It should be noted that, at t=0, RATS always moves to the left, even if the goal is further, since going to the right may be risky if the probabilities to go Up and Down increase. This corresponds to the careful, risk-averse, behavior. Conversely, DP-snapshot always moves to the right since MDP0 does not capture this risk. As a result, the ϵ=0 case reflects a favorable evolution for DP-snapshot and a bad one for RATS. The opposite occurs with ϵ=1 where the cautious behavior dominates over the risky one, and the in-between cases mitigate this effect.

In Figure 3(a), we display the achieved expected return for each algorithm as a function of ϵ, i.e. as a function of the possible evolutions of the NSMDP. As expected, the performance of DP-snapshot strongly depends on this evolution. It achieves high return for ϵ=0 and low return for ϵ=1. Conversely, the performance of RATS varies less across the different values of ϵ. The effect illustrated here is that RATS maximizes the minimal possible return given any evolution of the NSMDP. It provides the guarantee to achieve the best return in the worst-case. This behavior is highly desirable when one requires robust performance guarantees as, for instance, in critical certification processes.

(a) Discounted return vs ϵ, 50% of standard deviation.

(b) Discounted return distributions ϵ{0,0.5,1}.
Figure 4: Discounted return of the three algorithms for various values of ϵ.

Figure 3(b) displays the return distributions of the three algorithms for ϵ{0,0.5,1}. The effect seen here is the tendency for RATS to diminish the left tail of the distribution corresponding to low returns for each evolution. It corresponds to the optimized criteria, i.e. robustly maximizing the worst-case value. A common risk measure is the Conditional Value at Risk (CVaR) defined as the expected return in the worst q% cases. We illustrate the CVaR at 5% achieved by each algorithm in Table 2. Notice that RATS always maximizes the CVaR compared to both DP-snapshot and DP-NSMDP. Indeed, even if the latter uses the true model, the optimized criteria in DP is the expected return.

7 Conclusion

We proposed an approach for robust planning in non-stationary stochastic environments. We introduced the framework of Lipchitz Continuous Non-Stationary MDPs (NSMDPs) and derived the Risk-Averse Tree-Search (RATS) algorithm, to predict the worst-case evolution and to plan optimally w.r.t. this worst-case NSMDP. We analyzed RATS theoretically and showed that it approximates a worst-case NSMDP with a control parameter that is the depth of the search tree. We showed empirically the benefit of the approach that searches for the highest lower bound on the worst achievable score. RATS is robust to every possible evolution of the environment, i.e. maximizing the expected worst-case outcome on the whole set of possible NSMDPs. Our method was applied to the uncertainty on the evolution of a model. Generally, it could be extended to any uncertainty on the model used for planning, given bounds on the set of the feasible models. The purpose of this contribution is to lay a basis of worst-case analysis for robust solutions to NSMDPs. As is, RATS is computationally intensive and scaling the algorithm to larger problems is an exciting future challenge.

Acknowledgments

This research was supported by the Occitanie region, France.

References