atlas

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

CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity

ICLR 2024 · Reinforcement learning › Policy gradient and actor-critic · approach: Regularized scalable deep value learning

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 CrossQ, a UTD=1 SAC variant that removes target networks and uses Batch Renormalization in critics (with concatenated current and next-state batches in one forward pass) plus wider critics, matching REDQ/DroQ sample efficiency at a fraction of compute.

Problem

Sample efficiency in deep RL for continuous control; state-of-the-art high-UTD methods (REDQ, DroQ) achieve it only with UTD=20 and high computational cost.

Approach

Removing target networks and using BatchNorm (Batch Renormalization) in the critic, with current and next-state batches concatenated in one forward pass so normalization statistics cover both distributions, lets a UTD=1 SAC-based method learn as fast as high-UTD methods without ensembles or bias-reduction schemes.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
remove target networks and add BatchNorm (ablation)SAC · MuJoCo continuous-control environmentssample efficiencyHelped
remove target networks (ablation)SAC with tanh critic activations · MuJoCo continuous-control environmentssample efficiencyHelped
add BatchNorm while retaining target networks (ablation)SAC with tanh critic activations · MuJoCo continuous-control environmentssample efficiencyMixed
reduce critic width to 256 (ablation)CrossQ · MuJoCo continuous-control environmentssample efficiencyHurt
remove BRN layers (ablation)CrossQ · MuJoCo continuous-control environmentssample efficiencyHurt
replace BatchNorm with LayerNorm (ablation)CrossQ · MuJoCo continuous-control environmentssample efficiencyHurt
widen critic layers to 2048 (ablation)SAC · MuJoCo continuous-control environmentssample efficiencyNo clear effect
add BRN while retaining target networks (ablation)SAC · MuJoCo continuous-control environmentssample efficiencyHurt
use BRN in the critic only instead of in both actor and critic (ablation)CrossQ · MuJoCo continuous-control environmentssample efficiencyNo clear effect
replace BRN with original BatchNorm (ablation)CrossQ · MuJoCo continuous-control environmentstraining stabilityHurt
increase BRN moving-average momentum (ablation)CrossQ · MuJoCo continuous-control environmentstraining stabilityHelped
reduce Adam beta1 and use policy delay 3 (ablation)CrossQ · MuJoCo continuous-control environmentssample efficiencyNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MuJoCo Hopper MuJoCo Walker2d MuJoCo Ant MuJoCo Humanoid MuJoCo HalfCheetah HumanoidStandup DeepMind Control Suite

Benchmarks: MuJoCo Hopper MuJoCo Walker2d MuJoCo Ant MuJoCo Humanoid MuJoCo HalfCheetah HumanoidStandup DeepMind Control Suite

Metrics: episodic return IQM return wallclock time normalized Q estimation bias

Baselines: REDQ DroQ SAC TD3

Builds on: Soft Actor-Critic REDQ DroQ TD3 Batch Renormalization

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