AWAC: A CCELERATING O NLINE R EINFORCEMENT L EARNING WITH O FFLINE D ATASETS
Anonymous authors
Paper under double-blind review
A BSTRACT
Reinforcement learning provides an appealing formalism for learning control policies from experience.
However, the classic active formulation of reinforcement learning necessitates a lengthy active exploration process for each behavior, making it difﬁcult to apply in real-world settings.
If we can instead allow reinforcement learning to effectively use previously collected data to aid the online learning process, where the data could be expert demonstrations or more generally any prior experience, we could make reinforcement learning a substantially more practical tool.
While a number of recent methods have sought to learn ofﬂine from previously collected data, it remains exceptionally difﬁcult to train a policy with ofﬂine data and improve it further with online reinforcement learning.
In this paper we systematically analyze why this problem is so challenging, and propose an algorithm that combines sample-efﬁcient dynamic programming with maximum likelihood policy updates, providing a simple and effective framework that is able to leverage large amounts of ofﬂine data and then quickly perform online ﬁne-tuning of reinforcement learning policies.
We show that our method enables rapid learning of skills with a combination of prior demonstration data and online experience across a suite of difﬁcult dexterous manipulation and benchmark tasks.
I NTRODUCTION
Learning models that generalize effectively to complex open-world settings, from image recognition (Krizhevsky et al., 2012) to natural language processing (Devlin et al., 2019), relies on large, high-capacity models and large, diverse, and representative datasets.
Leveraging this recipe for reinforcement learning (RL) has the potential to yield real-world generalization for control applications such as robotics.
However, while deep RL algorithms enable the use of large models, the use of large datasets for real-world RL has proven challenging.
Most RL algorithms collect new data online every time a new policy is learned, which limits the size and diversity of the datasets for RL.
In the same way that powerful models in computer vision and NLP are often pre-trained on large, general-purpose datasets and then ﬁne-tuned on task-speciﬁc data, RL policies that generalize effectively to open-world settings will need to be able to incorporate large amounts of prior data effectively into the learning process, while still collecting additional data online for the task at hand.
For data-driven reinforcement learning, ofﬂine datasets consist of trajectories of states, actions and associated rewards.
This data can potentially come from demonstrations for the desired task (Schaal, 1997; Atkeson & Schaal, 1997), suboptimal policies (Gao et al., 2018), demonstrations for related tasks (Zhou et al., 2019), or even just random exploration in the environment.
Depending on the quality of the data that is provided, useful knowledge can be extracted about the dynamics of the world, about the task being solved, or both.
Effective data-driven methods for deep reinforcement learning should be able to use this data to pre-train ofﬂine while improving with online ﬁne-tuning.
Since this prior data can come from a variety of sources, we would like to design an algorithm that does not utilize different types of data in any privileged way.
For example, prior methods that incorporate demonstrations into RL directly aim to mimic these demonstrations (Nair et al., 2018), which is desirable when the demonstrations are known to be optimal, but imposes strict requirements on the type of ofﬂine data, and can cause undesirable bias when the prior data is not optimal.
While prior methods for fully ofﬂine RL provide a mechanism for utilizing ofﬂine data (Fujimoto et al., 2019; Kumar et al., 2019), as we will show in our experiments, such methods generally are not effective for ﬁne-tuning with online data as they are often too conservative.
In effect, prior methods require us to choose: Do we assume prior data is optimal or not?
Do we use only ofﬂine data, or only online data?
To make it feasible to learn policies for open-world settings, we need algorithms that learn successfully in any of these cases.
In this work, we study how to build RL algorithms that are effective for pre-training from offpolicy datasets, but also well suited to continuous improvement with online data collection.
We systematically analyze the challenges with using standard off-policy RL algorithms (Haarnoja et al., 2018; Kumar et al., 2019; Abdolmaleki et al., 2018) for this problem, and introduce a simple actor critic algorithm that elegantly bridges data-driven pre-training from ofﬂine data and improvement with online data collection.
Our method, which uses dynamic programming to train a critic but a supervised learning style update to train a constrained actor, combines the best of supervised learning and actor-critic algorithms.
Dynamic programming can leverage off-policy data and enable sample-efﬁcient learning.
The simple supervised actor update implicitly enforces a constraint that mitigates the effects of distribution shift when learning from ofﬂine data (Fujimoto et al., 2019;
Kumar et al., 2019), while avoiding overly conservative updates.
We evaluate our algorithm on a wide variety of robotic control and benchmark tasks across three simulated domains: dexterous manipulation, tabletop manipulation, and MuJoCo control tasks.
Our algorithm, Advantage Weighted Actor Critic (AWAC), is able to quickly learn successful policies on difﬁcult tasks with high action dimension and binary sparse rewards, signiﬁcantly better than prior methods for off-policy and ofﬂine reinforcement learning.
Moreover, AWAC can utilize different types of prior data without any algorithmic changes: demonstrations, suboptimal data, or random exploration data.
The contribution of this work is not just another RL algorithm, but a systematic study of what makes ofﬂine pre-training with online ﬁne-tuning unique compared to the standard RL paradigm, which then directly motivates a simple algorithm, AWAC, to address these challenges.
P RELIMINARIES
We consider the standard reinforcement learning notation, with states s , actions a , policy π ( a | s ) , rewards r ( s , a ) , and dynamics p ( s (cid:48) | s , a ) .
The discounted return is deﬁned as R t =
(cid:80)
Ti = t γ i r
( s i , a i ) , for a discount factor γ and horizon T which may be inﬁnite.
The objective of an RL agent is to maximize the expected discounted return J ( π ) =
E p π ( τ ) [ R 0 ] under the distribution induced by the policy.
The optimal policy can be learned directly by policy gradient, estimating ∇ J ( π ) (Williams, 1992), but this is often ineffective due to high variance of the estimator.
Many algorithms attempt to reduce this variance by making use of the value function V π ( s ) =
E p π ( τ ) [ R t | s ] , action-value function Q π ( s , a ) =
E p π ( τ ) [ R t | s , a ] , or advantage A π ( s , a ) =
Q π ( s , a ) − V π ( s ) .
The action-value function for a policy can be written recursively via the Bellman equation:
[Equation]
Instead of estimating policy gradients directly, actor-critic algorithms maximize returns by alternating between two phases (Konda & Tsitsiklis, 2000): policy evaluation and policy improvement.
During the policy evaluation phase, the critic Q π ( s , a ) is estimated for the current policy π .
This can be accomplished by repeatedly applying the Bellman operator B , corresponding to the right-hand side of Equation 1, as deﬁned below:
[Equation]
By iterating according to Q k +1 =
B π Q k , Q k converges to Q π (Sutton & Barto, 1998).
With function approximation, we cannot apply the Bellman operator exactly, and instead minimize the Bellman error with respect to Q-function parameters φ k :
[Equation]
During policy improvement, the actor π is typically updated based on the current estimate of Q π .
A commonly used technique (Lillicrap et al., 2016; Fujimoto et al., 2018; Haarnoja et al., 2018) is to update the actor π θ k ( a | s ) via likelihood ratio or pathwise derivatives to optimize the following objective, such that the expected value of the Q-function Q π is maximized:
[Equation]
Actor-critic algorithms are widely used in deep RL (Mnih et al., 2016;
Lillicrap et al., 2016; Haarnoja et al., 2018; Fujimoto et al., 2018).
With a Q-function estimator, they can in principle utilize off-policy data when used with a replay buffer for storing prior transition tuples, which we will denote β , to sample previous transitions, although we show that this by itself is insufﬁcient for our problem setting.
[Figure]
C HALLENGES IN O FFLINE RL WITH O NLINE F INE - TUNING
In this section, we study the unique challenges that exist when pre-training using ofﬂine data, followed by ﬁne-tuning with online data collection.
We ﬁrst describe the problem, and then analyze what makes this problem difﬁcult for prior methods.
Problem deﬁnition.
A static dataset of transitions, D = { ( s , a , s (cid:48) , r ) j } , is provided to the algorithm at the beginning of training.
This dataset can be sampled from an arbitrary policy or mixture of policies, and may even be collected by a human expert.
This deﬁnition is general and encompasses many scenarios, such as learning from demonstrations, random data, prior RL experiments, or even from multi-task data.
Given the dataset D , our goal is to leverage D for pre-training and use some online interaction to learn the optimal policy π ∗ ( a | s ) , with as few interactions with the environment as possible (depicted in Fig 1).
This setting is representative of many real-world RL settings, where prior data is available and the aim is to learn new skills efﬁciently.
We ﬁrst study existing algorithms empirically in this setting on the HalfCheetah-v2 Gym environment 1 .
The prior dataset consists of 15 demonstrations from an expert policy and 100 suboptimal trajectories sampled from a behavioral clone of these demonstrations.
All methods for the remainder of this paper incorporate the prior dataset, unless explicitly labeled “scratch”.
3.1) Data Efﬁciency.
One of the simplest ways to utilize prior data such as demonstrations for RL is to pre-train a policy with imitation learning, and ﬁne-tune with on-policy RL (Gupta et al., 2019;
Rajeswaran et al., 2018).
This approach has two drawbacks: (1) prior data may not be optimal; (2) on-policy ﬁne-tuning is data inefﬁcient as it does not reuse the prior data in the RL stage.
In our setting, data efﬁciency is vital.
To this end, we require algorithms that are able to reuse arbitrary offpolicy data during online RL for data-efﬁcient ﬁne-tuning.
We ﬁnd that algorithms that use on-policy ﬁne-tuning (Rajeswaran et al., 2018; Gupta et al., 2019), or Monte-Carlo return estimation (Peters &
Schaal, 2007; Wang et al., 2018; Peng et al., 2019) are generally much less efﬁcient than off-policy actor-critic algorithms, which iterate between improving π and estimating Q π via Bellman backups.
This can be seen from the results in Figure 2 plot 1, where on-policy methods like DAPG (Rajeswaran et al., 2018) and Monte-Carlo return methods like AWR (Peng et al., 2019) and MARWIL (Wang et al., 2018) are an order of magnitude slower than off-policy actor-critic methods.
Actor-critic methods, shown in Figure 2 plot 2, can in principle use off-policy data.
However, as we will discuss next, naïvely applying these algorithms to our problem suffers from a different set of challenges.
3.2) Bootstrap Error in Ofﬂine Learning with Actor-Critic Methods.
When standard off-policy actor-critic methods are applied to this problem setting, they perform poorly, as shown in the second plot in Figure 2: despite having a prior dataset in the replay buffer, these algorithms do not beneﬁt signiﬁcantly from ofﬂine training.
We evaluate soft actor critic (Haarnoja et al., 2018), a state-of-theart actor-critic algorithm for continuous control.
Note that “SAC-scratch,” which does not receive the prior data, performs similarly to “SACfD-prior,” which does have access to the prior data, indicating that the off-policy RL algorithm is not actually able to make use of the off-policy data for pre-training.
Moreover, even if the SAC is policy is pre-trained by behavior cloning, labeled “SACfD-pretrain”, we still observe an initial decrease in performance, and performance similar to learning from scratch.
This challenge can be attributed to off-policy bootstrapping error accumulation, as observed in several prior works (Sutton & Barto, 1998;
Kumar et al., 2019; Wu et al., 2020; Levine et al., 2020;
[Figure]
Fujimoto et al., 2019).
In actor-critic algorithms the target value Q ( s (cid:48) , a (cid:48) ) , with a (cid:48) ∼ π , is used to update Q ( s , a ) .
When a (cid:48) is outside of the data distribution, Q ( s (cid:48) , a (cid:48) ) will be inaccurate, leading to accumulation of error on static datasets.
Ofﬂine RL algorithms (Fujimoto et al., 2019; Kumar et al., 2019; Wu et al., 2020) propose to address this issue by explicitly adding constraints on the policy improvement update (Equation 4) to avoid bootstrapping on out-of-distribution actions, leading to a policy update of this form:
[Equation]
Here, π θ is the actor being updated, and π β ( a | s ) represents the (potentially unknown) distribution from which all of the data seen so far (both ofﬂine data and online data) was generated.
In the case of a replay buffer, π β corresponds to a mixture distribution over all past policies.
Typically, π β is not known, especially for ofﬂine data, and must be estimated from the data itself.
Many ofﬂine RL algorithms (Kumar et al., 2019; Fujimoto et al., 2019; Siegel et al., 2020) explicitly ﬁt a parametric model to samples for the distribution π β via maximum likelihood estimation, where samples from π β are obtained simply by sampling uniformly from the data seen thus far: ˆ π β = max
ˆ π β E s , a ∼ π β [log ˆ π β ( a | s )] .
After estimating ˆ π β , prior methods implement the constraint given in
Equation 5 in various ways, including penalties on the policy update (Kumar et al., 2019; Wu et al., 2020) or architecture choices for sampling actions for policy training (Fujimoto et al., 2019; Siegel et al., 2020).
As we will see next, the requirement for accurate estimation of ˆ π β makes these methods difﬁcult to use with online ﬁne-tuning.
3.3) Excessively Conservative Online Learning.
While ofﬂine RL algorithms with constraints (Kumar et al., 2019; Fujimoto et al., 2019; Wu et al., 2020) perform well ofﬂine, they struggle to improve with ﬁne-tuning, as shown in the third plot in Figure 2.
We see that the purely ofﬂine RL performance (at “0K” in Fig. 2) is much better than the standard off-policy methods shown in Section 3.2.
However, with additional iterations of online ﬁne-tuning, the performance increases very slowly (as seen from the slope of the BEAR curve in Fig 2).
What causes this phenomenon?
This can be attributed to challenges in ﬁtting an accurate behavior model as data is collected online during ﬁne-tuning.
In the ofﬂine setting, behavior models must only be trained once via maximum likelihood, but in the online setting, the behavior model must be updated online to track incoming data.
Training density models online (in the “streaming” setting) is a challenging research problem (Ramapuram et al., 2017), made more difﬁcult by a potentially complex multi-modal behavior distribution induced by the mixture of online and ofﬂine data.
To understand this, we plot the log likelihood of learned behavior models on the dataset during online and ofﬂine training for the HalfCheetah task.
As we can see in the plot, the accuracy of the behavior models ( log π β on the y-axis) reduces during online ﬁne-tuning, indicating that it is not ﬁtting the new data well during online training.
When the behavior models are inaccurate or unable to model new data well, constrained optimization becomes too conservative, resulting in limited improvement with ﬁne-tuning.
This analysis suggests that, in order to address our problem setting, we require an off-policy RL algorithm that constrains the policy to prevent ofﬂine instability and error accumulation, but not so conservatively that it prevents online ﬁne-tuning due to imperfect behavior modeling.
Our proposed algorithm, which we discuss in the next section, accomplishes this by employing an implicit constraint, which does not require any explicit modeling of the behavior policy.
A DVANTAGE W EIGHTED A CTOR C RITIC : A S IMPLE A LGORITHM FOR F INE - TUNING FROM O FFLINE D ATASETS
In this section, we will describe the advantage weighted actor-critic (AWAC) algorithm, which trains an off-policy critic and an actor with an implicit policy constraint.
We will show AWAC mitigates the challenges outlined in Section 3.
AWAC follows the design for actor-critic algorithms as described in Section 2, with a policy evaluation step to learn Q π and a policy improvement step to update π .
AWAC uses off-policy temporal-difference learning to estimate Q π in the policy evaluation step, and a policy improvement update that is able to obtain the beneﬁts of ofﬂine RL algorithms at training from prior datasets, while avoiding the overly conservative behavior described in Section 3.3.
We describe the policy improvement step in AWAC below, and then summarize the entire algorithm.
Policy improvement for AWAC proceeds by learning a policy that maximizes the value of the critic learned in the policy evaluation step via TD bootstrapping.
If done naively, this can lead to the issues described in Section 3.3, but we can avoid the challenges of bootstrap error accumulation by restricting the policy distribution to stay close to the data observed thus far during the actor update, while maximizing the value of the critic.
At iteration k , AWAC therefore optimizes the policy to maximize the estimated Q-function Q π k ( s , a ) at every state, while constraining it to stay close to the actions observed in the data, similar to prior ofﬂine RL methods, though this constraint will be enforced differently.
Note from the deﬁnition of the advantage in Section 2 that optimizing Q π k ( s , a ) is equivalent to optimizing A π k ( s , a ) .
We can therefore write this optimization as:
[Equation]
As we saw in Section 3.2, enforcing the constraint by incorporating an explicit learned behavior model (Kumar et al., 2019; Fujimoto et al., 2019; Wu et al., 2020; Siegel et al., 2020) leads to poor ﬁne-tuning performance.
Instead, we enforce the constraint implicitly , without learning a behavior model.
We ﬁrst derive the solution to the constrained optimization in Equation 6 to obtain a nonparametric closed form for the actor.
This solution is then projected onto the parametric policy class without any explicit behavior model.
The analytic solution to Equation 6 can be obtained by enforcing the KKT conditions (Peters & Schaal, 2007; Peters et al., 2010; Peng et al., 2019).
The Lagrangian is:
[Equation]
and the closed form solution to this problem is π ∗ ( a | s )
∝ π β
a | s ) exp
[Equation]
1 λ A π k
( s , a ) (cid:1) .
When using function approximators, such as deep neural networks as we do, we need to project the non-parametric solution into our policy space.
For a policy π θ with parameters θ , this can be done by minimizing the
KL divergence of π θ from the optimal non-parametric solution π ∗ under the data distribution ρ π β ( s ) :
[Equation]
Note that the parametric policy could be projected with either direction of KL divergence.
Choosing the reverse KL results in explicit penalty methods (Wu et al., 2020) that rely on evaluating the density of a learned behavior model.
Instead, by using forward KL, we can compute the policy update by sampling directly from β :
[Equation]
This actor update amounts to weighted maximum likelihood (i.e., supervised learning), where the targets are obtained by re-weighting the state-action pairs observed in the current dataset by the predicted advantages from the learned critic, without explicitly learning any parametric behavior model, simply sampling ( s, a ) from the replay buffer β .
See Appendix A.2 for a more detailed derivation and Appendix A.3 for speciﬁc implementation details.
Avoiding explicit behavior modeling.
Note that the update in Equation 9 completely avoids any modeling of the previously observed data β with a parametric model.
By avoiding any explicit learning of the behavior model AWAC is far less conservative than methods which ﬁt a model ˆ π β explicitly, and better incorporates new data during online ﬁne-tuning, as seen from our results in
Section 6.
This derivation is related to AWR (Peng et al., 2019), with the main difference that AWAC uses an off-policy Q-function Q π to estimate the advantage, which greatly improves efﬁciency and even ﬁnal performance (see results in Section 6.1).
The update also resembles ABM-MPO, but
ABM-MPO does require modeling the behavior policy which, as discussed in Section 3.3, can lead to poor ﬁne-tuning.
In Section 6.1, AWAC outperforms ABM-MPO on a range of challenging tasks.
Policy evaluation.
During policy evaluation, we estimate the action-value Q π ( s , a ) for the current policy π , as described in Section 2.
We utilize a temporal difference learning scheme for policy evaluation (Haarnoja et al., 2018; Fujimoto et al., 2018), minimizing the Bellman error as described in Equation 2.
This enables us to learn very efﬁciently from off-policy data.
This is particularly important in our problem setting to effectively use the ofﬂine dataset, and allows us to signiﬁcantly outperform alternatives using Monte-Carlo evaluation or TD( λ ) to estimate returns (Peng et al., 2019).
1: Dataset D = { ( s , a , s (cid:48) , r ) j } 2: Initialize buffer β
= D 3: Initialize π θ , Q φ 4: for iteration i = 1 , 2 , ... do 5:
Sample batch ( s , a , s (cid:48) , r ) ∼ β 6:
Update φ according to Eqn.7:
Update θ according to Eqn.8: if i > num_ofﬂine_steps then 9: τ 1 , . .
, τ K ∼ p π θ ( τ ) 10: β ← β ∪ { τ 1 , . . .
, τ K } 11: end if 12: end for
Algorithm summary.
The full AWAC algorithm for ofﬂine RL with online ﬁne-tuning is summarized in Algorithm 1.
In a practical implementation, we can parameterize the actor and the critic by neural networks and perform SGD updates from Eqn.
9 and Eqn. 3.
Speciﬁc details are provided in Appendix A.3.
AWAC ensures data efﬁciency with off-policy critic estimation via bootstrapping, and avoids ofﬂine bootstrap error with a constrained actor update.
By avoiding explicit modeling of the behavior policy, AWAC avoids overly conservative updates.
While AWAC is certainly quite related to several prior works, we note that there are key differences that make it particularly amenable to the problem setting we are considering - ofﬂine RL with online ﬁne-tuning, that none of the other methods are really able to tackle.
As we show in our experimental analysis with direct comparisons to prior work, every one of the design decisions being made in this work are important for algorithm performance.
As compared to AWR (Peng et al., 2019), AWAC uses TD bootstrapping for signiﬁcantly more efﬁcient and even asymptotically better performance.
As compared to ofﬂine RL techniques like ABM (Siegel et al., 2020), MPO (Abdolmaleki et al., 2018), BEAR (Kumar et al., 2019) or BCQ (Fujimoto et al., 2019) this work is able to avoid the need for any behavior modeling, thereby enabling the online ﬁne-tuning part of the problem much better.
As shown in Fig 3, when these seemingly ablations are made to AWAC, the algorithm performs signiﬁcantly worse.
R ELATED W ORK
Off-policy RL algorithms are designed to reuse off-policy data during training, and have been studied extensively (Konda & Tsitsiklis, 2000; Degris et al., 2012;
Mnih et al., 2016; Haarnoja et al., 2018;
Fujimoto et al., 2018; Bhatnagar et al., 2009; Peters & Schaal, 2008a; Zhang et al., 2019; Wawrzynski, 2009; Balduzzi & Ghifary, 2015).
While standard off-policy methods are able to beneﬁt from including data seen during a training run, as we show in Section 3.2 they struggle when training from previously collected ofﬂine data from other policies, due to error accumulation with distribution shift (Fujimoto et al., 2019; Kumar et al., 2019).
Ofﬂine RL methods aim to address this issue, often by constraining the actor updates to avoid excessive deviation from the data distribution (Lange et al., 2012; Thomas & Brunskill, 2016; Hallak et al., 2015; 2016; Hallak & Mannor, 2017; Agarwal et al., 2019; Kumar et al., 2019; Fujimoto et al., 2019; Fakoor et al., 2019; Nachum et al., 2019;
Siegel et al., 2020; Levine et al., 2020; Zhang et al., 2020).
One class of these methods utilize importance sampling (Thomas & Brunskill, 2016;
Zhang et al., 2020; Nachum et al., 2019; Degris et al., 2012; Jiang & Li, 2016; Hallak & Mannor, 2017).
Another class of methods perform ofﬂine reinforcement learning via dynamic programming, with an explicit constraint to prevent deviation from the data distribution (Lange et al., 2012; Kumar et al., 2019; Fujimoto et al., 2019; Wu et al., 2020; Jaques et al., 2019).
While these algorithms perform well in the purely ofﬂine settings, we show in Section 3.3 that such methods tend to be overly conservative, and therefore may not learn efﬁciently when ﬁne-tuning with online data collection.
In contrast, our algorithm AWAC is comparable to these algorithms for ofﬂine pre-training, but learns much more efﬁciently during subsequent ﬁne-tuning.
[Figure]
Prior work has also considered the special case of learning from demonstration data.
One class of algorithms initializes the policy via behavioral cloning from demonstrations, and then ﬁne-tunes with reinforcement learning (Peters & Schaal, 2008b; Ijspeert et al., 2002; Theodorou et al., 2010;
Kim et al., 2013; Rajeswaran et al., 2018; Gupta et al., 2019; Zhu et al., 2019).
Most such methods use on-policy ﬁne-tuning, which is less sample-efﬁcient than off-policy methods that perform value function estimation.
Other prior works have incorporated demonstration data into the replay buffer using off-policy RL methods (Ve ˇ cerík et al., 2017; Nair et al., 2017).
We show in Section 3.2 that these strategies can result in a large dip in performance during online ﬁne-tuning, due to the inability to pre-train an effective value function from ofﬂine data.
In contrast, our work shows that using supervised learning style policy updates can allow for better bootstrapping from demonstrations as compared to Veˇcerík et al.
(2017) and Nair et al.
Our method builds on algorithms that implement a maximum likelihood objective for the actor, based on an expectation-maximization formulation of RL (Peters & Schaal, 2007; Neumann &
Peters, 2008; Theodorou et al., 2010; Peters et al., 2010; Peng et al., 2019; Abdolmaleki et al., 2018;
Wang et al., 2018).
Most closely related to our method in this respect are the algorithms proposed by Peng et al.
(2019) (AWR) and Siegel et al.
(2020) (ABM).
Unlike AWR, which estimates the value function of the behavior policy, V π β via Monte-Carlo estimation or TD − λ , our algorithm estimates the Q-function of the current policy Q π via bootstrapping, enabling much more efﬁcient learning, as shown in our experiments.
Unlike ABM, our method does not require learning a separate function approximator to model the behavior policy π β , and instead directly samples the dataset.
As we discussed in Section 3.3, modeling π β can be a major challenge for online ﬁne-tuning.
While these distinctions may seem somewhat subtle, they are important and we show in our experiments that they result in a large difference in algorithm performance.
Finally, our work goes beyond the analysis in prior work, by studying the issues associated with pre-training and ﬁne-tuning in Section 3.
Concurrently to our work, Wang et al.
(2020) proposed critic regularized regression for ofﬂine RL, which uses off-policy Q-learning and an equivalent policy update.
In contrast to this concurrent work, we speciﬁcally study the ofﬂine pretraining online ﬁne-tuning problem, analyze why other methods are ineffective in this setting, and show that our approach achieves substantially better results.
E XPERIMENTAL E VALUATION
In our experiments, we ﬁrst compare our method against prior methods in the ofﬂine training and ﬁne-tuning setting.
We show that we can learn difﬁcult, high-dimensional, sparse reward dexterous manipulation problems from human demonstrations and off-policy data.
We then evaluate our method with suboptimal prior data generated by a random controller.
Finally, we study why prior methods struggle in this setting by analyzing their performance on benchmark MuJoCo tasks, and conduct further experiments to understand where the difﬁculty lies (also shown in Section 3).
6.1) Comparative Evaluation Learning From Prior Data
We
aim
to
study
tasks representative
of the difﬁculties of real-world robot learning, where ofﬂine learning and online ﬁne-tuning are
most relevant.
We begin our analysis with a set of challenging sparse reward dexterous manipulation tasks proposed by Rajeswaran et al.
These tasks involve complex manipulation skills using a 28-DoF ﬁve-ﬁngered hand in the MuJoCo simulator (Todorov et al., 2012) shown in Figure 3: in-hand rotation of a pen, opening a door by unlatching the handle, and picking up a sphere and relocating it to a target location.
These environments exhibit many challenges: high dimensional action spaces, complex manipulation physics with many intermittent contacts, and randomized hand and object positions.
The reward functions in these environments are binary 0-1 rewards for task completion.
2 Rajeswaran et al.
(2018) provide 25 human demonstrations for each task, which are not fully optimal but do solve the task.
Since this dataset is small, we generated another 500 trajectories of interaction data by constructing a behavioral cloned policy, and then sampling from this policy.
First, we compare our method on these dexterous manipulation tasks against prior methods for off-policy learning, ofﬂine learning, and bootstrapping from demonstrations.
Speciﬁc implementation details are discussed in Appendix A.5.
The results are shown in Fig.
Our method is able to leverage the prior data to quickly attain good performance, and the efﬁcient off-policy actor-critic component of our approach ﬁne-tunes much more quickly than demonstration augmented policy gradient (DAPG), the method proposed by Rajeswaran et al.
For example, our method solves the pen task in 120K timesteps, the equivalent of just 20 minutes of online interaction.
While the baseline comparisons and ablations are able to make some amount of progress on the pen task, alternative off-policy RL and ofﬂine RL algorithms are largely unable to solve the door and relocate task in the time-frame considered.
We ﬁnd that the design decisions to use off-policy critic estimation allow
AWAC to signiﬁcantly outperform AWR (Peng et al., 2019) while the implicit behavior modeling allows AWAC to signiﬁcantly outperform ABM (Siegel et al., 2020), although ABM does make some progress.
Rajeswaran et al.
(2018) show that DAPG can solve variants of these tasks with more well-shaped rewards, but still requires considerably more samples.
Additionally, we evaluated all methods on the Gym MuJoCo locomotion benchmarks, similarly providing demonstrations as ofﬂine data.
Due to space constraints, the results plots for these experiments are included in Appendix A.1.
These tasks are substantially easier than the sparse reward manipulation tasks described above, and a number of prior methods also perform well.
However, our method matches or exceeds the best prior method in all cases, whereas no other single prior method attains good performance on all of the tasks.
[Figure]
6.2) Fine-Tuning from Random Policy Data.
An advantage of using off-policy RL for reinforcement learning is that we can also incorporate suboptimal data, rather than demonstrations.
In this experiment, we evaluate on a simulated tabletop pushing environment with a Sawyer robot pictured in Fig 3 and described further in Appendix A.4.
To study the potential to learn from suboptimal data, we use an off-policy dataset of 500 trajectories generated by a random process.
The task is to push an object to a target location in a 40cm x 20cm goal space.
The results are shown in Figure 4.
We see that while many methods begin at the same initial performance, AWAC learns the fastest online and is actually able to make use of the ofﬂine dataset effectively.
D ISCUSSION AND F UTURE W ORK
We have discussed in detail the challenges existing RL methods face when ﬁne-tuning from prior datasets, and proposed an algorithm, AWAC, that is effective in this setting.
The key insight in
AWAC is that enforcing a policy update constraint implicitly on actor-critic methods results in a stable learning algorithm amenable for off-policy learning.
With an informative action-value estimate, the policy is weighted towards high-advantage actions in the data, resulting in policy improvement without conservative updates.
A direction of future work we plan to pursue is applying AWAC to solve difﬁcult robotic tasks in the real world.
More than just speeding up individual runs, incorporating prior data into the learning process enables continuously accumulating data by saving environment interactions of the robot - for instance, runs of RL with varying hyperparameters.
We hope that this enables a wider array of robotic applications than previously possible.
A A PPENDIX A. G YM B ENCHMARK R ESULTS F ROM P RIOR D ATA
In this section, we provide a comparative evaluation on MuJoCo benchmark tasks for analysis.
These tasks are simpler, with dense rewards and relatively lower action and observation dimensionality.
Thus, many prior methods can make good progress on these tasks.
These experiments allow us to understand more precisely which design decisions are crucial.
For each task, we collectdemonstration trajectories using a pre-trained expert on each task, and 100 trajectories of off-policy data by rolling out a behavioral cloned policy trained on the demonstrations.
The same data is made available to all methods.
The results are presented in Figure 5.
AWAC is consistently the best or on par with the best-performing method.
No other single method consistently attains the best results – on HalfCheetah, SAC + BC and BRAC are competitive, while on Ant-v2 ABM is competitive with
AWAC.
We summarize the results according to the challenges in Section 3.
Data efﬁciency.
The three methods that do not estimate Q π are DAPG (Abdolmaleki et al., 2018),
AWR (Peng et al., 2019), and MARWIL (Wang et al., 2018).
Across all three tasks, we see that these methods are somewhat worse ofﬂine than the best performing ofﬂine methods, and exhibit steady but very slow improvement during ﬁne-tuning.
In robotics, data efﬁciency is vital, so these algorithms are not good candidates for practical real-world applications.
Bootstrap error in ofﬂine learning.
For SAC (Haarnoja et al., 2018), across all three tasks, we see that the ofﬂine performance at epoch 0 is generally poor.
Due to the data in the replay buffer, SAC with prior data does learn faster than from scratch, but AWAC is faster to solve the tasks in general.
SAC with additional data in the replay buffer is similar to the approach proposed by Veˇcerík et al.
SAC+BC reproduces Nair et al.
but uses SAC instead of DDPG (Lillicrap et al., 2016) as the underlying RL algorithm.
We ﬁnd that these algorithms exhibit a characteristic dip at the start of learning.
Although this dip is only present in the early part of the learning curve, a poor initial policy and lack of steady policy improvement can be a safety concern and a signiﬁcant hindrance in real-world applications.
Moreover, recall that in the more difﬁcult dextrous manipulation tasks, these algorithms do not show any signiﬁcant learning.
Conservative online learning.
Finally, we consider conservative ofﬂine algorithms: ABM (Siegel et al., 2020), BEAR (Kumar et al., 2019), and BRAC (Wu et al., 2020).
We found that BRAC performs similarly to SAC for working hyperparameters.
BEAR trains well ofﬂine – on Ant and
Walker2d, BEAR signiﬁcantly outperforms prior methods before online experience.
However, online improvement is slow for BEAR and the ﬁnal performance across all three tasks is much lower than
AWAC.
The closest in performance to our method is ABM, which is comparable on Ant-v2, but much slower on other domains.
[Figure]
A. A LGORITHM D ERIVATION D ETAILS
The full optimization problem we solve, given the previous off-policy advantage estimate A π k and buffer distribution π β , is given below:
[Equation]
Our derivation follows Peters et al.
and Peng et al.
The analytic solution for the constrained optimization problem above can be obtained by enforcing the KKT conditions.
The
Lagrangian is:
[Equation]
Differentiating with respect to π gives:
[Equation]
Setting ∂ L ∂π to zero and solving for π gives the closed form solution to this problem:
[Equation]
Next
[Equation]
we project the solution into the space of parametric policies.
For a policy π θ with parameters θ , this can be done by minimizing the KL divergence of π θ from the optimal non-parametric solution π ∗ under the data distribution ρ π β ( s ) :
[Equation]
Note that in the projection step, the parametric policy could be projected with either direction of KL divergence.
However, choosing the reverse KL direction has a key advantage: it allows us to optimize θ as a maximum likelihood problem with an expectation over data s, a ∼ β , rather than sampling actions from the policy that may be out of distribution for the Q function.
In our experiments we show that this decision is vital for stable off-policy learning.
Furthermore, assume discrete policies with a minimum probably density of π θ ≥ α θ .
Then the upper bound:
[Equation]
holds by the Pinsker’s inequality, where D TV denotes the total variation distance between distributions.
Thus minimizing the reverse KL also bounds the forward KL.
Note that we can control the minimum α if desired by applying Laplace smoothing to the policy.
A. I MPLEMENTATION D ETAILS
We implement the algorithm building on top of twin soft actor-critic (Haarnoja et al., 2018), which incorporates the twin Q-function architecture from twin delayed deep deterministic policy gradient (TD3) from Fujimoto et al.
All off-policy algorithm comparisons (SAC, BRAC, MPO, ABM,
BEAR) are implemented from the same skeleton.
The base hyperparameters are given in Table 2.
The policy update is replaced with:
[Equation] [Equation]
[Table]
Similar to advantage weight regression (Peng et
, 2019) and other prior work (Neumann & Peters, 2008;
Wang et al.
Siegel et al., 2020), we disregard the per-state normalizing constant
Z ( s ) =
[Equation]
[Table]
[Equation]
We did experiment with estimating this expectation per batch element with K = 10 samples, but found that this generally made performance worse, perhaps because errors in the estimation of
Z ( s ) caused more harm than the beneﬁt the method derived from estimating this value.
We report success rate results for variants of our method with and without Z ( s ) estimation in Table 1.
While prior work (Neumann & Peters, 2008; Wang et al., 2018;
Peng et al., 2019) has generally ignored the omission of Z ( s ) without any speciﬁc justiﬁcation, it is possible to bound this value both above and below using the Cauchy-Schwarz and reverse CauchySchwarz (Polya-Szego) inequalities, as follows.
Let f ( a ) =
π ( a | s ) and g ( a ) =
exp( A ( s , a ) /λ ) .
Note f
( a ) > 0 for stochastic policies and g ( a ) > 0 .
By Cauchy-Schwarz, Z ( s ) =
(cid:82)
a f ( a ) g ( a ) d a ≤ (cid:113)(cid:82) a f ( a ) 2 d a (cid:82) a g ( a ) 2 d a = C 1 .
To apply Polya-Szego, let m f and m g be the minimum of f and g respectively and M f , M g be the maximum.
Then Z ( s ) ≥ 2( (cid:113)
M f M g m f m g + m f m g M f M g ) − 1 C
= C 2 .
We therefore have C 1 ≤ Z ( s ) ≤ C 2 , though the bounds are generally not tight.
A further, more intuitive argument for why omitting Z ( s ) may be harmless in practice comes from observing that this normalizing factor only affects the relative weight of different states in the training objective, not different actions.
The state distribution in β already differs from the distribution over states that will be visited by π θ , and therefore preserving this state distribution is likely to be of limited utility to downstream policy performance.
Indeed, we would expect that sufﬁciently expressive policies would be less affected by small to moderate variability in the state weights.
On the other hand, inaccurate estimates of Z ( s ) may throw off the training objective by increasing variance, similar to the effect of degenerate importance weights.
The Lagrange multiplier λ is treated as a hyperparameter in our method.
In this work we use λ = 0 .for
the manipulation environments and λ = 1 . 0 for the MuJoCo benchmark environments.
One could adaptively learn λ with a dual gradient descent procedure, but this would require access to π β .
As rewards for the dextrous manipulation environments are non-positive, we clamp the Q value for these experiments to be at most zero.
We ﬁnd this stabilizes training slightly.
A. E NVIRONMENT -S PECIFIC D ETAILS
We evaluate our method on three domains: dexterous manipulation environments, Sawyer manipulation environments, and MuJoCo benchmark environments.
In the following sections we describe speciﬁc details.
A.4. D EXTEROUS M ANIPULATION E NVIRONMENTS
These environments are modiﬁed from those proposed by Rajeswaran et al.
pen-binary-v0.
The task is to spin a pen into a given orientation.
The action dimension is 24 and the observation dimension is 45.
Let the position and orientation of the pen be denoted by x p and x o respectively, and the desired position and orientation be denoted by d p and d o respectively.
The reward function is r = 1 | x
p − d p |≤ 0 . 075 1 | x o · d o |≤ 0 .
In Rajeswaran et al.
(2018), the episode was terminated when the pen fell out of the hand; we did not include this early termination condition.
door-binary-v0.
The task is to open a door, which requires ﬁrst twisting a latch.
The action dimension is 28 and the observation dimension is 39.
Let d denote the angle of the door.
The reward function is r = 1 d> 1 .
[Table]
relocate-binary-v0.
The task is to relocate an object to a goal location.
The action dimension is 30 and the observation dimension is 39.
Let x p denote the object position and d p denote the desired position.
The reward is r = 1 | x p − d p |≤ 0 .
A.4. S AWYER M ANIPULATION E NVIRONMENT
SawyerPush-v0.
This environment is included in the Multiworld library.
The task is to push a puck to a goal position in a 40cm x 20cm, and the reward function is the negative distance between the puck and goal position.
When using this environment, we use hindsight experience replay for goal-conditioned reinforcement learning.
The random dataset for prior data was collected by rolling out an Ornstein-Uhlenbeck process with θ = 0 .
15 and σ = 0 .
A.4. O FF -P OLICY D ATA P ERFORMANCE
[Table]
[Table]
The performances of the expert data, behavior cloning (BC) on the expert data (1), and BC on the combined expert+BC data (2) are included in Table 3.
For Gym benchmarks we report average return, and expert data is collected by a trained SAC policy.
For dextrous manipulation tasks we report the success rate, and the expert data consists of human demonstrations provided by Rajeswaran et al.
A. B ASELINE I MPLEMENTATION D ETAILS
We used public implementations of prior methods (DAPG,
AWR) when available.
We implemented the remaining algorithms in our framework, which also allows us to understand the effects of changing individual components of the method.
In the section, we describe the implementation details.
The full overview of algorithms is given in
Figure 6.
[Figure]
Policy Objective
[Figure]
Constraint
[Figure]
No
None
[Figure]
No
None
[Figure]
Yes
[Figure]
( (cid:96) ∞ )
BEAR
Q
[Figure]
Q
[Figure]
Yes
Support (MMD)
[Figure]
No
Implicit
[Figure]
Prior
[Figure]
Yes
[Figure]
No
[Figure]
Behavior Cloning (BC).
This method learns a policy with supervised learning on demonstration data.
Soft Actor Critic (SAC).
Using the soft actor critic algorithm from (Haarnoja et al., 2018), we follow the exact same procedure as our method in order to incorporate prior data, initializing the policy with behavior cloning on demonstrations and adding all prior data to the replay buffer.
Behavior Regularized Actor Critic (BRAC).
We implement BRAC as described in (Wu et al., 2020) by adding policy regularization log( π β ( a | s )) where π β is a behavior policy trained with supervised learning on the replay buffer.
We add all prior data to the replay buffer before online training.
Advantage Weighted Regression (AWR).
Using the advantage weighted regression algorithm from (Peng et al., 2019), we add all prior data to the replay buffer before online training.
We use the implementation provided by Peng et al.
(2019), with the key difference from our method being that
AWR uses TD( λ ) on the replay buffer for policy evaluation.
Monotonic Advantage Re-Weighted Imitation Learning (MARWIL).
Monotonic advantage reweighted imitation learning was proposed by Wang et al.
(2018) for ofﬂine imitation learning.
MARWIL was not demonstrated in online RL settings, but we evaluate it for ofﬂine pretraining followed by online ﬁne-tuning as we do other ofﬂine algorithms.
Although derived differently,
MARWIL and AWR are similar algorithms and only differ in value estimation: MARWIL uses the on-policy single-path advantage estimate
A ( s, a ) =
Q π β ( s, a ) − V π β ( s ) instead of TD( λ ) as in
AWR.
Thus, we implement MARWIL by modifying the implementation of AWR.
Maximum a Posteriori Policy Optimization (MPO).
We evaluate the MPO algorithm presented by Abdolmaleki et al.
Due to a public implementation being unavailable, we modify our algorithm to be as close to MPO as possible.
In particular, we change the policy update in Advantage
Weighted Actor Critic to be:
[Equation]
Note that in MPO, actions for the update are sampled from the policy and the Q-function is used instead of advantage for weights.
We failed to see ofﬂine or online improvement with this implementation in most environments, so we omit this comparison in favor of ABM.
Advantage-Weighted Behavior Model (ABM).
We evaluate ABM, the method developed in Siegel et al.
As with MPO, we modify our method to implement ABM, as there is no public implementation of the method.
ABM ﬁrst trains an advantage model π θ abm ( a | s ) :
[Equation]
where f is an increasing non-negative function, chosen to be f = 1 + .
In place of an advantage computed by empirical returns R ( τ t : N ) − ˆ V ( s ) we use the advantage estimate computed per transition by the Q value Q ( s, a ) − V ( s ) .
This is favorable for running ABM online, as computing R ( τ t : N ) − ˆ V ( s ) is similar to AWR, which shows slow online improvement.
We then use the policy update:
[Equation]
Additionally, for this method, actions for the update are sampled from a behavior policy trained to match the replay buffer and the value function is computed as V π ( s )
=
Q π ( s, a ) s.t.
a ∼ π .
Demonstration Augmented Policy Gradient (DAPG).
We directly utilize the code provided in (Rajeswaran et al., 2018) to compare against our method.
Since DAPG is an on-policy method, we only provide the demonstration data to the DAPG code to bootstrap the initial policy from.
Bootstrapping Error Accumulation Reduction (BEAR).
We utilize the implementation of BEAR provided in rlkit.
We provide the demonstration and off-policy data to the method together.
Since the original method only involved training ofﬂine, we modify the algorithm to include an online training phase.
In general we found that the MMD constraint in the method was too conservative.
As a result, in order to obtain the results displayed in our paper, we swept the MMD threshold value and chose the one with the best ﬁnal performance after ofﬂine training with ofﬂine ﬁne-tuning.
A. E XTRA B ASELINE C OMPARISONS (CQL, A LGAE DICE)
In this section, we add comparisons to constrained Q-learning (CQL) (Kumar et al., 2020) and
AlgaeDICE (Nachum et al., 2019).
For CQL, we use the authors’ implementation, modiﬁed for additionally online-ﬁnetuning instead of only ofﬂine training.
For AlgaeDICE, we use the publicly available implementation, modiﬁed to load prior data and perform 25K pretraining steps before online
RL.
The results are presented in Figure 7.