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

TL;DR

HEPPO-GAE is an FPGA-based accelerator that significantly improves the efficiency of PPO training by optimizing the GAE computation stage with a novel parallel, pipelined architecture and memory standardization techniques.

Contribution

The paper presents a hardware-efficient FPGA accelerator for GAE in PPO, introducing a standardization and quantization method that reduces memory usage and boosts training speed.

Findings

Achieves 4x reduction in memory usage.

Increases PPO training speed by 30%.

Reduces memory access time significantly.

Abstract

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.