Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Efficient Wasserstein Natural Gradients for Reinforcement Learning

Oct 12, 2020
Ted Moskovitz, Michael Arbel, Ferenc Huszar, Arthur Gretton

Share this with someone who'll enjoy it:

A novel optimization approach is proposed for application to policy gradient methods and evolution strategies for reinforcement learning (RL). The procedure uses a computationally efficient Wasserstein natural gradient (WNG) descent that takes advantage of the geometry induced by a Wasserstein penalty to speed optimization. This method follows the recent theme in RL of including a divergence penalty in the objective to establish a trust region. Experiments on challenging tasks demonstrate improvements in both computational cost and performance over advanced baselines.

   Access Paper Source

Share this with someone who'll enjoy it: