arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.11491q-bio.NCcs.CVcs.LGcs.NE
收藏

Selective Particle Attention:深度强化学习中基于视觉感知的注意

Selective Particle Attention: Visual Feature-Based Attention in Deep Reinforcement Learning

Birkbeck, University of London(伦敦大学伯贝克学院)

Sam Blakeman, Denis Mareschal

摘要:

人类大脑使用选择性注意来过滤感知输入,以便仅使用其有限的计算资源来处理对行为有用的成分。我们专注于一种特殊形式的视觉注意力,称为基于特征的注意力,它涉及识别视觉输入的特征,这些特征对当前任务很重要,而不管它们的空间位置如何。基于视觉特征的注意力通过降低状态表征的维数和引导学习向相关特征方向发展来提高强化学习(RL)的效率。尽管在复杂的感知运动任务中达到了人类水平的性能,但深度RL算法一直因其效率低下和缺乏灵活性而受到批评。因此,基于视觉特征的注意力是解决这些批评的一个选择。尽管如此,大脑如何能够学习在RL期间注意哪些特征仍然是一个悬而未决的问题。为了帮助回答这个问题,我们提出了一种新的算法,称为选择性粒子注意力(SPA),它赋予深度RL代理执行选择性基于特征的注意力的能力。SPA根据自下而上的显著性以及它们预测未来奖励的准确性来学习要关注的特征组合。我们评估SPA的多项选择任务和2D视频游戏,都涉及原始像素输入和动态变化的任务结构。我们展示了SPA的各种好处,天真地参加所有或随机子集的功能。我们的研究结果表明:(1)深度RL模型中基于视觉特征的注意力如何提高其学习效率和处理任务结构突然变化的能力;(2)粒子滤波器可能代表了基于视觉特征的注意力如何在大脑中发生的可行计算。

英文摘要:

The human brain uses selective attention to filter perceptual input so that only the components that are useful for behaviour are processed using its limited computational resources. We focus on one particular form of visual attention known as feature-based attention, which is concerned with identifying features of the visual input that are important for the current task regardless of their spatial location. Visual feature-based attention has been proposed to improve the efficiency of Reinforcement Learning (RL) by reducing the dimensionality of state representations and guiding learning towards relevant features. Despite achieving human level performance in complex perceptual-motor tasks, Deep RL algorithms have been consistently criticised for their poor efficiency and lack of flexibility. Visual feature-based attention therefore represents one option for addressing these criticisms. Nevertheless, it is still an open question how the brain is able to learn which features to attend to during RL. To help answer this question we propose a novel algorithm, termed Selective Particle Attention (SPA), which imbues a Deep RL agent with the ability to perform selective feature-based attention. SPA learns which combinations of features to attend to based on their bottom-up saliency and how accurately they predict future reward. We evaluate SPA on a multiple choice task and a 2D video game that both involve raw pixel input and dynamic changes to the task structure. We show various benefits of SPA over approaches that naively attend to either all or random subsets of features. Our results demonstrate (1) how visual feature-based attention in Deep RL models can improve their learning efficiency and ability to deal with sudden changes in task structure and (2) that particle filters may represent a viable computational account of how visual feature-based attention occurs in the brain.

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

↑