arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.13629cs.LG
收藏

通过基于能量的归一化流实现最大熵强化学习

Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flow

National Tsing Hua University(国立清华大学);NVIDIA Corporation(英伟达公司)

Chen-Hao Chao, Chien Feng, Wei-Fang Sun, Cheng-Kuang Lee, Simon See, Chun-Yi Lee

AI总结:

本文提出 EBFlow 框架,用基于能量的归一化流统一 MaxEnt RL 的策略评估与改进,实现无 Monte Carlo 的 soft 价值计算、多模态动作建模和高效采样,并在 MuJoCo 与 Isaac Gym 任务中优于基线。

摘要:

现有的面向连续动作空间的最大熵(Maximum-Entropy,MaxEnt)强化学习(RL)方法通常基于 actor-critic 框架构建,并通过策略评估与策略改进的交替步骤进行优化。在策略评估步骤中,更新 critic 以捕捉 soft Q 函数;在策略改进步骤中,根据更新后的 soft Q 函数调整 actor。本文提出一种使用基于能量的归一化流(Energy-Based Normalizing Flows,EBFlow)建模的新型 MaxEnt RL 框架。该框架将策略评估步骤与策略改进步骤整合起来,形成单目标训练过程。我们的方法能够计算策略评估目标中使用的 soft 价值函数,而无需 Monte Carlo 近似。此外,该设计支持对多模态动作分布进行建模,同时便于高效的动作采样。为评估方法性能,我们在 MuJoCo 基准套件以及由 Omniverse Isaac Gym 模拟的多项高维机器人任务上进行了实验。评估结果表明,与广泛采用的代表性基线相比,我们的方法取得了更优的性能。

英文摘要:

Existing Maximum-Entropy (MaxEnt) Reinforcement Learning (RL) methods for continuous action spaces are typically formulated based on actor-critic frameworks and optimized through alternating steps of policy evaluation and policy improvement. In the policy evaluation steps, the critic is updated to capture the soft Q-function. In the policy improvement steps, the actor is adjusted in accordance with the updated soft Q-function. In this paper, we introduce a new MaxEnt RL framework modeled using Energy-Based Normalizing Flows (EBFlow). This framework integrates the policy evaluation steps and the policy improvement steps, resulting in a single objective training process. Our method enables the calculation of the soft value function used in the policy evaluation target without Monte Carlo approximation. Moreover, this design supports the modeling of multi-modal action distributions while facilitating efficient action sampling. To evaluate the performance of our method, we conducted experiments on the MuJoCo benchmark suite and a number of high-dimensional robotic tasks simulated by Omniverse Isaac Gym. The evaluation results demonstrate that our method achieves superior performance compared to widely-adopted representative baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑