atlas

Research map / Reinforcement learning / Policy gradient and actor-critic

SAPG: Split and Aggregate Policy Gradients

ICML 2024 · Reinforcement learning › Policy gradient and actor-critic · approach: Trajectory reuse in policy gradients · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SAPG, an on-policy RL method that splits parallel environments among multiple latent-conditioned policies and aggregates their data into a leader policy via clipped importance-sampled off-policy PPO updates, scaling to ~24k environments.

Problem

Standard on-policy RL such as PPO saturates in performance when scaled to tens of thousands of GPU-simulated parallel environments, because i.i.d. Gaussian action sampling yields many near-duplicate transitions and wasted environment capacity.

Approach

Divide environments into blocks, each run by a separate policy sharing a backbone with block-specific latent parameters (with entropy regularization on followers for diversity); a leader policy is updated with PPO on its own data plus importance-weighted off-policy data from the followers, while followers update only on their own data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use SAPG instead of PQLAllegroHand · AllegroHandepisode reward: 1.01 \mathrm{e}4 ± 5.28\mathrm{e}2 vs. 1.23 \mathrm{e}4 ± 3.29\mathrm{e}2Helped
use SAPG instead of PBTRegrasping · Regraspingsuccesses: 31.9 ± 2.26 vs. 35.7 ± 1.46Helped
use SAPG instead of PBTThrow · Throwsuccesses: 19.2 ± 1.07 vs. 23.7 ± 0.74Helped
use SAPG instead of PBTReorientation · Reorientationsuccesses: 23.2 ± 4.86 vs. 38.6 ± 0.63Helped
use SAPG instead of PBTTwo Arms Reorientation · Two Arms Reorientationsuccesses: 14.46 ± 2.91 vs. 28.58 ± 1.55Helped
use SAPG instead of PQLAllegroHand · AllegroHandepisode reward: 21%Helped
use SAPG instead of PQLShadowHand · ShadowHandepisode rewardNo clear effect
add entropy regularization with coefficient 5e-3 (ablation)SAPG · Reorientationsuccesses: up to 16.5%Helped
add entropy regularization (ablation)SAPG · AllegroHandepisode reward: 1.23 \mathrm{e}4 ± 3.29\mathrm{e}2 -> 9.14 \mathrm{e}3 ± 8.38\mathrm{e}2Hurt
remove leader-follower aggregation and use symmetric aggregation (ablation)SAPG · all tested environmentsperformanceHurt
remove importance-sampled off-policy data aggregation (ablation)SAPG · all tested environmentsperformanceHurt
remove subsampling and use the entire off-policy dataset (ablation)SAPG · AllegroHand and ShadowHandepisode rewardHurt

Surprising findings

Limitations

Experimental setup

Datasets: IsaacGym simulated Allegro-Kuka, Allegro Hand and Shadow Hand tasks

Benchmarks: AllegroKuka Regrasping AllegroKuka Throw AllegroKuka Reorientation AllegroKuka Two Arms Reorientation AllegroHand in-hand reorientation ShadowHand in-hand reorientation

Metrics: number of successes per episode episode reward

Baselines: PPO PBT (DexPBT) PQL (Parallel Q-Learning)

Builds on: PPO Population-Based Training (DexPBT) Parallel Q-Learning IsaacGym

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy