arXiv ID:
2605.13013
arXiv 提交日期: 2026-05-13
JEDI:面向在线基于模型强化学习的联合嵌入扩散世界模型 / JEDI: Joint Embedding Diffusion World Model for Online Model-Based Reinforcement Learning
1️⃣ 一句话总结
本文提出JEDI,一种端到端训练的潜在扩散世界模型,通过将扩散去噪损失与JEPA预测性表征学习框架结合,在在线强化学习中既大幅降低了计算成本(显存减少43%、采样速度提升3倍以上),又在Atari100k任务上达到与像素级扩散模型相当甚至更优的性能。