Provably efficient reconstruction of policy networks

1 Introduction

In the reinforcement learning (RL) framework, the goal of a rational agent consists in maximizing the expected rewards in a dynamical system by finding a suitable conditional distribution known as policy. This conditional distribution can be found using policy iteration and any suitable function approximator, ranging from linear models to neural networks. Due to their high representation power and efficient scalability, neural networks are often used for this matter (Lillicrap et al., 2015; Schulman et al., 2015). They have shown great successes in simulated environments such as games (e.g. Atari) or robotic control (e.g. MuJoCo).

However, in situations with limited memory capacity, e.g. deployment on mobile devices (Liu et al., 2018) or field drones, finding a lower-dimensional embedding of a policy network with bounds on its performance and error rate is challenging. A typical policy network can contain redundant parameters that must be pruned to save storage, a process for which knowledge distillation (Rusu et al., 2015) is not well-suited due to lack of guarantees. Other parameters reduction attempts include Mazoure et al. (2019); Doan et al. (2019), but authors do not provide theoretical guarantees or interpretations for this improvement.

It is in fact crucial to have theoretical guarantees when conducting policy reconstruction, especially for real world applications. Unlike simulated environments, where the agent can explore freely and interact without incurring unexpected consequences, such a setup is not applicable to real life scenarios. In multiple risk-sensitive domains, failures can lead to drastic damages either in term of human lives (robotic surgery (Ubee et al., 2019), self driving car (Shalev-Shwartz et al., 2017), air traffic control (Kim, 2015; Moon et al., 2011)) or financial consequences (warehouse robots (Bogue, 2017), advertisement campaign). Before deploying the compressed models, one might request performance guarantees on the reconstructed policy’s performance (Berkenkamp et al., 2017). In this case, black-box methods such as neural networks are disadvantaged due to the lack of theoretical guarantees, therefore are unlikely to be applied to real-world scenarios (Ghavamzadeh et al., 2016).

In this paper, we address two drawbacks of policies represented by neural networks: lack of (1) performance guarantees and (2) control over which redundant parameters are pruned. We treat the policy embedding problem as finding a low-dimensional representation of a given density function over state action space, and propose a general framework for representing any given policy with provable guarantees. We show that the performance of the reconstructed policy depends on the projection algorithm, as well as on the policy’s long-term behaviour. To the best of our knowledge, this is the first work proposing a general framework to learn a compressed policy with performance and long-term behaviour guarantees.

The experiments section explores various policy representation methods based on our framework, which are compared on average returns as well as state coverage. More precisely, we perform experiments with a wide range of different types of basis, including orthogonal basis such as singular value decomposition (SVD,

Halko et al. 2011) and non-orthogonal basis such as Gaussian mixtures (GMM, Rasmussen 2000

) which can be directly learned from data. In addition, we evaluate the performance of pre-defined orthogonal basis such as Fourier transformations (FFT,

Stein & Shakarchi 2003) and Daubechies wavelet (DB4, Daubechies 1988). Our empirical results suggest that the framework we propose is general enough to handle various classes of RL problems, whether the basis needs to be learned or arises naturally.

2 Preliminary

In this section, we provide a brief introduction to reinforcement learning, density approximation, as well as singular value decomposition.

2.1 Reinforcement learning

We consider a problem modeled by a discounted Markov Decision Process (MDP)

M := ⟨ S, A, r, P, S_{0}, γ ⟩

, where

S

is the state space;

A

is the action space; given states

s

s^{'} \in S

, action

a \in A

P (s^{'} | s, a)

is the transition probability of transferring

s

s^{'}

under the action

a

and

r (s, a)

is the reward collected at state

s

after executing the action

a

;

S_{0} \subseteq S

is the set of initial states and

γ

is the discount factor.

The agent in RL environment often execute actions according to some policies. We define a stochastic policy $π : S \times A \to [0, 1]$ such that $π (a | s)$ is the conditional probability that the agent takes action $a \in A$ after observing the state $s \in S$ . The objective is to find a policy $π$ which maximizes the expected discounted reward:

η (π) = E_{π} [\infty \sum t = 0 γ^{t} r (s_{t}, a_{t})]

(1)

We define the state value function $V_{π}$ and the state-action value function $Q_{π}$ as:

	$V_{π} (s)$	$= E_{π} [\infty \sum k = t γ^{k - t} r (s_{k}, a_{k}) ∣ ∣ ∣ s_{t} = s],$
	$Q_{π} (s, a)$	$= E_{π} [\infty \sum k = t γ^{k - t} r (s_{k}, a_{k}) ∣ ∣ ∣ s_{t} = s, a_{t} = a]$

The gap between $Q_{π}$ and $V_{π}$ is known as the advantage function:

A_{π} (s, a)

= Q_{π} (s, a) - V_{π} (s),

where $a \sim π (\cdot | s)$ and $s^{'} \sim P (\cdot | s, a)$ . The coverage of policy $π$ is defined as the stationary state visitation probability $ρ_{π} (s) = P (s_{t} = s)$ under $π$ .

2.2 Function approximation in inner product spaces

The theory of inner product spaces has been widely used to characterize approximate learning of Markov decision processes (Puterman, 2014; Tsitsiklis & Van Roy, 1999). In this subsection, we provide a short overview of inner product spaces of square-integrable functions on a closed interval $I$ denoted $L^{2} (I)$ , and mention drawbacks of working purely in the continuous setting.

The $L^{2} (I)$ space is known to be separable for $I = [0, 1]$ , i.e., it admits a countable orthonormal basis of functions ${ω_{k}}_{k = 1}^{\infty}$ , such that $⟨ ω_{i}, ω_{j} ⟩ = δ_{i j}$ for all $i, j$ and $δ$ being Kronecker’s delta. This property makes possible computations with respect to the inner product

⟨ f, g ⟩ = \int_{t \in I} f (t) ¯ g (t) d t,

(2)

and corresponding norm $| | f | | = \sqrt{⟨ f, f ⟩}$ , for $f, g \in L^{2} (I)$ .

That is, for any $f \in L^{2} (I)$ with basis ${ω_{k}}_{k = 1}^{\infty}$ , there exist scalars ${ξ_{k}}_{k = 1}^{\infty}$ such that

^f (t) = \infty \sum k = 1 ξ_{k} ω_{k} (t) .

(3)

where the coefficients ${ξ_{k}}_{k = 1}^{\infty}$ are given by

ξ_{k} = ⟨ f, ω_{k} ⟩ .

(4)

If ${^f}_{K} (t) = \sum_{k = 1}^{K} ξ_{k} ω_{k} (t)$ , then adding $ξ_{K + 1} ω_{K + 1}$ to ${^f}_{K}$

can be thought of as adding ”non-redundant” information to the previous estimate.

Eq. (3) suggests a simple compression rule, where one would project a density function onto a fixed basis and store only the first ${ξ_{k}}_{k = 1}^{K}$

coefficients. Which components to pick will depend on the nature of the basis: harmonic and wavelet bases are ranked according to amplitude, while singular vectors are sorted by corresponding singular values.

The choice of the basis ${ω_{k}}_{k = 1}^{\infty}$ plays a crucial role in the quality of approximation and should be taken while keeping in mind properties of the compressed function such as smoothness and periodicity. Below, we provide some examples of well-known orthonormal bases of $L^{2} (I)$ :

Fourier: $ω_{k} (t) = e^{2 π i k t}$ ;
Haar: $ω_{k, k^{'}} (t) = 2^{k / 2} ω (2^{k} t - k^{'})$ ;

In order to allow our learning framework to have strong convergence results in the inner product sense as well as a countable set of basis functions, we restrict ourselves to separable Hilbert spaces.

While convergence guarantees are mostly known for the closed interval $I = [0, 1]$ , previous works have studied properties of the above functions on the whole real line and in multiple dimensions (Egozcue et al., 2006). Weaker convergence results in Hilbert spaces can be stated with respect to the space’s inner product. If, for every $g \in L^{2} (I)$ ,

lim n \to \infty ⟨ f_{n}, g ⟩ \to ⟨ f, g ⟩,

(5)

then the sequence $f_{n}$ is said to converge weakly to $f$ . Stronger theorems are known for specific bases but require stronger assumptions on $f$ . Some are presented in Section 2.3.

2.3 Universal density approximators

Universal density approximators are a class of models which, given enough parameters, can approximate any smooth density up to an arbitrary error level. Over the separable Hilbert space of square-integrable real functions, such approximations typically rely on a countable set of orthonormal elementary functions and include but are not limited to Fourier, Hermite, Haar and Daubechies bases (Stein & Shakarchi, 2003; Haar, 1909; Daubechies, 1988). Moreover, it is known that a mixture model with countable number of components is a universal density approximator (McLachlan & Peel, 2004); in such case, the basis consists of Gaussian density functions and is not necessarily orthonormal.

Our approach for policy compression can be summarized in the following steps: (1) pick a set of known basis over the space, (2) project the policy onto the first $K$ basis functions and (3) store the vector of projection weights and optionally the corresponding basis functions.

Below, we mention existing results for the $K^{th}$ Fourier partial sum, as well as finite $K -$ component smooth mixture model.

Rate of convergence of Fourier approximation

Let $π_{K}^{Fourier}$ denote the density approximated by the first $K^{th}$ Fourier partial sums (Stein & Shakarchi, 2003), then a result from (Jackson, 1930) shows that

| | π - π_{K}^{Fourier} | |_{1} = O (K^{- 1}) .

(6)

More recent results (Giardina & Chirlian, 1972) provide even tighter bounds for continuous and periodic functions with $m - 1$ continuous derivatives and bounded $m^{th}$ derivative. In such case,

| | π - π_{K}^{Fourier} | |_{\infty} = O (K^{- (2 m + 1)}) .

(7)

Rate of convergence of mixture approximation

Let $π_{K}^{MM}$ denote a (finite) $K -$ component mixture model. A result from (Rakhlin et al., 2005) shows the following result for $π_{K}^{MM}$ in the class of mixtures with $K$ marginally-independent scaled density functions and $π$

in the class of lower-bounded probability density functions:

K L (π | | π_{K}^{MM}) = O (K^{- 1} + n^{- 1}),

(8)

where $n$ is the number of i.i.d. samples used in the learning of $π_{K}^{MM}$ .

The constants hidden inside the big-O notation depend on the nature of the function and can become quite large, which can explain differences in empirical evaluation.

2.4 Low-rank matrix factorization

Matrix decomposition algorithms such as the truncated singular value decomposition (SVD) are popular compression methods due to their simplicity and robustness. Although SVD is commonly used in neural network compression schemes (Lu et al., 2017; Goetschalckx et al., 2018)

, most works apply it on the weights rather than the outputs of the function. The latter makes the process easy to implement, but error guarantees are harder to derive due to the non-linear activation functions of neural networks. Similarly to the density approximators above, a key appeal of SVD is that the error rate of truncation can be controlled efficiently through the magnitude of the truncated singular values

^*^**see Eckart-Young-Mirsky theorem.

2.5 Time complexity comparison

The time complexity for a $k$ -truncated SVD decomposition of a matrix $A \in R^{n \times m}$ is $O (m n k)$ as discussed in Du et al. (2017). Fast Fourier transform can be done in $O (m n log (m n))$ for the 2-dimensional case (Lohne, 2017). Hence, SVD is expected to be faster whenever $k < log (n m)$ . The discrete wavelet transform’s (DWT) time complexity depends on the choice of mother wavelets. When using Mallat’s algorithm, the 2-dimensional DWT is known to have complexities ranging from $O (n^{2} log n)$ to as low as $O (n^{2})$ , for a square matrix of size $n \times n$ and depending on the choice of mother wavelet (Resnikoff et al., 2012). Finally, we expect FFT and Daubechies wavelets to have less parameters than SVD, since the former use pre-defined orthonormal bases, while the later must store the left and right singular vectors.

3 Reconstruction algorithm

In this section, we illustrate our method in a single state case setting (i.e. bandit), before generalizing to multiple states.

3.1 Bandit example

Consider the special case of a multi-armed bandit over $A$ with reward function $r : A \to R$ and some given policy $π$ . The average rewards collected by $π$ are a simplification of Eq. (1):

η (π) = E_{π} [r (a)]

(9)

If $A$ is finite, $π$ is a vector which can be approximated with

{^π}_{K} (t) = K \sum k = 1 ξ_{k} ω_{k} (t), K \leq | A | .

(10)

The compression algorithm should therefore pick the $K$ coefficients which minimize the projection error onto $ω$ .

In an empirical setting, one can monitor the difference between average actions $a$ with respect to ground truth and approximate policies, which is upper-bounded by 1-Wasserstein distance (Dudley, 2018):

\begin{matrix} ∣ ∣ ∣ E_{a \sim π} [a] - E_{a \sim^π} [a] ∣ ∣ ∣ & = ∣ ∣ ∣ E_{u \sim U (0, 1)} [Q_{a} (u) - {^Q}_{a} (u)] ∣ ∣ ∣ \leq E_{u \sim U (0, 1)} [∣ ∣ ∣ Q_{a} (u) - {^Q}_{a} (u) ∣ ∣ ∣] = W_{1} (Π,^Π), \end{matrix}

(11)

where $Q_{a} (p) = inf {a \in A | p \leq Π_{a} (a)}$

is the quantile function and

Π_{a} (a) = P [A \leq a]

is the cumulative distribution function of the action random variable

A

. Therefore, it is sufficient to keep track of

W_{1} (Π,^Π)

, which can be efficiently computed for discrete distributions.

An extension of the multi-armed bandit problem to continuous action spaces implies that the probability of pulling each arm is given by the function $π (a) = {lim}_{h \to 0} \int_{a}^{a + h} P [A = t] d t$ . The extension of discrete quantities to their continuous counterparts is discussed in the next section.

3.2 Discretization of continuous policies

Consider the case when $f$ is a continuous, positive and decreasing function with a discrete sequence $a_{n} = f (n)$ . For example, the reconstruction error $W_{1} (Π,^Π)$ as well as difference in returns $| η (π) - η (^π) |$ fall under this family. Then, $\int_{0}^{\infty} f (t) d t < \infty$ implies that $\sum_{i = 0}^{\infty} a_{n} < \infty$ . Under these assumptions, convergence guarantees (e.g. on monotonically decreasing reconstruction error) in continuous space imply convergence in the discrete (empirical) setting. Hence, we operate on discrete spaces rather than continuous ones.

We now discuss the construction of $⌈ π ⌉$ , the discretized policy. The core idea consists in grouping together similar state-action pairs (in term of visitation frequency). To this end, we suggest using the quantile state visitation function $Q_{S}$ and the state visitation distribution function $Π$ , as well as their empirical counterparts, ${^Q}_{S}$ and ${^Π}_{S}$ . Quantile and distribution functions for the action space are defined analogously.

Algorithm 1 outlines the proposed discretization process allowing one to approximate any continuous policy (e.g., computed by a neural network) by a 2-dimensional table indexed by discrete states and actions. Binning is done via quantile discretization in order to have maximal resolution in frequently visited areas. It also allows for faster sampling during rollouts, since the probability of falling in each bin is uniform. The number of state and action bins per dimension is denoted $b_{S}$ and $b_{A}$ , respectively.

Input: Policy $π$ , number of state, action bins $b_{S}, b_{A}$

Result: Discrete policy $⌈ π ⌉$

Collect a set of states $S \subseteq S$ and set of actions $A \subseteq A$ via rollouts of $π$

Build the empirical c.d.f. ${^Π}_{s}$ from set $S$

Build the empirical c.d.f. ${^Π}_{a}$ from set $A$

Find numbers $i = i_{1}, . ., i_{b_{S}}$ s.t. ${^Π}_{s} (i_{l}) - {^Π}_{s} (i_{l - 1}) = \frac{1}{b_{S}}$ using ${^Q}_{s}$ for all $s \in S, l = 1, . ., b_{S}$

Find numbers $j = j_{1}, . ., j_{b_{A}}$ s.t. ${^Π}_{a} (j_{l}) - {^Π}_{a} (j_{l - 1}) = \frac{1}{b_{A}}$ using ${^Q}_{a}$ for all $a \in S, l = 1, . ., b_{A}$

if $i_{l - 1} \leq s \leq i_{l}$ and $j_{l^{'} - 1} \leq a \leq j_{l^{'}}$ then

Assign

(s, a)

(l, l^{'})

end if

Set

{⌈ π ⌉}_{l l^{'}}

π (\frac{i_{l^{'} - 1} + i_{l^{'}}}{2}, \frac{i_{l - 1} + i_{l}}{2})

Algorithm 1 Policy discretization

All further computations of distance between two discretized policies will have to be computed over the corresponding discrete grid $(i, j), i = 1, . ., b_{S}, j = 1, . ., b_{A}$ . Similarly to Eq. 11, one can look at the average action taken by two MDP policies at state $s$ :

(12)

This relationship is one of the motivations behind using $W_{1}$ to assess goodness-of-fit in the experiments section. Moreover, since the support is finite and discrete, $W_{1}$ computation can be done efficiently (Rubner et al., 1998).

3.3 Pruning unvisited states

As often some states might be never visited and one might wonder how much loss in performance is incurred by pruning those states. Let $N (s)$ denote the number of times $π$ visits state $s$ and $N (s, a)$ the number of times action $a$ is taken from $s$ over $N$ trajectories. It can happen that $N (s) = 0$ and $s$ is never visited. Hence, some rarely visited state-action pairs are not required and can be pruned in order to reduce the number of parameters in the discretized policy. If, at inference time, one of these pruned states is encountered, we can simply act uniformly at random, knowing that the performance will lightly be affected.

This leads us to define a pruned $π_{p r u n e d}$ policy as one which acts randomly under some state visitation threshold. Precisely

π_{pruned} (a | s) = ⎧ ⎪ ⎨ ⎪ ⎩ \begin{matrix} \frac{N (s, a)}{N (s)}, & if N (s) > 0 \frac{1}{b_{A}}, & otherwise \end{matrix}

(13)

Pruning rarely visited states can drastically reduce the number of parameters, while maintaining high probability performance guarantees for $π_{pruned}$ .

Theorem 3.1.

(Policy pruning) Let $N$ be the number of rollout trajectories of the policy $π$ , $r_{M}$ be the largest reward. With high probability $1 - 2 δ$ , we have a guarantee on the performance of the reconstructed policy $π_{pruned}$ :

∣ ∣ ∣ ∣ ∣ ∣ η (π) - η (π_{pruned}) ∣ ∣ ∣ {∣ ∣ ∣}_{1} \leq \frac{2 r_{M}}{1 - γ} \sqrt{\frac{3 | S | | A | + 4 log \frac{1}{δ}}{2 N}}

(14)

The proof is shown in Theorem 2 of (Simão et al., 2019). The theorem allows us to discretize only visited states and still ensuring a strong performance guarantee with high probability.

3.4 Acting with tabular policies

In order to execute the learned policy, one has to be able to (1) generate samples conditioned on a given state and (2) evaluate the log-probability of state-action pairs. Acting with a tabular density can be done via various sampling techniques. For example, importance sampling of the atoms corresponding to each bin is possible when $b_{A}$

is small, while the inverse c.d.f. method is expected to be faster in larger dimensions. The process can be further optimized on a case-per-case basis for each method. For example, sampling directly from the real part of a characteristic function can be done with the algorithm defined in

(Devroye, 1986)

Furthermore, it is possible to use any of the above algorithms to jointly sample $(s, a)$ pairs under the assumption that states were discretized by quantiles. First, uniformly sample a state bin and then use any of the conditional sampling algorithms to sample action $a$ . Optionally, one can add uniform noise (clipped to the range of the bin) to the sampled action. This naive trick would transform discrete actions into continuous approximations of the policy network.

3.5 Method

We propose the following algorithm which projects the function described by a given neural network onto some inner product space:

Input: Policy $π$ , number of components $K$ , basis $ω = ω_{1}, ω_{2}, . ., ω_{K}$

Result: A set of coefficients $ξ_{1}, . . ., ξ_{K}$

Rollout $π$ in the environment to estimate ${^Q}_{s}, {^Q}_{a}$

Discretize $π$ into $⌈ π ⌉$ using Alg. 1

Prune $⌈ π ⌉$ with ${^Q}_{s}, {^Q}_{a}$ using Eq. 13

$ξ_{1}, . ., ξ_{K} \leftarrow$ Project $⌈ π ⌉$ onto $ω$ using the corresponding algorithm

Algorithm 2 Joint policy reconstruction

Different projection algorithms are expected to perform differently depending on the curvature of the policy. For example, if $π$

is a Gaussian distribution for each state

s

in a state space of dimension

K

, then the table

⌈ π ⌉

will have exactly

K

modes. Moreover, methods decomposing signals in probability space such as FFT and SVD should be able to find structural patterns in the probability domain.

As a rule of thumb, if $π$ contains few distinct modes, state-action space methods (e.g. mixture models) are expected to perform better than probability space methods.

4 Theoretical guarantees

In this section, we provide bounds controlling the impact of the proposed approximation method on collected rewards and policy coverage.

4.1 Performance bound (single state case)

Storing an approximation of the ground truth policy implies a trade-off between performance and model size. Our framework allows to control the difference in collected reward as a function of reconstruction error in probability space.

Lemma 4.1.

Let $π_{1}, π_{2}$ be policies in $L^{2} (A)$ , $r : A \to R$ be the bandit’s reward function. Let $δ_{q} \geq 0$ be such that $| | π_{1} - π_{2} | |_{1} < δ_{q}$ and $p, q > 0$ s.t. $\frac{1}{p} + \frac{1}{q} = 1$ . Then

| η (π_{1}) - η (π_{2}) | < | | r | |_{p} δ_{q}

Lemma 4.1 (whose proof can be found in appendix) guarantees that rewards collected by $π_{2}$ will be no further than $| | r | |_{p} δ_{q}$ away from rewards of $π_{1}$ .

4.2 Performance bounds (multi-state case)

Similar guarantees can be obtained when the state space is not reduced to a single state.

Theorem 4.2.

(Multi-state policy reconstruction) Let $π_{1}, π_{2}$ be square-summable policies. The following holds

∣ ∣ ∣ η (π_{2}) - η (π_{1}) ∣ ∣ ∣ \leq C \cdot K L_{M A X} (π_{1} | | π_{2}) + ϵ

(15)

where $ϵ = {max}_{s, a} | A_{π_{1}} (s, a) |$ , $A_{π_{1}}$ is the advantage function, $C = \frac{4 γ ϵ}{(1 - γ)^{2}}$ and $K L_{M A X} (π_{1} | | π_{2}) = {max}_{s} K L (π_{1} (. | s) | | π_{2} (. | s))$ .

The proof can be found in the Appendix and relies on techniques found in (Schulman et al., 2015). A similar relation is presented in the next section for stationary distributions of $π$ .

A key technical difference in the multi-state setting is that the discretization of the state space and pruning of the policy must be taken into account. Consequently, the performance error is now decomposed by triangle inequality into the discretization error (Eq. (12)), the pruning error (Thm 3.1) and the projection error which is controlled by specific function approximation theorems:

	$∣ ∣ ∣ η (π) - η (^π)$	$∣ ∣ ∣ \leq ∣ ∣ ∣ η (π) - η (⌈ π ⌉) ∣ ∣ ∣ +$		(16)
		$∣ ∣ ∣ η (⌈ π ⌉) - η (π_{p r u n e d}) ∣ ∣ ∣ + ∣ ∣ ∣ η (π_{p r u n e d}) - η (^π) ∣ ∣ ∣$

Loose bounds on the reconstruction error are given in Eq. (6) and Eq. (8), tighter versions can be found in the literature (Giardina & Chirlian, 1972; Rakhlin et al., 2005).

To estimate the total error, one can separately estimate each of the three terms of Eq. (16). Empirical results of theses three error terms are shown in the next section.

4.3 Connection between reconstruction error and policy coverage

A useful metric to assess the long-term behaviour of a given policy in an MDP is the stationary distribution $ρ_{π}$

of the Markov chain induced by

π

. For a given policy,

ρ_{π}

measures the state coverage of the model averaged with respect to that policy.

If one wishes to switch between the MDP and Markov chain frameworks for a fixed policy, it is possible to define the expected transition model $(P_{π})_{s s^{'}} = \sum_{a \in A} π (a | s) P (s^{'} | s, a)$

. In tensor form, it can be represented as

A (T_{(2)} Π)

, where

Π_{a s} = π (a | s)

is the policy matrix,

A = I ⊙ I

is the Khatri-Rao product and

T_{(2)}

is the second matricization of the transition tensor

T_{s a s^{'}} = P (s^{'} | = s, a)

If the Markov chain defined above is irreducible and homogeneous, then its stationary distribution corresponds to the state occupancy probability^†^††The existence and uniqueness of $ρ$ follow from the Perron-Frobenius theorem.

given by the principal left eigenvector of

P_{π}

The following theorem bridges the error made on the reconstruction of long-run distribution as a function of the system’s transition dynamics and policy reconstruction error.

Theorem 4.3 (Approximate policy coverage).

Let $| | \cdot | |_{S_{p}}$ be the Schatten $p$ -norm and $| | \cdot | |_{p}$ be the vector $p$ -norm. If ( $P_{π}$ , $| S |^{- 1} 1_{| S |}$ ) is an irreducible, homogeneous Markov chain, then the following holds:

| | ρ_{π}^{⊤} - ρ_{~ π}^{⊤} | |_{S_{1}} \leq | | Z | |_{S_{\infty}} | | ~ Π - Π | |_{S_{1}} | | T_{(3)} | |_{S_{2}},

(17)

where $Z = (I - P_{π} + 1_{| S |} ρ_{π}^{⊤})^{- 1}$ and $1_{| S |}$ is a vector of all ones.

A detailed proof can be found in the Appendix.

Note that the upper bound depends on both the environment’s structure as well as on the policy reconstruction quality. It is thus expected that, for MDPs with particularly large singular values of $T_{(2)}$ , the bound converges less quickly than for those with smaller singular values. A visualization of this bound is provided in Figure 1.

5 Experimental results

In this section, we visualize the rate of convergence of a reconstructed stationary distribution via a toy example to illustrate Eq. (17). We then assess the validity of our method on several environments and compare the quality of the reconstructed policy for different projection methods, in terms of performance as well as computation time.

5.1 Rate of convergence of stationary distribution under random policy

We validate the rate of convergence of Thm. 4.3 in the toy experiment described below. Consider a deterministic chain MDP with $N$ states. A fixed policy in state $i$ transitions to $i + 1$ with probability $α$ and to $i - 1$ with probability $1 - α$ . If in states $1$ or $N$ , the agent remains there with probability $1 - α$ and $α$ , respectively. We consider the expected transition model $P_{π}$ obtained using the policy reconstructed through discrete Fourier transform. A visualization of Theorem 4.3 is shown in Fig. 1.

Figure 1: Upper bound on $| | ρ_{π} - ρ_{^π} | |$ in the chain MDP task as a function of number of Fourier components and of the number of states.

5.2 Control domains

We consider a range of reconstruction tasks from three environments: a toy bandit problem inspired by Fellows et al. (2018), as well as the classical control problems of ContinuousMountainCar-v0 and Pendulum-v0 from OpenAI Gym (Brockman et al., 2016).

In all environments but the first, we omit GMM and fixed basis GMM methods, since the expectation-maximization algorithm runs into stability problems when dealing with extremely high number of components. Instead, we use SVD for low-rank matrix factorization and the fourth order Daubechies wavelet

(Daubechies, 1988) (DB4) as an orthonormal basis alternative to Fourier (DFT). We use

b_{S} = 35

and

b_{A} = 15

as initial estimates for the discretization for all experiments except for Mountain Car (

b_{A} = 10

5.2.1 Bandit turntable

We consider a bandit problem in which rewards are spread in a circle as shown in Figure 2 a). Actions consist of angles in the interval $[- π, π]$ . Given the corresponding Boltzmann policy, we compare the reconstruction quality of the discrete Fourier transform and GMM. In addition to Fourier and GMM, we also compare with fixed basis GMM, where the means are sampled uniformly over $[- π, π]$

and variance is drawn uniformly from

{0.1, 0.5, 1, 2}

; only mixing coefficients are learned. This provides insight into whether the reconstruction algorithms are learning the policy, or if

π

is so simple that a random set of Gaussian can already approximate it well.

Figure 2: Bandit turntable environment in (a) polar coordinates. Intensity of modes is proportional to the reward. Actions are absolute angles in interval $[- π, π]$ . b) shows the $W_{1}$ distance between ground truth and order $K$ approximations. Dotted line indicates the number of modes in the ground truth density. While GMM shows instability with more components, Fourier transform achieves low $W_{1}$ with only a few bases.

Results are reported in Figure 2 b), where the $W_{1}$ metric was obtained by first computing the row-wise (i.e. action-specific) scores $W_{1} (π (\cdot | s),^π (\cdot | s))$ for every state, and subsequently averaging over all possible states. As $K$

increases, EM has more trouble converging, as can be seen in the green curve’s trend and confidence intervals from

K = 60

to 100. Fixed basis GMM struggles to accurately approximate the policy. On the other hand, DFT shows a stable performance after a threshold of

K = 10

components.

5.2.2 Mountain Car

In this environment, the agent (a car) must reach the flag located at the top of a hill. It needs to go back and forth in order to generate sufficient momentum to reach the top. The agent is allowed to apply a speed motion in the interval $[- 1, 1]$ . We use Soft Actor-Critic (SAC, Haarnoja et al. 2018) to train the agent until convergence ( $25 k$ steps).

We compare all reconstruction methods with a maximum likelihood estimate (MLE) of $π$ , which consists of approximating the discrete policy with $K$ samples of that discrete policy. This method is referred as ”oracle” in the plots, since no reconstruction is performed. As shown in Figure 3, DFT, SVD and DB4 reach same reconstruction error as the oracle method. All methods achieve good performance with one order of magnitude less than the neural network in term of parameters. Note that DB4 shows slightly better performance than DFT.

Figure 3: Difference in returns between reconstructed and ground truth policies for the Mountain Car task with respect of the number of parameters used (X-axis). Our reconstructed models used drastically less than neural network parameters.

5.2.3 Pendulum

This classical mechanic task consists of a pendulum which needs to swing up in order to stay upright. The actions represent the joint effort (torque) between $- 2$ and $2$ units. We train SAC until convergence and save snapshots of the actor and critic after $2 k$ and $5 k$ steps. The reconstruction task is to recover both the Gaussian policy (actor) and the Boltzmann-Q policy (critic, temperature set to 1). While the Gaussian policy is unimodal, the Boltzmann-Q policy can be multi-modal and is a good challenge to demonstrates properties of each method. As shown in Figure 5, when policy has converged to a spiky shape ( $5, 000$ steps), all methods show comparable performance (in terms of convergence). DFT shows better convergence at early stages of training ( $2, 000$ steps), that is when the ground truth policy has a large variance. Note that all methods use less parameters than the original neural network.

Figure 4: Plots of Gaussian $Π$ after $5 k$ training steps in polar coordinates. Each circle of different radius represents a states in the Pendulum environment as shown by the snapshots. Higher intensity colors (red) represent higher density mass on the given angular action.

Figure 5: Absolute difference in returns collected by discretized and reconstructed policies, averaged over 5 trials. Blue dots represent the number of parameters of the neural network policy. SVD, DFT and DB4 projections need an order of magnitude less in term of parameters to reconstruct the original policy.

Figure 6: $W_{1}$ distance between the true and reconstructed stationary distributions, averaged over $5$ trials. SVD, DFT and DB4 methods show a fast convergence to the oracle’s stationary distribution using only $10 %$ of the neural network’s parameters.

Figure 7: Plots of discretized and pruned (a) Gaussian policy and (b) Boltzmann-Q policy in polar coordinates for $5$ thousands training steps of soft actor-critic. Rewards collected by their respective continuous networks are indicated in parentheses. Each circle of radius $s$ corresponds to $π (\cdot | s)$ for a discrete $s = 1, 2, . ., 750$ . All densities are on the same scale.

Figure 8: Plots of discretization and pruning errors for (a) the Gaussian policy of ContinuousMountainCar-v0 and (b) Boltzmann-Q policy of Pendulum-v0. As expected, more bins induce lower discretization error (green curve), while collecting more trajectories guarantees confidence for pruning unvisited states (blue curve).

Results for the Boltzmann-Q policy presented in the Appendix exhibit patterns similar to Gaussian’s. The discretization and pruning errors are illustrated in Figure 8 and agree with error bounds of Thm 3.1.

6 Conclusion

In this work, we introduced a general framework for provably efficient policy reconstruction. It allowed to drastically compress policy densities in our experiments by a projection onto $K$ basis functions. Moreover, the reconstruction error has been shown to depend on the discretization, pruning and projection algorithms. We conducted experiments to demonstrate the behaviour of four sets of basis functions on a set of continuous control tasks, which exhibit desirable performance guarantees. A potential extension to our framework would be to directly operate in the continuous space, avoiding the discretization step procedure.

References

Baumgartner (2011) Baumgartner, B. An inequality for the trace of matrix products, using absolute values. arXiv preprint arXiv:1106.6189, 2011.
Berkenkamp et al. (2017) Berkenkamp, F., Turchetta, M., Schoellig, A., and Krause, A. Safe model-based reinforcement learning with stability guarantees. In Advances in neural information processing systems, pp. 908–918, 2017.
Bogue (2017) Bogue, R. Robots that interact with humans: a review of safety technologies and standards. Industrial Robot, 44:395–400, 2017.
Brockman et al. (2016) Brockman, G., Cheung, V., Pettersson, L., Schneider, J., Schulman, J., Tang, J., and Zaremba, W. Openai gym. CoRR, abs/1606.01540, 2016. URL http://arxiv.org/abs/1606.01540.
Cho & Meyer (2001) Cho, G. E. and Meyer, C. D. Comparison of perturbation bounds for the stationary distribution of a markov chain. Linear Algebra and its Applications, 335(1-3):137–150, 2001.
Daubechies (1988) Daubechies, I. Orthonormal bases of compactly supported wavelets. Communications on pure and applied mathematics, 41(7):909–996, 1988.
Devroye (1986) Devroye, L. An automatic method for generating random variates with a given characteristic function. In SIAM journal on applied mathematics, 1986.
Doan et al. (2019) Doan, T., Mazoure, B., Durand, A., Pineau, J., and Hjelm, R. D. Attraction-repulsion actor-critic for continuous control reinforcement learning. arXiv preprint arXiv:1909.07543, 2019.
Du et al. (2017) Du, S. S., Wang, Y., and Singh, A. On the power of truncated svd for general high-rank matrix estimation problems. In Advances in neural information processing systems, pp. 445–455, 2017.
Dudley (2018) Dudley, R. M. Real analysis and probability. Chapman and Hall/CRC, 2018.
Egozcue et al. (2006) Egozcue, J. J., Díaz-Barrero, J. L., and Pawlowsky-Glahn, V. Hilbert space of probability density functions based on aitchison geometry. Acta Mathematica Sinica, 22(4):1175–1182, 2006.
Fellows et al. (2018) Fellows, M., Ciosek, K., and Whiteson, S. Fourier Policy Gradients. In ICML, 2018.
Ghavamzadeh et al. (2016) Ghavamzadeh, M., Petrik, M., and Chow, Y. Safe policy improvement by minimizing robust baseline regret. In Advances in Neural Information Processing Systems, pp. 2298–2306, 2016.
Giardina & Chirlian (1972) Giardina, C. and Chirlian, P. Bounds on the truncation error of periodic signals. IEEE Transactions on Circuit Theory, 19(2):206–207, 1972.
Goetschalckx et al. (2018) Goetschalckx, K., Moons, B., Wambacq, P., and Verhelst, M. Efficiently combining svd, pruning, clustering and retraining for enhanced neural network compression. In Proceedings of the 2nd International Workshop on Embedded and Mobile Deep Learning
, pp. 1–6, 2018.
Haar (1909) Haar, A. Zur theorie der orthogonalen funktionensysteme. Georg-August-Universitat, Gottingen., 1909.
Haarnoja et al. (2018) Haarnoja, T., Zhou, A., Abbeel, P., and Levine, S. Soft actor-critic: Off-policy maximum entropy deep reinforcement learning with a stochastic actor. CoRR, abs/1801.01290, 2018. URL http://arxiv.org/abs/1801.01290.
Halko et al. (2011) Halko, N., Martinsson, P.-G., and Tropp, J. A. Finding structure with randomness: Probabilistic algorithms for constructing approximate matrix decompositions. SIAM review, 53(2):217–288, 2011.
Jackson (1930) Jackson, D. The theory of approximation. The American mathematical society, 1930.
Kim (2015) Kim, H.-Y.
Statistical notes for clinical researchers: Type i and type ii errors in statistical decision.
Restorative dentistry & endodontics, 40(3):249–252, 2015.
Lillicrap et al. (2015) Lillicrap, T. P., Hunt, J. J., Pritzel, A., Heess, N., Erez, T., Tassa, Y., Silver, D., and Wierstra, D. Continuous control with deep reinforcement learning. arXiv preprint arXiv:1509.02971, 2015.
Liu et al. (2018) Liu, S., Lin, Y., Zhou, Z., Nan, K., Liu, H., and Du, J. On-demand deep model compression for mobile devices: A usage-driven model selection framework. In Proceedings of the 16th Annual International Conference on Mobile Systems, Applications, and Services, pp. 389–400. ACM, 2018.
Lohne (2017) Lohne, M. The computational complexity of the fast fourier transform. Technical report, 2017.
Lu et al. (2017) Lu, Y., Kumar, A., Zhai, S., Cheng, Y., Javidi, T., and Feris, R. Fully-adaptive feature sharing in multi-task networks with applications in person attribute classification. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition
, pp. 5334–5343, 2017.
Mazoure et al. (2019) Mazoure, B., Doan, T., Durand, A., Hjelm, R. D., and Pineau, J. Leveraging exploration in off-policy algorithms via normalizing flows. CoRR, abs/1905.06893, 2019. URL http://arxiv.org/abs/1905.06893.
McLachlan & Peel (2004) McLachlan, G. and Peel, D. Finite mixture models. John Wiley & Sons, 2004.
Moon et al. (2011) Moon, W.-C., Yoo, K.-E., Choi, Y.-C., et al. Air traffic volume and air traffic control human errors. Journal of Transportation Technologies, 1(03):47, 2011.
Puterman (2014) Puterman, M. L. Markov decision processes: discrete stochastic dynamic programming. John Wiley & Sons, 2014.
Rakhlin et al. (2005) Rakhlin, A., Panchenko, D., and Mukherjee, S. Risk bounds for mixture density estimation. ESAIM: Probability and Statistics, 9:220–229, 2005.
Rasmussen (2000) Rasmussen, C. E.
The infinite gaussian mixture model.
In Advances in neural information processing systems, pp. 554–560, 2000.
Resnikoff et al. (2012) Resnikoff, H. L., Raymond Jr, O., et al. Wavelet analysis: the scalable structure of information. Springer Science & Business Media, 2012.
Rubner et al. (1998) Rubner, Y., Tomasi, C., and Guibas, L. J. A metric for distributions with applications to image databases. In Sixth International Conference on Computer Vision (IEEE Cat. No. 98CH36271), pp. 59–66. IEEE, 1998.
Rusu et al. (2015) Rusu, A. A., Colmenarejo, S. G., Gulcehre, C., Desjardins, G., Kirkpatrick, J., Pascanu, R., Mnih, V., Kavukcuoglu, K., and Hadsell, R. Policy distillation. arXiv preprint arXiv:1511.06295, 2015.
Schulman et al. (2015) Schulman, J., Levine, S., Abbeel, P., Jordan, M., and Moritz, P. Trust region policy optimization. In Bach, F. and Blei, D. (eds.), Proceedings of the 32nd International Conference on Machine Learning, volume 37 of Proceedings of Machine Learning Research, pp. 1889–1897, Lille, France, 07–09 Jul 2015. PMLR. URL http://proceedings.mlr.press/v37/schulman15.html.
Schweitzer (1968) Schweitzer, P. J. Perturbation theory and finite markov chains. Journal of Applied Probability, 5(2):401–413, 1968.
Shalev-Shwartz et al. (2017) Shalev-Shwartz, S., Shammah, S., and Shashua, A. On a formal model of safe and scalable self-driving cars. arXiv preprint arXiv:1708.06374, 2017.
Simão et al. (2019) Simão, T. D., Laroche, R., and des Combes, R. T. Safe policy improvement with an estimated baseline policy, 2019.
Stein & Shakarchi (2003) Stein, E. and Shakarchi, R. Fourier Analysis: An Introduction. Princeton University Press, 2003. ISBN 9780691113845. URL https://books.google.ca/books?id=I6CJngEACAAJ.
Tsitsiklis & Van Roy (1999) Tsitsiklis, J. N. and Van Roy, B. Optimal stopping of markov processes: Hilbert space theory, approximation algorithms, and an application to pricing high-dimensional financial derivatives. IEEE Transactions on Automatic Control, 44(10):1840–1851, 1999.
Ubee et al. (2019) Ubee, S. S., Selvan, M., Chandrashekar, R., and Cooke, P. Safety considerations for performing robotic surgery in the presence of a permanent pacemaker. Journal of Perioperative Practice, 29(7-8):242–246, 2019.

7 Appendix

Reproducibility Checklist

We follow the reproducibility checklist ”The machine learning reproducibility checklist” and point to relevant sections explaining them here.
For all algorithms presented, check if you include:

A clear description of the algorithm, see main paper and included codebase. The proposed approach is completely described by Alg. 2.
An analysis of the complexity (time, space, sample size) of the algorithm. The space complexity of our algorithm depends on the number of desired basis. It is $4 K$ for a 1-dimension $K -$ component GMM with diagonal covariance, $m K + K^{2} + n K$ for a $K -$ truncated SVD of an $m \times n$ matrix, and only $K$ for the wavelet and Fourier bases. Note that a simple implementation of SVD requires to find the three matrices $U, D, V^{⊤}$ before truncation.

The time complexity depends on the reconstruction, pruning and discretization algorithms, which involves conducting rollouts w.r.t. policy $π$ , training a quantile encoder on the state space trajectories, and finally embedding the the pruned matrix using the desired algorithm. We found that the Gaussian mixture is the slowest to train, due to shortcomings of the expectation-maximization algorithm.
A link to a downloadable source code, including all dependencies. The code is included with Supplemental Files as a zip file; all dependencies can be installed using Python’s package manager. Upon publication, the code would be available on Github. Additionally, we include the model’s weights as well as the discretized policy for Pendulum-v0 environment.

For all figures and tables that present empirical results, check if you include:

A complete description of the data collection process, including sample size. We use standard benchmarks provided in OpenAI Gym (Brockman et al., 2016).
A link to downloadable version of the dataset or simulation environment. Not applicable.
An explanation of how samples were allocated for training / validation / testing.
We do not use a training-validation-test split, but instead report the mean performance (and one standard deviation) of the policy at evaluation time across 5 trials.
An explanation of any data that were excluded. The only data exclusion was done during policy pruning, as outlined in the main paper.
The exact number of evaluation runs. 5 trials to obtain all figures, and 200 rollouts to determine $ρ_{π}$ .
A description of how experiments were run. See Section Experimental Results in the main paper and didactic example details in Appendix.
A clear definition of the specific measure or statistics used to report results. Undiscounted returns across the whole episode are reported, and in turn averaged across 5 seeds. Confidence intervals shown in Fig. 8 were obtained using the pooled variance formula from a difference of means $t -$ test.
Clearly defined error bars. Confidence intervals are always mean $\pm 1$ standard deviation over 5 trials.
A description of results with central tendency (e.g. mean) and variation (e.g. stddev). All results use the mean and standard deviation.
A description of the computing infrastructure used. All runs used 1 CPU for all experiments with $8$ Gb of memory.

7.1 Discretization techniques

Naively discretizing some function $f$ over a fixed interval $[a, b]$ can be done by $n$ equally spaced bins. To pass from a continuous value $f (x)$ to discrete, one would find $i \in 1, . ., n$ such that $\frac{b - a}{n} i \leq x \leq \frac{b - a}{n} (i + 1)$ . However, using a uniform grid for a non-uniform $f$ wastes representation power. To re-allocate bins in low density areas, we use the quantile binning method, which first computes the cumulative distribution function of $f$ , called $F$ . Then, it finds $n$ points such that the probability of falling in each bin is equal. Quantile binning can be see as uniform binning in probability space, and exactly corresponds to uniform discretization if $f$

is constant (uniform distribution).

Below, we present an example of discretizing 4,000 sample points taken from four

N (μ_{i}, 0.3)

distributions, using the quantile binning.

Figure 9: Quantile binning for the four Gaussians problem, using 10 bins per dimension.

In Figure 9, we are only allowed to allocate 10 bins per dimension. Note how the grid is denser around high-probability regions, while the furthermost bins are the widest. This uneven discretization, if applied to the policy density function, allows the agent to have higher detail around high-probability regions.
In Python, the function sklearn.preprocessing.KBinsDiscretizer can be used to easily discretize samples from the target density.

Figure 10: Quantile binning of unnormalized state visitation counts of DQN policy in the discrete Mountain Car task. Bins closer to the starting states are visited more often than those at the end of the trajectory. Blue dots represents the bins limits.

Since policy densities are expected to be more complex than the previous example, we analyze quantile binning in the discrete Mountain Car environment. To do so, we train a DQN policy until convergence (i.e. collected rewards at least $- 110$ ), and perform 500 rollouts using the greedy policy. Trajectories are used to construct a 2-dimensional state visitation histogram. At the same time, all visited states are given to the quantile binning algorithm, which is used to assign observations to bins.

Figure 10 presents the state visitation histogram with $n_{S} = 30$ bins, where color represents the state visitation count. Bins are shown by dotted lines.

7.2 Proofs

Proof.

(Lemma 4.1)

where the before last line is a direct application of Holder inequality.

∎

Proof.

(Theorem 4.2)

Following (Schulman et al., 2015), we take

L_{π} (π_{2}) = η (π_{1}) + \sum s ρ_{π_{1}} (s) \sum a π_{2} (a | s) A_{π_{1}} (s, a)

Using inequality (45) from (Schulman et al., 2015), we have:

∣ ∣ ∣ η (π_{2}) - L_{π_{1}} (π_{2}) ∣ ∣ ∣ \leq C \cdot α^{2}

Expanding $L_{π} (π_{2})$ and using triangle inequality, we have:

\begin{matrix} ∣ ∣ ∣ η (π_{2}) - {η (π_{1}) + \sum s ρ_{π} (s) \sum a π_{2} (a | s) A_{π_{1}} (s, a)} ∣ ∣ ∣ & = ∣ ∣ ∣ η (π_{2}) - η (π_{1}) - \sum s ρ_{π} (s) \sum a π_{2} (a | s) A_{π_{1}} (s, a) ∣ ∣ ∣ \geq ∣ ∣ ∣ η (π_{2}) - η (π_{1}) ∣ ∣ ∣ - ∣ ∣ ∣ \sum s ρ_{π} (s) \sum a π_{2} (a | s) A_{π_{1}} (s, a) ∣ ∣ ∣ \end{matrix}

Combining with inequality (45):

∎

Proof.

(Theorem 4.3) We first represent the approximate transition matrix $P_{~ π}$ induced by the policy $~ π$ as a perturbation of the true transition:

\begin{matrix} P_{~ π} & = P_{π} + (P_{~ π} - P_{π}) = P_{π} + E \end{matrix}

(18)

Then, the difference between stationary distributions $ρ_{~ π}$ and $ρ_{π}$ is equal to (Schweitzer, 1968; Cho & Meyer, 2001):

ρ_{π}^{⊤} - ρ_{~ π}^{⊤} = ρ_{~ π}^{⊤} E Z,

(19)

where $Z = (I - P_{π} + 1_{| S |} ρ_{π}^{⊤})^{- 1}$ is the fundamental matrix of the Markov chain induced by $P_{π}$ and $1_{| S |}$ is a vector of ones.

In particular, the above result holds for Schatten norms (Baumgartner, 2011):

| | ρ_{π} - ρ_{~ ρ} | |_{2} \leq | | ρ_{~ π} | |_{2} | | Z | |_{S_{\infty}} | | E | |_{S_{\infty}} \leq | | Z | |_{S_{\infty}} | | E | |_{S_{1}}

(20)

So far, this result is known for irreducible, homogeneous Markov chains and has no decision component.
Consider the matrix $E$ , which is the difference between expected transition models of true and approximate policies. It can be expanded into products of matricized tensors:

\begin{matrix} E & = P_{~ π} - P_{π} = A (~ Π \otimes I) T_{(3)} - A (Π \otimes I) T_{(3)}^{⊤} = A ((~ Π - Π) \otimes I) T_{(3)}^{⊤} \end{matrix}

(21)

The norm of $E$ can also be upper bounded as follows:

\begin{matrix} | | E | |_{S_{1}} & \leq | | A | |_{S_{1}} | | (~ Π - Π) \otimes I T_{(3)}^{⊤} | |_{S_{\infty}} \leq | | A | |_{S_{1}} | | (~ Π - Π) \otimes I T_{(3)}^{⊤} | |_{S_{1}} \leq | | A | |_{S_{1}} | | (~ Π - Π) \otimes I | |_{S_{2}} | | T_{(3)}^{⊤} | |_{S_{2}} \leq | | A | |_{S_{1}} | | (~ Π - Π) \otimes I | |_{S_{1}} | | T_{(3)}^{⊤} | |_{S_{2}} \leq | | A | |_{S_{1}} | | ~ Π - Π | |_{S_{1}} | | I | |_{S_{\infty}} | | T_{(3)}^{⊤} | |_{S_{2}} = | | ~ Π - Π | |_{S_{1}} | | T_{(3)} | |_{S_{2}} \end{matrix}

(22)

Combining this result from that of (Schweitzer, 1968) yields

| | ρ_{π} - ρ_{~ ρ} | |_{S_{1}} \leq | | Z | |_{S_{\infty}}      \begin{matrix} Depends on MDP + π \end{matrix} | | ~ Π - Π | |_{S_{1}}      \begin{matrix} Depends on π \end{matrix} | | T_{(3)} | |_{S_{2}}      \begin{matrix} Depends on% MDP \end{matrix}

(23)

∎

Provably efficient reconstruction of policy networks

Related Research

Planning in Hierarchical Reinforcement Learning: Guarantees for Using Local Policies

Reinforcement Learning from a Mixture of Interpretable Experts

Towards Mixed Optimization for Reinforcement Learning with Program Synthesis

A functional mirror ascent view of policy gradient methods with function approximation

State Augmented Constrained Reinforcement Learning: Overcoming the Limitations of Learning with Rewards

Sample Complexity of Nonparametric Off-Policy Evaluation on Low-Dimensional Manifolds using Deep Networks

Active Information Acquisition

1 Introduction

2 Preliminary

2.1 Reinforcement learning

2.2 Function approximation in inner product spaces

2.3 Universal density approximators

Rate of convergence of Fourier approximation

Rate of convergence of mixture approximation

2.4 Low-rank matrix factorization

2.5 Time complexity comparison

3 Reconstruction algorithm

3.1 Bandit example

3.2 Discretization of continuous policies

3.3 Pruning unvisited states

Theorem 3.1.

3.4 Acting with tabular policies

3.5 Method

4 Theoretical guarantees

4.1 Performance bound (single state case)

Lemma 4.1.

4.2 Performance bounds (multi-state case)

Theorem 4.2.

4.3 Connection between reconstruction error and policy coverage

Theorem 4.3 (Approximate policy coverage).

5 Experimental results

5.1 Rate of convergence of stationary distribution under random policy

5.2 Control domains

5.2.1 Bandit turntable

5.2.2 Mountain Car

5.2.3 Pendulum

6 Conclusion

References

7 Appendix

Reproducibility Checklist

7.1 Discretization techniques

7.2 Proofs

Proof.

Proof.

Proof.

7.3 Additional plots

7.3.1 Pendulum (Gaussian policy)

7.3.2 Pendulum (Boltzmann-Q policy)