Timezone: »

Neural Proximal/Trust Region Policy Optimization Attains Globally Optimal Policy
Boyi Liu · Qi Cai · Zhuoran Yang · Zhaoran Wang

Wed Dec 11 05:00 PM -- 07:00 PM (PST) @ East Exhibition Hall B + C #210

Proximal policy optimization and trust region policy optimization (PPO and TRPO) with actor and critic parametrized by neural networks achieve significant empirical success in deep reinforcement learning. However, due to nonconvexity, the global convergence of PPO and TRPO remains less understood, which separates theory from practice. In this paper, we prove that a variant of PPO and TRPO equipped with overparametrized neural networks converges to the globally optimal policy at a sublinear rate. The key to our analysis is the global convergence of infinite-dimensional mirror descent under a notion of one-point monotonicity, where the gradient and iterate are instantiated by neural networks. In particular, the desirable representation power and optimization geometry induced by the overparametrization of such neural networks allow them to accurately approximate the infinite-dimensional gradient and iterate.

Author Information

Boyi Liu (Northwestern University)
Qi Cai (Northwestern University)
Zhuoran Yang (Princeton University)
Zhaoran Wang (Northwestern University)

More from the Same Authors