arXiv:2501.12703cs.ARcs.AI2025-01中稿 · the 2024 Internati…被引 1

用FPGA加速PPO中的广义优势估计,提升训练效率

HEPPO-GAE: Hardware-Efficient Proximal Policy Optimization with Generalized Advantage Estimation

  • 在单片SoC上实现并行流水线架构,优化GAE计算
  • 内存使用减少4倍,累计奖励提升1.5倍,速度加快30%
  • 适合需要高效强化学习硬件部署的研究与工业场景

本文提出HEPPPO-GAE,一种基于FPGA的加速器,用于优化强化学习中近端策略优化(PPO)的广义优势估计(GAE)阶段。不同于以往聚焦轨迹收集与演员-评论家更新的工作,本方案通过单片系统级芯片(SoC)上的并行流水线架构,应对GAE的高计算需求。关键创新在于融合动态奖励归一化与块归一化,并采用8位均匀量化,有效稳定学习过程,提升性能并缓解内存瓶颈,实现内存占用降低4倍、累积奖励提升1.5倍。该单芯片方案结合可编程逻辑与嵌入式处理器,吞吐量远超传统CPU-GPU系统,显著降低通信延迟与吞吐瓶颈,实验表明PPO训练速度提升30%,内存访问时间大幅减少,展现出在硬件高效强化学习中的广泛应用潜力。

原文摘要 · Abstract (English)

This paper introduces HEPPO-GAE, an FPGA-based accelerator designed to optimize the Generalized Advantage Estimation (GAE) stage in Proximal Policy Optimization (PPO). Unlike previous approaches that focused on trajectory collection and actor-critic updates, HEPPO-GAE addresses GAE's computational demands with a parallel, pipelined architecture implemented on a single System-on-Chip (SoC). This design allows for the adaptation of various hardware accelerators tailored for different PPO phases. A key innovation is our strategic standardization technique, which combines dynamic reward standardization and block standardization for values, followed by 8-bit uniform quantization. This method stabilizes learning, enhances performance, and manages memory bottlenecks, achieving a 4x reduction in memory usage and a 1.5x increase in cumulative rewards. We propose a solution on a single SoC device with programmable logic and embedded processors, delivering throughput orders of magnitude higher than traditional CPU-GPU systems. Our single-chip solution minimizes communication latency and throughput bottlenecks, significantly boosting PPO training efficiency. Experimental results show a 30% increase in PPO speed and a substantial reduction in memory access time, underscoring HEPPO-GAE's potential for broad applicability in hardware-efficient reinforcement learning algorithms.

强化学习FPGA加速PPOGAE

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。