arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.09754cs.NEcs.AIcs.LG

基于脉冲 Q-learning 的深度强化学习

Deep Reinforcement Learning with Spiking Q-learning

Shanghai Jiao Tong University(上海交通大学);Peking University(北京大学)

Ding Chen, Peixi Peng, Tiejun Huang, Yonghong Tian

AI总结:

针对现有 SNN 强化学习方法泛化不足或依赖 ANN 估值的问题,提出 DSQN,以非脉冲神经元膜电压表示 Q 值并实现端到端学习,在 17 款 Atari 游戏中多数优于 DQN,且更稳定、抗攻击。

摘要:

借助专用神经形态硬件,脉冲神经网络(SNN)有望以更低能耗实现人工智能(AI)。将 SNN 与深度强化学习(RL)相结合,可为现实控制任务提供一种颇具前景的节能途径。目前,现有的基于 SNN 的 RL 方法仍然较少。其中大多数方法要么缺乏泛化能力,要么在训练中采用人工神经网络(ANN)来估计价值函数。前者需要针对每个场景调整大量超参数,而后者限制了不同类型 RL 算法的应用,并且忽视了训练过程中的巨大能耗。为了开发一种鲁棒的基于脉冲的 RL 方法,我们从昆虫中发现的非脉冲中间神经元获得启发,提出了深度脉冲 Q 网络(DSQN),将非脉冲神经元的膜电压用作 Q 值的表示,从而能够通过端到端 RL 直接从高维感官输入中学习鲁棒策略。在 17 款 Atari 游戏上进行的实验表明,DSQN 是有效的,甚至在大多数游戏中优于基于 ANN 的深度 Q 网络(DQN)。此外,实验还表明 DSQN 具有更优的学习稳定性以及对抗攻击鲁棒性。

英文摘要:

With the help of special neuromorphic hardware, spiking neural networks (SNNs) are expected to realize artificial intelligence (AI) with less energy consumption. It provides a promising energy-efficient way for realistic control tasks by combining SNNs with deep reinforcement learning (RL). There are only a few existing SNN-based RL methods at present. Most of them either lack generalization ability or employ Artificial Neural Networks (ANNs) to estimate value function in training. The former needs to tune numerous hyper-parameters for each scenario, and the latter limits the application of different types of RL algorithm and ignores the large energy consumption in training. To develop a robust spike-based RL method, we draw inspiration from non-spiking interneurons found in insects and propose the deep spiking Q-network (DSQN), using the membrane voltage of non-spiking neurons as the representation of Q-value, which can directly learn robust policies from high-dimensional sensory inputs using end-to-end RL. Experiments conducted on 17 Atari games demonstrate the DSQN is effective and even outperforms the ANN-based deep Q-network (DQN) in most games. Moreover, the experiments show superior learning stability and robustness to adversarial attacks of DSQN.

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

↑