arXiv:2505.10010cs.LG2025-05被引 3

用大模型生成虚拟数据,评估强化学习在新任务上的表现

ImagineBench: Evaluating Reinforcement Learning with Large Language Model Rollouts

  • 构建首个融合真实与大模型生成数据的强化学习评测基准
  • 在复杂任务中仅35.44%成功率,远低于真实数据训练的64.37%
  • 适合研究大模型辅助强化学习、零样本迁移的学者参考

强化学习依赖大量真实交互数据来学习特定任务策略。近期研究表明,大语言模型(LLMs)可通过生成合成经验(称为虚拟推演)缓解这一限制,但该领域进展受限于缺乏标准评测基准。为此,我们提出ImagineBench,首个全面评估结合真实与LLM虚拟推演的离线强化学习算法的基准。其关键特征包括:(1) 包含环境采集与LLM生成推演的数据集;(2) 覆盖运动、机器人操作和导航等多样任务域;(3) 提供不同复杂度的自然语言指令,支持语言条件策略学习。对先进离线强化学习算法的系统评估显示,直接应用现有算法在未见任务上表现不佳,复杂任务成功率为35.44%,远低于真实数据训练的64.37%。这凸显了改进算法以更好利用虚拟推演的必要性。此外,我们识别出未来研究机遇:包括更优利用虚拟数据、快速在线适应与持续学习,以及扩展至多模态任务。代码已开源:https://github.com/LAMDA-RL/ImagineBench。

原文摘要 · Abstract (English)

A central challenge in reinforcement learning (RL) is its dependence on extensive real-world interaction data to learn task-specific policies. While recent work demonstrates that large language models (LLMs) can mitigate this limitation by generating synthetic experience (noted as imaginary rollouts) for mastering novel tasks, progress in this emerging field is hindered due to the lack of a standard benchmark. To bridge this gap, we introduce ImagineBench, the first comprehensive benchmark for evaluating offline RL algorithms that leverage both real rollouts and LLM-imaginary rollouts. The key features of ImagineBench include: (1) datasets comprising environment-collected and LLM-imaginary rollouts; (2) diverse domains of environments covering locomotion, robotic manipulation, and navigation tasks; and (3) natural language task instructions with varying complexity levels to facilitate language-conditioned policy learning. Through systematic evaluation of state-of-the-art offline RL algorithms, we observe that simply applying existing offline RL algorithms leads to suboptimal performance on unseen tasks, achieving 35.44% success rate in hard tasks in contrast to 64.37% of method training on real rollouts for hard tasks. This result highlights the need for algorithm advancements to better leverage LLM-imaginary rollouts. Additionally, we identify key opportunities for future research: including better utilization of imaginary rollouts, fast online adaptation and continual learning, and extension to multi-modal tasks. Our code is publicly available at https://github.com/LAMDA-RL/ImagineBench.

强化学习大模型虚拟数据评测基准

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