Coarse-to-Fine Q-Network (CQN) in RL
- Coarse-to-Fine Q-Network (CQN) is a reinforcement learning method that uses adaptive hierarchical action discretization to solve continuous control tasks.
- It employs decoupled Q-learning with dimension-wise value decomposition and n-step temporal-difference updates to manage high-dimensional continuous spaces.
- CQN integrates behavior cloning, prioritized replay, and sequence-based value prediction to enhance data efficiency and achieve robust empirical performance.
The Coarse-to-Fine Q-Network (CQN), sometimes called Growing Q-Network (GQN), is a family of value-based reinforcement learning algorithms that solve continuous control problems by dynamically discretizing the action space at multiple resolutions. CQN leverages hierarchical or adaptive action discretization, decoupled Q-value architectures, and techniques including distributional Q-learning, behavior cloning regularization, and prioritized replay to achieve data efficiency, scalability, and strong empirical performance in both simulated and real-world robotic tasks. Recent work demonstrates that CQN and its variants outperform both static discretization and standard actor–critic baselines, especially in high-dimensional, sparse-reward, or long-horizon settings (Seyde et al., 2024, Seo et al., 2024, Seo et al., 2024, James et al., 2022).
1. Adaptive and Hierarchical Action Discretization
CQN restructures control in continuous action spaces—the standard setting in robotics and physical simulation—into a sequence or schedule of discretization steps. The action space is covered by a grid, with each dimension partitioned into bins at the highest resolution. At any stage, only a subgrid (with bins per dimension, ) is active; the agent chooses actions from this coarser mask and the Bellman backup maximizes only over this subgrid.
Adaptation of control resolution is achieved by:
- A fixed linear schedule: increment every episodes.
- An adaptive criterion: increment when the running mean return and standard deviation fail to improve beyond a threshold .
Alternative CQN frameworks use a hierarchical “zoom-in” progression: for each of discretization levels, each dimension is iteratively subdivided into bins, and the highest-valued subinterval is recursively refined. The process either finishes at the finest level or dynamically adapts to task learning progress (Seyde et al., 2024, Seo et al., 2024, Seo et al., 2024).
2. Q-Learning Updates and Decoupled Value Decomposition
CQN algorithms perform temporal-difference (TD) learning with n-step targets and double Q-learning, using a target network ():
To make training tractable in high action dimensions, decoupled Q-networks (DecQN) decompose the Q-function as:
is a univariate head for the -th action dimension, each outputting a vector of Q-values. At any epoch, only the first bins are unmasked. The policy is derived as a sum of independent utilities, providing scalability to dimensions (Seyde et al., 2024).
The CQN architecture also extends to action-sequence Q-values (CQN-AS): the critic outputs -values for a sequence (or block) of future actions, enabling richer temporal credit assignment and regularization (Seo et al., 2024).
3. Learning Objectives and Auxiliary Losses
CQN combines several objective components:
- n-step TD loss (with distributional or standard regression): per minibatch, minimize or its distributional equivalent.
- Behavior cloning (BC) loss on demonstration data: a large-margin hinge or MSE term that encourages the Q-value of expert actions to surpass other actions by a margin. For example,
The overall loss is weighted,
with typical weights , (Seo et al., 2024).
Other enhancements include prioritized experience replay, target network Polyak averaging (e.g., ), and exploration via -greedy or policy noise.
4. Network Architecture and Implementation
The canonical CQN employs:
- A shared multi-layer perceptron backbone (2 layers, 512 ReLU/SiLU units, LayerNorm).
- Per-action-dimension output heads, linear over bins.
- Input: concatenation of visual encoded features, proprioception, previous chosen actions, one-hot discretization level encoding.
- Typical settings: levels, –$5$ bins, total action dimensions –$39$ (Seyde et al., 2024, Seo et al., 2024).
For CQN-AS, block inputs at each step in the sequence comprise a GRU-RNN embedding to share temporal context across the action sequence (Seo et al., 2024).
Replay buffers mix online experiences and expert demonstrations; training augments the demo buffer with successful online episodes. Gradients are updated via Adam or AdamW ( or ), batch size –$256$, target network update every 1k steps.
5. Empirical Performance and Comparative Evaluation
CQN and its variants have been extensively evaluated in:
- DeepMind Control Suite (continuous control, up to ): Growing Q-Network outperforms static DecQN discretizations, matching continuous D4PG and DMPO. Adaptive scheduling converges faster and more stably (Seyde et al., 2024).
- RLBench and real-world robotic manipulation: CQN attains 80% success in steps, exceeding DrQ-v2+ and behavior cloning-only baselines. With sufficient demos and distributional critics, CQN solves 4 real-world tasks in $600$–$1000$ steps (5–8 minutes) (Seo et al., 2024).
- MetaWorld and MyoSuite (biomechanical, ): Larger MLPs and coarse-to-fine schedules enable scaling to very high dimensions. GQN- achieves high final success (Seyde et al., 2024).
- Hard, sparse, or long-horizon tasks: CQN-AS surpasses ACT (action-chunking BC), vanilla CQN (no-sequence), and SAC on BiGym and RLBench, particularly where long sequences or global–local strategy is required (Seo et al., 2024).
- Ablations confirm that CQN’s adaptive/coarse-to-fine schedule outperforms static binning, and that BC loss and distributional critics are critical for stability and performance.
6. Theoretical Rationale and Insights
Coarse discretization (e.g., two bins—bang-bang control) incentivizes wide exploration by covering large regions of action space, evading local optima introduced by action penalties. Progressive refinement (“coarse-to-fine”) introduces fine granularity only as the task requires, yielding precision and actuation smoothness.
Decoupled (dimensionwise) Q-learning transforms the exponential joint space into linearly-scaling heads, permitting tractable learning and inference in high dimensions (Seyde et al., 2024). Adaptive schedules using performance-based thresholds prevent premature over-refinement, dynamically trading off exploration and exploitation.
In action-sequence CQN (CQN-AS), sequence prediction structures the value function for multi-step credit assignment and regularizes away mode collapse or excessive temporal variance (Seo et al., 2024).
7. Representative Algorithm and Configuration Table
| Variant | Key Ideas | Empirical Setting |
|---|---|---|
| GQN (CQN) | Grow active bin set; DecQN heads | DeepMind Control Suite, M=38 |
| CQN (Seo et al., 2024) | Hierarchical “zoom-in” via L,B | RLBench/Real-World, L=3, B=5 |
| CQN-AS | Sequence value prediction | BiGym (K=16), RLBench (K=4) |
Each variant implements hierarchical or adaptive bin discretization, dimensionwise decoupling, and multi-task deployment with shared architecture and replay. Hyperparameters including levels , bins , n-step return, replay weighting, and network size are selected empirically and tuned per benchmark.
8. References
- “Growing Q-Networks: Solving Continuous Control Tasks with Adaptive Control Resolution” (Seyde et al., 2024)
- “Continuous Control with Coarse-to-fine Reinforcement Learning” (Seo et al., 2024)
- “Coarse-to-fine Q-Network with Action Sequence for Data-Efficient Robot Learning” (Seo et al., 2024)
- “Coarse-to-fine Q-attention with Tree Expansion” (James et al., 2022)