SNN系列｜学习算法篇(3)ReSuMe

ReSuMe

原文传送

适用于单层网络或液体状态机，采用精确时间编码，结合了STDP与anti-STDP机制，对多种神经元模型和突触类型均适用，因为权重的调整仅取决于脉冲输出与目标输出序列有关。核心的更新公式如下：
$$\frac{d}{dt}{\omega }_{oi}(t)=\left[S_d(t)-S_o(t)\right]\left[a_d+{\int }_0^{\infty }{a}_{di}(s)S_i(t-s)ds\right]$$

Anti-Hebbian

Hebb定律提到"fire together, wire together"，Hebb本人也提到突触前后神经元的协同活动会提高突触前神经元的传递效率，这对兴奋性突触时没有问题的，但是对抑制性神经元来说，相关性活动却会造成突触传递效率的降低。与Hebb学习法则一样，anti-Hebb只是多了一个负号
$$\frac{d\omega }{dt}=-\eta xy$$
anti-Hebb通常与Hebb一起使用，比如可以用于侧向神经元的相互抑制。Hebbian学习通常与LTP相关，anti-Hebbian通常与LTD相关，STDP一般也认为与anti-hebb有关。

A phenomenological model for STDP

该模型认为，除了突触延迟以外，所有的突触权重变化都可以由突触前后神经元脉冲来描述。突触前、突触后、目标脉冲序列分别表示为$S_i(t),S_o(t),S_d(t)$，每个序列都是单个脉冲的叠加$S(t)={\sum }_f\delta (t-t_f)$。

突触权重的改变包括三部分：当不考虑$a$和Non-Hebbian时，同经典STDP。

• 无脉冲时的自衰减$a$
• 突触前脉冲带来的影响
  • Non-Hebbian：无突触后脉冲时的自影响
  • 突触前后脉冲时序的影响
• 突触后脉冲带来的影响
  • Non-Hebbian：无突触前脉冲时的自影响
  • 突触前后脉冲时序的影响

突触权重的更新如下：
$$\frac{d\omega (t)}{dt}=a+S_i(t)\left[a_i+{\int }_0^{\infty }{a}_{id}(s)S_d(t-s)ds\right]+S_d(t)\left[a_d+{\int }_0^{\infty }{a}_{di}(s)S_i(t-s)ds\right]$$

注意积分时从0开始的，也就是说当突触前脉冲到来的时候，只看在此之前的突触后脉冲，按照经典STDP，此时应该是抑制作用；反之为增强作用。对于突触前后脉冲时序$s=t_i^f-t_d^f$的影响，$a_{id},a_{di}$定义窗口函数如下：
$$W(s)=\{\begin{array}{cl}{A}_{+}\exp \left[s/{\tau }_{+}\right]& \text{for }s<0\\ A_{-}\exp \left[-s/{\tau }_{-}\right]& \text{for }s>0\end{array}$$
即$a_{id}(s)$对应$s>0$，$a_{di}(s)$对应$s<0$。

ReSuMe

该算法在上述基础上，将STDP与Anti-STDP相结合，即突触权重的更新满足
$$\frac{d{\omega }_{oi}(t)}{dt}=X_{di}(t)+X_{oi}(t)$$
其中，$X_{di}(t),X_{oi}(t)$分别表示突触前脉冲序列与目标序列的STDP过程以及突触前脉冲序列与实际序列的anti-STDP过程，根据上一节的描述，二者可形式化给出
$$\begin{gathered} X_{di}(t)=a+S_i(t)\left[a_i+{\int }_0^{\infty }{a}_{id}(s)S_d(t-s)ds\right]+S_d(t)\left[a_d+{\int }_0^{\infty }{a}_{di}(s)S_i(t-s)ds\right] \\ {X}_{oi}(t)=b+S_i(t)\left[b_i+{\int }_0^{\infty }{b}_{io}(s)S_o(t-s)ds\right]+S_o(t)\left[b_o+{\int }_0^{\infty }{b}_{oi}(s)S_i(t-s)ds\right] \end{gathered}$$
当输出脉冲序列与目标脉冲序列完全一致时，应满足如下条件
$$\begin{gathered} X_{di}(t)+X_{oi}(t)=0\iff S_o(t)=S_d(t) \\ ⟹\{\begin{array}{l}-b=a,-b_i=a_i,-b_o=a_d\\ A_{oi}=A_{di},A_{io}=A_{id}\\ {\tau }_{oi}={\tau }_{di},{\tau }_{io}={\tau }_{id}\end{array} \end{gathered}$$
使用上述条件化简公式
$$\begin{array}{rl}\frac{d}{dt}{\omega }_{oi}(t)& =S_i(t){\int }_0^{\infty }{a}_{id}(s)\left[S_d(t-s)-S_o(t-s)\right]ds\\ & +\left[S_d(t)-S_o(t)\right]\left[a_d+{\int }_0^{\infty }{a}_{di}(s)S_i(t-s)ds\right]\end{array}$$
实验发现，$a_{id}(s)=0$能获得更好的收敛性和学习效果，因此，ReSuMe的学习规则如下：
$$\frac{d}{dt}{\omega }_{oi}(t)=\left[S_d(t)-S_o(t)\right]\left[a_d+{\int }_0^{\infty }{a}_{di}(s)S_i(t-s)ds\right]$$

代码

略

参考

[1] Spike-timing dependent synaptic plasticity: a phenomenological framework

[2] Encyclopedia of Computianal Neuroscience

[3] Supervised Learning in Spiking Neural Networks with ReSuMe: Sequence Learning, Classification, and Spike Shifting