跳到正文
arXiv cs.LG· Daniel Lawson, Dhruv Sreenivas, Adriana Hugessen, Khimya Khetarpal, Glen Berseth·· 7 小时前AI 评分22

用几何视界模型想象目标,提升策略泛化能力

Imagining Goals with Geometric Horizon Models for Policy Generalization

AI 导读

研究者提出 TD-Aug,通过从几何视界模型(geometric horizon model)中采样,直接想象由已有知识组合而成的新结果,以缓解世界模型在长视界生成上的困难。将这些未来结果作为目标用于目标条件 BC 策略训练,在基于拼接的 OGBench 任务上显著提升了零样本目标泛化能力。

来源:arXiv cs.LG · arxiv.org