atlas

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

ReLU to the Rescue: Improve Your On-Policy Actor-Critic with Positive Advantages

ICML 2024 · Reinforcement learning › Policy gradient and actor-critic · approach: Actor-critic theory · 2 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 VSOP, an A3C modification with ReLU-clipped advantages, spectral normalization, and dropout-based approximate Bayesian inference (Thompson sampling) for on-policy actor-critic, evaluated on MuJoCo and ProcGen.

Problem

Exploration in on-policy actor-critic deep RL is state-agnostic or non-adaptive; approximate Bayesian inference over actor weights is hard with the policy gradient objective.

Approach

Clip advantages to be non-negative (ReLU) so that the actor update fits a gamma-likelihood-style Bayesian form, use dropout as Bayesian approximation for actor and critic, spectral-normalize critic weights (motivated by a Lipschitz bound on an additive term), and sample actor parameters for Thompson-sampling exploration.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use VSOP instead of PPO, A3C, SAC, or TD3MuJoCo continuous control · MuJoCoMedianHelped
Use VSOP instead of PPO, A3C, SAC, or TD3MuJoCo continuous control · MuJoCoIQMHelped
Use VSOP instead of PPOProcGen generalization · ProcGenPPO-normalized scoresHelped
Use VSOP instead of PPOProcGen generalization · ProcGenMin-Max normalized IQMHelped
Use VSOP instead of PPOProcGen generalization · ProcGentest episodic returnMixed
Use VSOP instead of A3CVSOP with 2048 threads and 10 steps per rollout · MuJoCo Braxepisodic returnHelped
Use VSOP instead of PPOVSOP with 2048 threads and 10 steps per rollout · MuJoCo Braxepisodic returnHurt
Remove positive-advantage clipping (ablation)VSOP · MuJoCoperformanceHurt
Use all-actions policy optimization instead of single-action optimization (ablation)VSOP · MuJoCoperformanceHurt
Remove spectral normalization (ablation)VSOP · MuJoCo high-dimensional tasksperformanceHurt
Remove Thompson sampling (ablation)VSOP · MuJoCo high-dimensional tasksperformanceHurt
Add spectral normalization (ablation)VSOP · MuJoCo Brax massively parallel settingperformanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: MuJoCo ProcGen

Benchmarks: MuJoCo ProcGen

Metrics: median interquartile mean mean optimality gap probability of improvement area under return curve

Baselines: A3C PPO SAC TD3 RMPG VSPPO EDE

Builds on: A3C dropout as Bayesian approximation generalized advantage estimation regret matching policy gradient PPO

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