arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03329cs.CVcs.AI

无监督显著块选择用于数据高效强化学习

Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learning

UM-SJTU Joint Institute(上海交通大学密西根联合学院)

Zhaohui Jiang, Paul Weng

AI总结:

提出SPIRL方法,利用自监督预训练的视觉Transformer自动选择图像中的显著块,通过注意力模块处理,提高视觉RL的样本效率,并在Atari游戏上验证了其数据效率优势。

摘要:

为了提高基于视觉的深度强化学习(RL)的样本效率,我们提出了一种名为SPIRL的新方法,用于从输入图像中自动提取重要块。遵循掩码自编码器,SPIRL基于视觉Transformer模型,这些模型以自监督方式预训练,从随机采样的块中重建图像。这些预训练模型随后可用于检测和选择显著块,这些块被定义为难以从相邻块重建的块。在RL中,SPIRL智能体通过注意力模块处理选定的显著块。我们在Atari游戏上实证验证了SPIRL,以测试其相对于相关最先进方法(包括一些传统的基于模型的方法和基于关键点的模型)的数据效率。此外,我们分析了模型的可解释性能力。

英文摘要:

To improve the sample efficiency of vision-based deep reinforcement learning (RL), we propose a novel method, called SPIRL, to automatically extract important patches from input images. Following Masked Auto-Encoders, SPIRL is based on Vision Transformer models pre-trained in a self-supervised fashion to reconstruct images from randomly-sampled patches. These pre-trained models can then be exploited to detect and select salient patches, defined as hard to reconstruct from neighboring patches. In RL, the SPIRL agent processes selected salient patches via an attention module. We empirically validate SPIRL on Atari games to test its data-efficiency against relevant state-of-the-art methods, including some traditional model-based methods and keypoint-based models. In addition, we analyze our model's interpretability capabilities.

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

↑