用W形网络增强强化学习的自适应势函数,让高维游戏更快学会。
APF+: Boosting adaptive-potential function reinforcement learning methods with a W-shaped network for high-dimensional games
- 设计W形网络编码器,分离捕捉游戏背景与移动物体特征
- 在20个Atari游戏中,14次超越基础DDQN,13次胜过其他编码方法
- 无需真实状态信息,也能达到接近真实状态表示的效果
近年来,奖励塑形在强化学习中备受关注,因其能加速训练过程。此前工作提出自适应势函数(APF),在低维环境下显著提升多层感知机的Q-learning效率。本文将APF扩展为APF+,引入编码器以处理像素输入的高维游戏环境,采用一种W形网络(W-Net)将高维游戏帧映射为低维嵌入。W-Net可同时编码背景和运动物体,其生成的嵌入包含两个潜在向量:一个表示输入状态,另一个表示状态表示的偏差。将W-Net与APF结合,训练下游双通道深度Q网络(DDQN),形成APF-WNet-DDQN,并在Atari游戏任务中验证有效性。对比两类基线:(i) 基础DDQN;(ii) 用ST-DIM或真实状态接口(ARI)替换编码器的APF-DDQN。实验显示,在20个Atari游戏中,APF-WNet-DDQN在14个上优于DDQN,13个上优于APF-ST-DIM-DDQN;与使用真实状态信息的APF-ARI-DDQN相比,性能相当。
原文摘要 · Abstract (English)
Studies in reward shaping for reinforcement learning (RL) have flourished in recent years due to its ability to speed up training. Our previous work proposed an adaptive potential function (APF) and showed that APF can accelerate the Q-learning with a Multi-layer Perceptron algorithm in the low-dimensional domain. This paper proposes to extend APF with an encoder (APF+) for RL state representation, allowing applying APF to the pixel-based Atari games using a state-encoding method that projects high-dimensional game's pixel frames to low-dimensional embeddings. We approach by designing the state-representation encoder as a W-shaped network (W-Net), by using which we are able to encode both the background as well as the moving entities in the game frames. Specifically, the embeddings derived from the pre-trained W-Net consist of two latent vectors: One represents the input state, and the other represents the deviation of the input state's representation from itself. We then incorporate W-Net into APF to train a downstream Dueling Deep Q-Network (DDQN), obtain the APF-WNet-DDQN, and demonstrate its effectiveness in Atari game-playing tasks. To evaluate the APF+W-Net module in such high-dimensional tasks, we compare with two types of baseline methods: (i) the basic DDQN; and (ii) two encoder-replaced APF-DDQN methods where we replace W-Net by (a) an unsupervised state representation method called Spatiotemporal Deep Infomax (ST-DIM) and (b) a ground truth state representation provided by the Atari Annotated RAM Interface (ARI). The experiment results show that out of 20 Atari games, APF-WNet-DDQN outperforms DDQN (14/20 games) and APF-STDIM-DDQN (13/20 games) significantly. In comparison against the APF-ARI-DDQN which employs embeddings directly of the detailed game-internal state information, the APF-WNet-DDQN achieves a comparable performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。