arXiv:2502.04141cs.LG2025-02ICLR被引 6

用行为熵指导数据生成,提升离线强化学习的探索效率。

Behavioral Entropy-Guided Dataset Generation for Offline Reinforcement Learning

  • 基于行为熵设计连续空间探索目标,实现更优状态覆盖。
  • 在MuJoCo上实验显示,该方法生成的数据使下游任务性能提升80%以上。
  • 适合需要高效探索的复杂环境离线强化学习研究者。

基于熵的目标广泛用于强化学习中的状态空间探索及离线强化学习的数据集生成。行为熵(BE)是一种结合了智能体认知与感知偏见的严格推广,在离散场景中被提出并展现出在机器人探索问题上的潜力。本文将BE扩展至连续空间,推导出可计算的k近邻估计器,给出其理论保证,并设计适用于标准RL方法的奖励函数,以学习最大化行为熵的策略。在标准MuJoCo环境中,我们对比了使用BE、Rényi、Shannon熵最大化策略以及SMM和RND算法生成的数据集在多种下游任务上的表现。结果表明,使用BE生成的数据集训练的离线强化学习算法在所有任务上均优于使用Shannon熵、SMM和RND生成的数据集,且在80%的任务上优于Rényi熵生成的数据集。

原文摘要 · Abstract (English)

Entropy-based objectives are widely used to perform state space exploration in reinforcement learning (RL) and dataset generation for offline RL. Behavioral entropy (BE), a rigorous generalization of classical entropies that incorporates cognitive and perceptual biases of agents, was recently proposed for discrete settings and shown to be a promising metric for robotic exploration problems. In this work, we propose using BE as a principled exploration objective for systematically generating datasets that provide diverse state space coverage in complex, continuous, potentially high-dimensional domains. To achieve this, we extend the notion of BE to continuous settings, derive tractable $k$-nearest neighbor estimators, provide theoretical guarantees for these estimators, and develop practical reward functions that can be used with standard RL methods to learn BE-maximizing policies. Using standard MuJoCo environments, we experimentally compare the performance of offline RL algorithms for a variety of downstream tasks on datasets generated using BE, Rényi, and Shannon entropy-maximizing policies, as well as the SMM and RND algorithms. We find that offline RL algorithms trained on datasets collected using BE outperform those trained on datasets collected using Shannon entropy, SMM, and RND on all tasks considered, and on 80% of the tasks compared to datasets collected using Rényi entropy.

强化学习数据生成行为熵离线学习

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