arXiv reaDer
暗黙の注意による強化学習のためのピクセルのロック解除
Unlocking Pixels for Reinforcement Learning via Implicit Attention
最近、視覚ベースの環境で強化学習(RL)エージェントをトレーニングすることに大きな関心が寄せられています。これは、高次元性や疑似相関による観測的過剰適合の可能性など、多くの課題をもたらします。これらの問題の両方を解決するための有望なアプローチは、自己注意のボトルネックです。これは、気が散る場合でも、高性能のポリシーを学習するためのシンプルで効果的なフレームワークを提供します。ただし、アテンションアーキテクチャのスケーラビリティが低いため、これらの方法は、大きなパッチ(したがって小さなアテンションマトリックス)を使用して、低解像度の視覚入力を超えて拡張することはできません。この論文では、トランスフォーマーに非常に効果的であることが最近示された新しい効率的な注意アルゴリズムを利用し、これらの新しい手法をRL設定に適用できることを示します。これにより、アテンションベースのコントローラーをより大きな視覚入力にスケーリングし、個々のピクセルでさえもより小さなパッチの使用を容易にして、一般化を改善できます。さらに、角度カーネルの理論を活用して、ソフトマックス注意をハイブリッドランダム機能と呼ぶものに近似する新しい効率的なアルゴリズムを提案します。視覚ベースのRLに注意を使用する場合、ハイブリッドランダム機能が有望なアプローチであることを理論的および経験的に示します。
There has recently been significant interest in training reinforcement learning (RL) agents in vision-based environments. This poses many challenges, such as high dimensionality and potential for observational overfitting through spurious correlations. A promising approach to solve both of these problems is a self-attention bottleneck, which provides a simple and effective framework for learning high performing policies, even in the presence of distractions. However, due to poor scalability of attention architectures, these methods do not scale beyond low resolution visual inputs, using large patches (thus small attention matrices). In this paper we make use of new efficient attention algorithms, recently shown to be highly effective for Transformers, and demonstrate that these new techniques can be applied in the RL setting. This allows our attention-based controllers to scale to larger visual inputs, and facilitate the use of smaller patches, even individual pixels, improving generalization. In addition, we propose a new efficient algorithm approximating softmax attention with what we call hybrid random features, leveraging the theory of angular kernels. We show theoretically and empirically that hybrid random features is a promising approach when using attention for vision-based RL.
updated: Mon Feb 08 2021 17:00:26 GMT+0000 (UTC)
published: Mon Feb 08 2021 17:00:26 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト