arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2310.16655cs.LG
收藏

面向图像强化学习中以控制为中心的表示

Towards Control-Centric Representations in Reinforcement Learning from Images

Beijing Institute of Electronic System Engineering(北京电子系统工程研究所);Peking University(北京大学);Beijing Institute of Technology(北京理工大学);University of the Sunshine Coast(阳光海岸大学)

Chen Liu, Hongyu Zang, Xin Li, Yong Heng, Yifei Wang, Zhen Fang, Yisen Wang, Mingzhong Wang

AI总结:

本文提出 ReBis,通过 Transformer 隐式建模动态并结合互模拟损失与非对称重构损失,在稀疏奖励下学习以控制为中心的图像强化学习表示,并在 Atari 和 DeepMind Control Suit 上验证了其优越性。

摘要:

基于图像的强化学习是一项实用但具有挑战性的任务。一个主要障碍在于提取以控制为中心的表示,同时忽略无关信息。尽管遵循互模拟原则的方法在通过学习状态表示来解决该问题上展现出潜力,但它们仍然受限于潜在动态的有限表达能力以及对稀疏奖励环境的不适应性。为了解决这些局限性,我们提出了 ReBis,其目标是通过将无奖励的控制信息与奖励特定的知识相结合来捕获以控制为中心的信息。ReBis 利用 Transformer 架构隐式建模动态,并引入分块掩码以消除时空冗余。此外,ReBis 将基于互模拟的损失与非对称重构损失相结合,以防止在稀疏奖励环境中发生特征坍缩。在包括 Atari 游戏和 DeepMind Control Suit 在内的两个大型基准上的实证研究表明,与现有方法相比,ReBis 具有更优越的性能,证明了其有效性。

英文摘要:

Image-based Reinforcement Learning is a practical yet challenging task. A major hurdle lies in extracting control-centric representations while disregarding irrelevant information. While approaches that follow the bisimulation principle exhibit the potential in learning state representations to address this issue, they still grapple with the limited expressive capacity of latent dynamics and the inadaptability to sparse reward environments. To address these limitations, we introduce ReBis, which aims to capture control-centric information by integrating reward-free control information alongside reward-specific knowledge. ReBis utilizes a transformer architecture to implicitly model the dynamics and incorporates block-wise masking to eliminate spatiotemporal redundancy. Moreover, ReBis combines bisimulation-based loss with asymmetric reconstruction loss to prevent feature collapse in environments with sparse rewards. Empirical studies on two large benchmarks, including Atari games and DeepMind Control Suit, demonstrate that ReBis has superior performance compared to existing methods, proving its effectiveness.

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

↑