Deep reinforcement learning (DRL) has significantly advanced artificial agents in complex environments by integrating deep learning with reinforcement learning, demonstrating success in domains such as robotics, reinforcement learning from human feedback (RLHF), and game-playing. However, the alternating training and execution phases, particularly in on-policy methods, introduce substantial synchronization overhead, limiting efficiency. To address this challenge, we analyze the computational interplay between these phases and propose CoTraX, a novel framework that strategically overlaps training and execution to optimize resource utilization and accelerate training. Furthermore, we develop an adaptive control algorithm to mitigate potential adverse effects of overlapping. Extensive experiments demonstrate that CoTraX reduces training time by an average of \(9.89\%\) without compromising performance.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CoTraX: An Efficient Parallel Training Method for On-Policy Deep Reinforcement Learning

  • Hao Dai,
  • Yuqing Zhu,
  • Xueying Zhu,
  • Chong Tang,
  • Ni Li,
  • Yang Wang

摘要

Deep reinforcement learning (DRL) has significantly advanced artificial agents in complex environments by integrating deep learning with reinforcement learning, demonstrating success in domains such as robotics, reinforcement learning from human feedback (RLHF), and game-playing. However, the alternating training and execution phases, particularly in on-policy methods, introduce substantial synchronization overhead, limiting efficiency. To address this challenge, we analyze the computational interplay between these phases and propose CoTraX, a novel framework that strategically overlaps training and execution to optimize resource utilization and accelerate training. Furthermore, we develop an adaptive control algorithm to mitigate potential adverse effects of overlapping. Extensive experiments demonstrate that CoTraX reduces training time by an average of \(9.89\%\) without compromising performance.