用流模型优化策略,让强化学习更好处理多解问题。
Flow-based Policy With Distributional Reinforcement Learning in Trajectory Optimization
- 用流匹配建模策略,能拟合复杂多模态分布。
- 结合分布强化学习,优化回报全分布而非单一均值。
- 在MuJoCo上性能超前,适合多解控制任务。
强化学习在复杂控制与决策任务中表现优异,但传统算法通常将策略参数化为对角高斯分布,难以捕捉多模态分布,导致在多解问题中无法覆盖全部最优解,且回报仅被简化为均值,丢失其多模态特性,从而削弱对策略更新的指导作用。为此,我们提出一种名为基于流的策略与分布强化学习(FP-DRL)的算法。该算法采用流匹配建模策略,兼具计算效率与拟合复杂分布的能力;同时引入分布强化学习,建模并优化完整的回报分布,更有效地引导多模态策略更新,提升智能体性能。在MuJoCo基准测试中的实验表明,FP-DRL在多数控制任务中达到当前最优(SOTA)性能,且流策略展现出卓越的表示能力。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) has proven highly effective in addressing complex control and decision-making tasks. However, in most traditional RL algorithms, the policy is typically parameterized as a diagonal Gaussian distribution, which constrains the policy from capturing multimodal distributions, making it difficult to cover the full range of optimal solutions in multi-solution problems, and the return is reduced to a mean value, losing its multimodal nature and thus providing insufficient guidance for policy updates. In response to these problems, we propose a RL algorithm termed flow-based policy with distributional RL (FP-DRL). This algorithm models the policy using flow matching, which offers both computational efficiency and the capacity to fit complex distributions. Additionally, it employs a distributional RL approach to model and optimize the entire return distribution, thereby more effectively guiding multimodal policy updates and improving agent performance. Experimental trails on MuJoCo benchmarks demonstrate that the FP-DRL algorithm achieves state-of-the-art (SOTA) performance in most MuJoCo control tasks while exhibiting superior representation capability of the flow policy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。