arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.15840cs.RO
收藏

QueST:面向连续控制学习的自监督技能抽象方法

QueST: Self-Supervised Skill Abstractions for Learning Continuous Control

Georgia Institute of Technology(佐治亚理工学院);Nvidia(英伟达)

Atharva Mete, Haotian Xue, Albert Wilcox, Yongxin Chen, Animesh Garg

AI总结:

针对机器人学习泛化能力不足的问题,提出量化技能Transformer(QueST),通过带因果归纳偏置的灵活潜编码学习可迁移的时序动作抽象,在多任务和少样本基准上表现优于现有基线。

摘要:

泛化能力不足是机器人学习领域最重要的未解决问题之一,尽管已有多项大规模研究着手解决这一问题,但该问题仍未得到攻克。本文提出假设:使用潜变量模型(LVMs,学习将数据映射到压缩潜空间并反向重建的模型)学习时序动作抽象,是获取可直接用于新任务的底层技能的一个有前景的方向。尽管已有多项研究试图证明这一点,但这些研究普遍受限于架构,无法忠实地捕获可共享的表示。为解决这一问题,我们提出量化技能Transformer(Quantized Skill Transformer,QueST),该模型学习一种规模更大、更灵活的潜编码,更有能力建模多种任务所需的各类底层技能。为利用这种额外的灵活性,QueST将动作序列数据中的因果归纳偏置注入潜空间,从而得到更具语义实用性、可迁移性更强的表示。我们将其与当前最优的模仿学习和LVM基线进行对比,结果表明QueST的架构在多项多任务和少样本学习基准上实现了优异性能。更多结果和视频可访问https://quest-model.github.io/查看。

英文摘要:

Generalization capabilities, or rather a lack thereof, is one of the most important unsolved problems in the field of robot learning, and while several large scale efforts have set out to tackle this problem, unsolved it remains. In this paper, we hypothesize that learning temporal action abstractions using latent variable models (LVMs), which learn to map data to a compressed latent space and back, is a promising direction towards low-level skills that can readily be used for new tasks. Although several works have attempted to show this, they have generally been limited by architectures that do not faithfully capture shareable representations. To address this we present Quantized Skill Transformer (QueST), which learns a larger and more flexible latent encoding that is more capable of modeling the breadth of low-level skills necessary for a variety of tasks. To make use of this extra flexibility, QueST imparts causal inductive bias from the action sequence data into the latent space, leading to more semantically useful and transferable representations. We compare to state-of-the-art imitation learning and LVM baselines and see that QueST's architecture leads to strong performance on several multitask and few-shot learning benchmarks. Further results and videos are available at https://quest-model.github.io/

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑