Search papers, labs, and topics across Lattice.
This paper investigates whether the bias introduced by discontinuous dynamics in differentiable simulators is the primary obstacle to effective policy gradient estimation. They introduce DDCG, a lightweight test that switches between 1st-order and 0th-order estimators in nonsmooth regions, and IVW-H, a per-step inverse-variance weighting method for differentiable robotics control. Results suggest that while estimator switching improves robustness in controlled settings, variance control is often more critical for practical deployments.
Discontinuous dynamics aren't the biggest problem for policy gradients in differentiable simulators; variance control often matters more.
In policy gradient reinforcement learning, access to a differentiable model enables 1st-order gradient estimation that accelerates learning compared to relying solely on derivative-free 0th-order estimators. However, discontinuous dynamics cause bias and undermine the effectiveness of 1st-order estimators. Prior work addressed this bias by constructing a confidence interval around the REINFORCE 0th-order gradient estimator and using these bounds to detect discontinuities. However, the REINFORCE estimator is notoriously noisy, and we find that this method requires task-specific hyperparameter tuning and has low sample efficiency. This paper asks whether such bias is the primary obstacle and what minimal fixes suffice. First, we re-examine standard discontinuous settings from prior work and introduce DDCG, a lightweight test that switches estimators in nonsmooth regions; with a single hyperparameter, DDCG achieves robust performance and remains reliable with small samples. Second, on differentiable robotics control tasks, we present IVW-H, a per-step inverse-variance implementation that stabilizes variance without explicit discontinuity detection and yields strong results. Together, these findings indicate that while estimator switching improves robustness in controlled studies, careful variance control often dominates in practical deployments.
