通过直接训练的深度脉冲Q网络实现人类水平的控制

Human-Level Control Through Directly Trained Deep Spiking Q-Networks

IEEE Transactions on Cybernetics · 2022
被引 51
ABS 3

中文导读

提出一种直接训练的深度脉冲Q网络(DSQN),基于LIF神经元和DQN架构,在17款Atari游戏上性能、稳定性、泛化性和能效均优于现有转换方法,首次实现直接训练SNN在多个游戏上的最优表现。

Abstract

As the third-generation neural networks, spiking neural networks (SNNs) have great potential on neuromorphic hardware because of their high energy efficiency. However, deep spiking reinforcement learning (DSRL), that is, the reinforcement learning (RL) based on SNNs, is still in its preliminary stage due to the binary output and the nondifferentiable property of the spiking function. To address these issues, we propose a deep spiking Q -network (DSQN) in this article. Specifically, we propose a directly trained DSRL architecture based on the leaky integrate-and-fire (LIF) neurons and deep Q -network (DQN). Then, we adapt a direct spiking learning algorithm for the DSQN. We further demonstrate the advantages of using LIF neurons in DSQN theoretically. Comprehensive experiments have been conducted on 17 top-performing Atari games to compare our method with the state-of-the-art conversion method. The experimental results demonstrate the superiority of our method in terms of performance, stability, generalization and energy efficiency. To the best of our knowledge, our work is the first one to achieve state-of-the-art performance on multiple Atari games with the directly trained SNN.

脉冲神经网络深度强化学习类脑计算人工智能