深層強化学習におけるロバスト探索のための自己教師あり逐次情報ボトルネック
Self-supervised Sequential Information Bottleneck for Robust Exploration in Deep Reinforcement Learning
時系列観測の逐次予測情報を圧縮する情報ボトルネック目的を導入し、ノイズに頑健なタスク関連の新規性に基づく内在的報酬で探索を効率化する手法を提案。
著者: Bang You, Jingming Xie, Youping Chen, Jan Peters, Oleg Arenz
分類: cs.LG, cs.RO
原文アブストラクト
Effective exploration is critical for reinforcement learning agents in environments with sparse rewards or high-dimensional state-action spaces. Recent works based on state-visitation counts, curiosity and entropy-maximization generate intrinsic reward signals to motivate the agent to visit novel states for exploration. However, the agent can get distracted by perturbations to sensor inputs that contain novel but task-irrelevant information, e.g. due to sensor noise or changing background. In this work, we introduce the sequential information bottleneck objective for learning compressed and temporally coherent representations by modelling and compressing sequential predictive information in time-series observations. For efficient exploration in noisy environments, we further construct intrinsic rewards that capture task-relevant state novelty based on the learned representations. We derive a variational upper bound of our sequential information bottleneck objective for practical optimization and provide an information-theoretic interpretation of the derived upper bound. Our experiments on a set of challenging image-based simulated control tasks show that our method achieves better sample efficiency, and robustness to both white noise and natural video backgrounds compared to state-of-art methods based on curiosity, entropy maximization and information-gain.
