arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.02437cs.ROcs.LG
收藏

模仿学习中的数据质量

Data Quality in Imitation Learning

Stanford University(斯坦福大学)

Suneel Belkhale, Yuchen Cui, Dorsa Sadigh

AI总结:

本文从分布偏移视角形式化模仿学习的数据质量,提出动作散度与转移多样性两项属性,并通过理论与实证分析揭示其对策略性能的交互影响。

摘要:

在监督学习中,近年来数据质量与数据整理的问题被能够摄取互联网规模数据的日益强大且富有表达力的模型所掩盖。然而,在机器人的离线学习中,我们根本没有互联网规模的数据,因此高质量数据集是必需品。在模仿学习(IL)这种利用专家演示进行机器人学习的样本高效范式中尤其如此。通过 IL 学到的策略在测试时会因动作预测中的误差累积而遭受状态分布偏移,这会导致策略无法从中恢复的未见过状态。与其设计新算法来解决分布偏移,另一种视角是开发评估和整理数据集的新方法。越来越多的证据表明,相同的 IL 算法在不同数据集上可能表现出显著不同的性能。这就需要一种形式化方法来定义“数据质量”指标,并进一步将其用于数据整理。在本工作中,我们通过分布偏移的视角,迈出了将模仿学习中的数据质量形式化的第一步:高质量数据集会促使策略在测试时保持在分布内。我们提出了塑造数据集质量的两个基本属性:i)动作散度:在某些状态下专家与所学策略之间的不匹配;以及 ii)转移多样性:在给定状态和动作下系统中存在的噪声。我们从理论上研究了模仿学习中这两个关键属性的联合效应,并对在多种不同数据源上训练的模型进行了实证分析。我们表明,状态多样性并不总是有益的,并展示了动作散度与转移多样性在实践中如何相互作用。

英文摘要:

In supervised learning, the question of data quality and curation has been over-shadowed in recent years by increasingly more powerful and expressive models that can ingest internet-scale data. However, in offline learning for robotics, we simply lack internet scale data, and so high quality datasets are a necessity. This is especially true in imitation learning (IL), a sample efficient paradigm for robot learning using expert demonstrations. Policies learned through IL suffer from state distribution shift at test time due to compounding errors in action prediction, which leads to unseen states that the policy cannot recover from. Instead of designing new algorithms to address distribution shift, an alternative perspective is to develop new ways of assessing and curating datasets. There is growing evidence that the same IL algorithms can have substantially different performance across different datasets. This calls for a formalism for defining metrics of "data quality" that can further be leveraged for data curation. In this work, we take the first step toward formalizing data quality for imitation learning through the lens of distribution shift: a high quality dataset encourages the policy to stay in distribution at test time. We propose two fundamental properties that shape the quality of a dataset: i) action divergence: the mismatch between the expert and learned policy at certain states; and ii) transition diversity: the noise present in the system for a given state and action. We investigate the combined effect of these two key properties in imitation learning theoretically, and we empirically analyze models trained on a variety of different data sources. We show that state diversity is not always beneficial, and we demonstrate how action divergence and transition diversity interact in practice.

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

↑