Research map / Reinforcement learning / Policy gradient and actor-critic
Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulation
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 Adaptive Horizon Actor-Critic (AHAC), a first-order model-based RL method on differentiable simulation that adapts rollout horizons to truncate at stiff contact, outperforming PPO/SAC/SVG/SHAC on locomotion tasks.
Problem
First-order model-based RL with differentiable simulation suffers from gradient sample error under stiff contact dynamics and long horizons, leading to suboptimal policies compared with model-free RL in contact-rich continuous control.
Approach
Adapt the model-based rollout horizon via a Lagrangian constraint so that rollouts are truncated when contact stiffness ||∇f|| exceeds a threshold, avoiding erroneous first-order gradients; combined with a double critic, iterative critic training, and GPU-parallel rollouts.
Key results
- AHAC reaches 40% higher PPO-normalized IQM reward than the best MFRL baseline PPO across five locomotion tasks (10 seeds).
- On SNU Humanoid (152 action dims), AHAC achieves 6746 raw reward vs. PPO 4114 and SHAC 5924.
- On Ant, AHAC achieves 9313 raw reward vs. PPO 6605 and SHAC 7662.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Use AHAC instead of PPO | Locomotion tasks overall · Locomotion tasks overall | asymptotic reward: +40% reward | Helped |
| Use AHAC instead of PPO | AHAC · Ant | normalized asymptotic reward: 1.00 ± 0.12 -> 1.41 ± 0.08 | Helped |
| Use AHAC instead of SHAC | AHAC · Ant | normalized asymptotic reward: 1.16 ± 0.13 -> 1.41 ± 0.08 | Helped |
| Use AHAC instead of PPO | AHAC · SNU Humanoid | normalized asymptotic reward: +64% reward | Helped |
| Add an adaptive horizon objective (ablation) | SHAC · Ant | asymptotic episodic reward | Helped |
| Set a static horizon to H=29 (ablation) | SHAC · Ant | asymptotic episodic reward | Helped |
| Replace the static H=29 with an adaptive horizon (ablation) | SHAC · Ant | asymptotic episodic reward | Helped |
| Add a double critic (ablation) | AHAC · Ant | run-to-run reward variance | Helped |
| Train the critic iteratively until convergence (ablation) | AHAC · Humanoid and SNU Humanoid | asymptotic performance | Helped |
| Add graph truncation at stiff contacts to SHAC short-horizon rollouts (ablation) | SHAC · Contact-rich locomotion tasks | performance | No clear effect |
| Vary SHAC rollout horizon to H=28 (ablation) | SHAC · Ant | reward: H=28 | Helped |
Surprising findings
- Model-free PPO and SAC do not reach AHAC's asymptotic reward even when given roughly 100x more samples and 10x more training time, suggesting zeroth-order gradient variance limits MFRL optima.
- Truncating SHAC's short horizon at stiff contact (graph truncation) did not improve performance in the parallel setting.
Limitations
- Requires a differentiable simulator (dflex), which limits applicability to real-world settings with non-differentiable or unavailable dynamics.
- Results rely on dflex, which is no longer actively developed; the paper leaves transfer to real robots as a challenge.
- The contact threshold C and iterative critic criteria were not tuned, and AHAC was under-tuned relative to SHAC.
Experimental setup
Datasets: Hopper Ant Anymal Humanoid SNU Humanoid
Benchmarks: Hopper Ant Anymal Humanoid SNU Humanoid
Metrics: 50% IQM episodic reward 95% confidence interval PPO-normalized reward score distributions
Baselines: PPO SAC SVG SHAC
Builds on: SHAC PPO SAC SVG
Closest papers by meaning
- Differentiable Information Enhanced Model-Based Reinforcement Learning (AAAI 2025 · similarity 0.88)
- Accelerated Policy Learning with Parallel Differentiable Simulation (ICLR 2022 · similarity 0.88)
- Gradient Information Matters in Policy Optimization by Back-propagating through Model (ICLR 2022 · similarity 0.88)
- Robust Model Based Reinforcement Learning Using $\mathcal{L}_1$ Adaptive Control (ICLR 2024 · similarity 0.86)
- OCEAN-MBRL: Offline Conservative Exploration for Model-Based Offline Reinforcement Learning (AAAI 2024 · similarity 0.86)
- Diffusion Actor-Critic: Formulating Constrained Policy Iteration as Diffusion Noise Regression for Offline Reinforcement Learning (ICLR 2025 · similarity 0.85)
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.
