用世界模型辅助Q-learning,提升样本效率与性能。
Q-Learning With World Models

- 在真实环境中训练Q-learning,用世界模型在推理时搜索想象轨迹
- 在Robomimic和LIBERO上显著优于现有方法,样本效率更高
- 避免模型偏差累积,适合高维复杂任务如机器人操控
离线策略强化学习(RL)已实现更高的样本效率,可应用于视觉-语言-动作模型的微调以生成高性能策略。世界模型通过预测状态变化而非仅预测动作,进一步提升样本效率,但其成功主要局限于监督式策略学习。先前基于模型的强化学习方法通常直接在想象轨迹上优化策略或价值函数,易产生累积偏差,难以扩展至大型高维问题(如现实世界机器人),且随着任务周期和视觉复杂度增加而恶化。本文提出一种新框架QWM,将世界模型直接应用于标准Q-learning,在测试时通过想象轨迹搜索选择高价值动作,同时保持策略和价值函数在真实在线环境中训练。由于策略和价值函数仅基于真实过渡数据训练,QWM避免了模型偏差的累积,仍获得预测搜索带来的样本效率优势。在挑战性的机械臂操控基准任务Robomimic和LIBERO上,QWM在样本效率和性能上均显著超越现有先进方法。
原文摘要 · Abstract (English)
Off-policy reinforcement learning (RL) has become increasingly sample-efficient, enabling applications such as RL fine-tuning of Vision-Language-Action models into reliable, high-performing policies. World models offer a further lever for sample efficiency, as they predict state changes rather than actions alone, but their success has largely been confined to supervised policy learning. Prior model-based RL methods often optimize the policy or value function directly on imagined rollouts, which is prone to compounding bias and struggles to scale to large, high-dimensional problems such as real-world robotics, a problem that worsens with task horizon and visual complexity. In this work, we instead ask whether we can leverage world models directly on top of standard Q-learning to improve performance, while remaining trained and grounded in the real, online setting. We propose QWM, a framework that leverages world models to perform test-time search over imagined trajectories on top of Q-learning to select high-value actions during both online rollouts and evaluation. Since the policy and value function are trained only on real transitions, QWM avoids compounding model bias while still gaining the sample-efficiency benefits of predictive search. On challenging manipulation benchmarks Robomimic and LIBERO, QWM significantly outperforms strong prior state-of-the-art methods on both sample efficiency and performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。