arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04419cs.LGcs.AIcs.CV
收藏

面向强化学习的以对象为中心表示预训练研究

An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning

SAP(思爱普);Rutgers University(罗格斯大学);ETRI(韩国电子通信研究院);KAIST(韩国科学技术院)

Jaesik Yoon, Yi-Fu Wu, Heechul Bae, Sungjin Ahn

AI总结:

本文通过引入以对象为中心的视觉强化学习基准,系统评估了无监督OCR预训练对图像强化学习的有效性,揭示了其在性能提升、分布外泛化方面的作用及在复杂场景中的局限。

摘要:

无监督的以对象为中心表示(OCR)学习近年来作为一种新的视觉表示范式引起了关注。这是因为它有潜力作为一种有效的预训练技术,用于各种下游任务,在样本效率、系统泛化和推理方面具有优势。尽管基于图像的强化学习(RL)是最重要且因此最常被提及的此类下游任务之一,但迄今为止,其在强化学习中的益处尚未得到系统性的研究。相反,大多数评估都集中在较为间接的指标上,例如分割质量和对象属性预测准确率。在本文中,我们通过实证实验研究了OCR预训练对基于图像的强化学习的有效性。为了进行系统性评估,我们引入了一个简单的以对象为中心的视觉强化学习基准,并进行了实验以回答诸如“OCR预训练是否能提高以对象为中心任务的性能?”以及“OCR预训练能否帮助实现分布外泛化?”等问题。我们的结果为深入理解OCR预训练在强化学习中的有效性及其在某些场景下使用的潜在局限性提供了实证证据。此外,本研究还考察了在强化学习中引入OCR预训练的关键方面,包括在视觉复杂环境中的性能以及用于聚合对象表示的合适的池化层。

英文摘要:

Unsupervised object-centric representation (OCR) learning has recently drawn attention as a new paradigm of visual representation. This is because of its potential of being an effective pre-training technique for various downstream tasks in terms of sample efficiency, systematic generalization, and reasoning. Although image-based reinforcement learning (RL) is one of the most important and thus frequently mentioned such downstream tasks, the benefit in RL has surprisingly not been investigated systematically thus far. Instead, most of the evaluations have focused on rather indirect metrics such as segmentation quality and object property prediction accuracy. In this paper, we investigate the effectiveness of OCR pre-training for image-based reinforcement learning via empirical experiments. For systematic evaluation, we introduce a simple object-centric visual RL benchmark and conduct experiments to answer questions such as ``Does OCR pre-training improve performance on object-centric tasks?'' and ``Can OCR pre-training help with out-of-distribution generalization?''. Our results provide empirical evidence for valuable insights into the effectiveness of OCR pre-training for RL and the potential limitations of its use in certain scenarios. Additionally, this study also examines the critical aspects of incorporating OCR pre-training in RL, including performance in a visually complex environment and the appropriate pooling layer to aggregate the object representations.

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

↑