Dual-Flow Reinforcement Learning with State-Aware Exploration
About
In complex continuous-control reinforcement learning tasks, multimodal optimal actions often coincide with uncertain, multimodal return distributions, making reliable value estimation and multimodal exploration challenging. Existing value estimation methods using unimodal Gaussians restrict expressiveness and yield biased estimates. Recent generative policies can represent multimodal actions but often collapse to a few modes and under-explore high-value areas of the action space. Motivated by these challenges, we propose Dual-Flow RL, a unified actor-critic framework that jointly models a continuous return distribution and a multimodal policy distribution using conditional flow matching (CFM). This design supports reliable value estimation and sustained multimodal exploration. To further enhance exploration, we introduce an Entropy-Covariance Exploration Regulator (ECER) that enables state-aware exploration regulation leveraging policy entropy and action-uncertainty covariance. Experiments on DeepMind Control Suite and Humanoid-Bench show that Dual-Flow RL achieves state-of-the-art performance on most tasks, significantly outperforming prior diffusion-based and flow-based methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Locomotion | DeepMind Control suite Dog-Trot | Final Return938 | 17 | |
| Dog-run | DeepMind Control Suite (DMC) | Total Average Return628 | 13 | |
| H1balance_hard | Humanoid-bench | Total Average Return73 | 13 | |
| H1balance_simple | Humanoid-bench | Total Return354 | 13 | |
| H1crawl | Humanoid-bench | Total Average Return961 | 13 | |
| H1maze | Humanoid-bench | Total Average Return458 | 13 | |
| H1reach | Humanoid-bench | Total Average Return6.76e+3 | 13 | |
| H1sit_hard | Humanoid-bench | Total Average Return837 | 13 | |
| Humanoid Run | Humanoid-bench | Total Average Return433 | 13 | |
| Walker Walk | DeepMind Control Suite (DMC) | Total Average Return988 | 13 |