Reinforcement Learning with Feedback-modulated TD-STDP

Abstract

Spiking neuron networks have been used successfully to solve simple reinforcement learning tasks with continuous action set applying learning rules based on spike-timing-dependent plasticity (STDP). However, most of these models cannot be applied to reinforcement learning tasks with discrete action set since they assume that the selected action is a deterministic function of firing rate of neurons, which is continuous. In this paper, we propose a new STDP-based learning rule for spiking neuron networks which contains feedback modulation. We show that the STDP-based learning rule can be used to solve reinforcement learning tasks with discrete action set at a speed similar to standard reinforcement learning algorithms when applied to the CartPole and LunarLander tasks. Moreover, we demonstrate that the agent is unable to solve these tasks if feedback modulation is omitted from the learning rule. We conclude that feedback modulation allows better credit assignment when only the units contributing to the executed action and TD error participate in learning.

arXiv comment17 pages, 4 figures

References in corpus (6)

• Adam: A Method for Stochastic Optimization
• Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computation
• STDP-based spiking deep convolutional neural networks for object recognition
• Demonstrating Advantages of Neuromorphic Computation: A Pilot Study
• Understanding Straight-Through Estimator in Training Activation Quantized Neural Nets
• Reinforcement learning with a network of spiking agents