arXiv:2411.04534cs.LG2024-11中稿 · MLMI 2025

提出超立方体正则化框架,提升离线强化学习在低质量数据下的表现。

Hypercube Policy Regularization Framework for Offline Reinforcement Learning

  • 用超立方体约束策略,允许智能体在相似状态下探索邻近动作。
  • 在D4RL上优于IQL、CQL等主流算法,尤其在低质量数据下提升显著。
  • 适用于数据质量差但需高效策略的离线强化学习场景。

离线强化学习因其无需智能体与环境交互即可通过静态数据集学习策略而受到广泛关注。然而,传统强化学习方法在离线场景中因训练时遭遇数据集未覆盖的分布外状态动作而表现不佳。为此,直接模仿静态数据集中策略的正则化方法因其简洁有效受到重视。但此类方法通常过于保守,导致策略次优,尤其在低质量数据下问题更严重。本文提出超立方体策略正则化框架,通过允许智能体在相似状态下探索对应动作,缓解了传统约束的过度保守性,提升了算法在低质量数据上的有效性。理论证明该框架可显著改善原算法性能。将其与TD3-BC和Diffusion-QL结合,在D4RL数据集上分别命名为TD3-BC-C和Diffusion-QL-C,实验结果表明二者在多数环境中表现优于IQL、CQL、TD3-BC和Diffusion-QL等当前最优算法,且接近实时性能。

原文摘要 · Abstract (English)

Offline reinforcement learning has received extensive attention from scholars because it avoids the interaction between the agent and the environment by learning a policy through a static dataset. However, general reinforcement learning methods cannot get satisfactory results in offline reinforcement learning due to the out-of-distribution state actions that the dataset cannot cover during training. To solve this problem, the policy regularization method that tries to directly clone policies used in static datasets has received numerous studies due to its simplicity and effectiveness. However, policy constraint methods make the agent choose the corresponding actions in the static dataset. This type of constraint is usually over-conservative, which results in suboptimal policies, especially in low-quality static datasets. In this paper, a hypercube policy regularization framework is proposed, this method alleviates the constraints of policy constraint methods by allowing the agent to explore the actions corresponding to similar states in the static dataset, which increases the effectiveness of algorithms in low-quality datasets. It was also theoretically demonstrated that the hypercube policy regularization framework can effectively improve the performance of original algorithms. In addition, the hypercube policy regularization framework is combined with TD3-BC and Diffusion-QL for experiments on D4RL datasets which are called TD3-BC-C and Diffusion-QL-C. The experimental results of the score demonstrate that TD3-BC-C and Diffusion-QL-C perform better than state-of-the-art algorithms like IQL, CQL, TD3-BC and Diffusion-QL in most D4RL environments in approximate time.

离线强化学习策略正则化D4RLTD3-BC

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。