用联想记忆增强潜在世界模型

研究者提出一种记忆增强的潜在世界模型，在 JEPA 式预测中加入内容可寻址记忆库，直接从观测学习高层表示，并在共享潜在空间内跨多个时间尺度运作，通过自适应层归一化调节预测。为防止表示坍塌并保证记忆有效利用，设计了两个互补损失以促进槽位多样性和时间利用均衡。在 PushT、Tworoom、Reacher 和 OGBCube 上，该方法以轻量架构取得 SOTA，长时程规划任务优于传统潜在世界模型。