Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

XQC: Well-conditioned Optimization Accelerates Deep Reinforcement Learning

About

Sample efficiency is a central property of effective deep reinforcement learning algorithms. Recent work has improved this through added complexity, such as larger models, exotic network architectures, and more complex algorithms, which are typically motivated purely by empirical performance. We take a more principled approach by focusing on the optimization landscape of the critic network. Using the eigenspectrum and condition number of the critic's Hessian, we systematically investigate the impact of common architectural design decisions on training dynamics. Our analysis reveals that a novel combination of batch normalization (BN), weight normalization (WN), and a distributional cross-entropy (CE) loss produces condition numbers orders of magnitude smaller than baselines. This combination also naturally bounds gradient norms, a property critical for maintaining a stable effective learning rate under non-stationary targets and bootstrapping. Based on these insights, we introduce XQC: a well-motivated, sample-efficient deep actor-critic algorithm built upon soft actor-critic that embodies these optimization-aware principles. We achieve state-of-the art sample efficiency across 55 proprioception and 15 vision-based continuous control tasks, all while using significantly fewer parameters than competing methods. Our code is available at danielpalenicek.github.io/projects/xqc.

Daniel Palenicek, Florian Vogt, Joe Watson, Ingmar Posner, Jan Peters• 2025

Related benchmarks

TaskDatasetResultRank
Sparse-reward manipulationThreading simulated environment
Success Rate60
6
Sparse-reward manipulationNut Assembly simulated environment
Success Rate34
6
Sparse-reward manipulationCoffee simulated environment
Success Rate96
6
Sparse-reward manipulationMug Cleanup simulated environment
Success Rate72
6
Sparse-reward manipulationHammer Cleanup simulated environment
Success Rate100
6
Sparse-reward manipulationSquare simulated environment
Success Rate84
6
Showing 6 of 6 rows

Other info

Follow for update