研究人员开发了JEDI,一种用于在线基于模型的强化学习的新型联合嵌入扩散世界模型。该模型通过在JEPA框架内使用去噪目标端到端地学习其潜在空间,解决了现有扩散世界模型在计算成本和性能之间的权衡问题。实证表明,JEDI在Atari100k上表现出具有竞争力的性能,优于具有单独训练的潜在空间的模型,同时还显著降低了VRAM使用量并加快了训练和采样时间。 AI
影响 通过将扩散模型与JEPA集成,引入了一种更有效的强化学习方法,有望提高性能并降低计算要求。
排序理由 发布了一篇详细介绍新型模型架构及其经验结果的新研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
在 Hugging Face Daily Papers 阅读 →
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →