2019/05/27 by Chuheng Zhang, Zhang, Chuheng, Yuanqi Li +3
Computer Science · Engineering · Mathematics · #Adversarial Robustness in Machine Learning #FOS: Computer and information sciences #Fuel Cells and Related Materials #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Reinforcement Learning in Robotics #cs.LG #stat.ML
paper · pdf · doi:10.48550/arxiv.1905.11041
AAAI-20 (oral)
openalex publication_date 2019/05/27 · arxiv created 2019/11/18 · arxiv updated 2019/11/19 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
We observe that several existing policy gradient methods (such as vanilla policy gradient, PPO, A2C) may suffer from overly large gradients when the current policy is close to deterministic (even in some very simple environments), leading to an unstable training process. To address this issue, we propose a new method, called target distribution learning (TDL), for policy improvement in reinforcement learning. TDL alternates between proposing a target distribution and training the policy network to approach the target distribution. TDL is more effective in constraining the KL divergence between updated policies, and hence leads to more stable policy improvements over iterations. Our experiments show that TDL algorithms perform comparably to (or better than) state-of-the-art algorithms for most continuous control tasks in the MuJoCo environment while being more stable in training.