-pre-training a policy with BC and subsequent fine-tuning with policy gradient + augmented loss to stay close to demonstration
- purely trained in MuJoCo, demonstrations collected using VR, model-free RL
- sparse rewards
policy gradient: ∇θlnπ(a∣s)A(s,a)
-∇θlnπ(a∣s) — how to move the weights to increase the likelihood of taking a from s -A(s,a) — which direction to move the weights in - A>0 increase likelihood of a
- A<0 decrease likelihood of a
- advantage calculate using methods like GAE or A(s,a)=Q(s,a)−V(s)
Demo Augmented Policy Gradient (DAPG)
1.Pre-training with BC using a maximum-likelihood objective:
- ρD — demonstration data
- learns (μθ(s),σθ(s))
maximize(s,a)∈ρD∑lnπθ(a∣s) 2.RL fine-tuning with augmented loss
- BC doesn’t optimally use the information present in demonstration data
- augmented policy gradient:
gaug=(s,a)∈ρπ∑∇θlnπθ(a∣s)Aπ(s,a)+(s,a)∈ρD∑∇θlnπθ(a∣s)w(s,a) - use w(s,a)=λ0λ1max(s′,a′)∈ρπAπ(s′,a′) for all (s,a)∈ρD, where λ0=0.1 and λ1=0.95