Awesome World Model Hub 论文 · 数据集 · 项目
← 返回论文列表

STORM: 基于随机Transformer的高效世界模型用于强化学习

NIPS 23 2023 80.2 method

TLDR

STORM结合Transformer与VAE构建高效随机世界模型,在Atari 100k上达126.7%人类表现且训练快速。

评分理由

The paper presents a novel architecture that effectively integrates sequence modeling and stochasticity, achieving state-of-the-art results on a standard benchmark with notable training efficiency. However, the evaluation is limited to the Atari domain, and the abstract does not discuss real-world applications or limitations beyond simulation.

Read-first 评分解释

综合优先阅读分 80.2,由主题、引用、图谱、方法、可复现性和近期性等信号加权得到。 原始总分保留为 58。

主题相关性 42%
82.9

使用现有 LLM 关键词相关性评分,并归一化到 0-100。 关键词:world model、world simulator、generative world model、interactive world model、video world model、world dynamics prediction、model-based reinforcement learning world model

可复现性 25%
81

检查链接和可见文本中的论文、代码、数据集、工件与仓库信号。 论文:有;代码:有;数据:无;命中信号:代码、GitHub

方法质量 25%
80

检查可见的摘要与分析字段,寻找实验、数据集、基线、指标和局限性等方法证据。 命中信号:基准、评估、指标、结果

近期性 8%
65.1

使用温和的时间衰减,让近期论文更容易浮现,同时保留较早基础工作的价值。 年份:2023

研究版图角色

方法锚点复现锚点

排序敏感性

稳定性:volatile;排名波动范围:47。

关键词评分

world model
10
model-based reinforcement learning world model
10
generative world model
9
world simulator
8
world dynamics prediction
8
interactive world model
7
video world model
6

深度分析

创新点

  • 将Transformer架构与变分自编码器相结合,用于基于模型的强化学习中的随机世界建模
  • 在不使用前瞻搜索技术的情况下,在Atari 100k基准测试上达到126.7%的平均人类表现,创下新纪录
  • 高效训练:在单张NVIDIA GeForce RTX 3090上仅需4.3小时即可完成1.85小时实时交互经验的训练

方法

STORM是一种世界模型架构,它整合了Transformer的序列建模与生成能力以及变分自编码器的随机特性。世界模型通过自监督学习在环境的视觉输入上进行训练,智能体的策略通过利用模型生成的想象轨迹进行改进。在Atari 100k基准测试上进行评估,与不使用前瞻搜索的最先进方法进行比较,评估指标包括平均人类表现和训练时间。

关键结果

STORM在Atari 100k基准测试上达到了126.7%的平均人类表现,创下了不使用前瞻搜索方法的新纪录。在单张RTX 3090上,训练具有1.85小时实时交互经验的智能体仅需4.3小时,相比先前方法展现了更高的效率。

技术栈

TransformerVariational Autoencoder (VAE)Atari 100kNVIDIA GeForce RTX 3090

标签