スパイキングニューラルネットワーク(SNN)は、一時的な情報処理機能、低消費電力、および高い生物学的妥当性により、多大な研究関心を集めています。ただし、SNNの効率的で高性能な学習アルゴリズムの定式化は依然として困難です。ほとんどの既存の学習方法は重みのみを学習し、単一のスパイキングニューロンのダイナミクスを決定する膜関連パラメーターの手動調整を必要とします。これらのパラメータは通常、すべてのニューロンで同じになるように選択されます。これにより、ニューロンの多様性が制限され、結果として得られるSNNの表現力が制限されます。本論文では、膜関連パラメータが脳領域間で異なるという観察からインスピレーションを得て、シナプスの重みだけでなく、SNNの膜時定数も学習できるトレーニングアルゴリズムを提案します。学習可能な膜時定数を組み込むと、ネットワークの初期値に対する感度が低下し、学習が高速化されることを示します。さらに、SNNのプーリング方法を再評価し、最大プーリングによって情報が大幅に失われることはなく、計算コストが低く、バイナリ互換性があるという利点があることを確認しました。従来の静的MNIST、Fashion-MNIST、CIFAR-10データセット、およびニューロモルフィックN-MNIST、CIFAR10-DVS、DVS128ジェスチャデータセットの両方で、画像分類タスクの提案された方法を評価します。実験結果は、提案された方法が、より少ない時間ステップを使用して、ほぼすべてのデータセットで最先端の精度を上回っていることを示しています。コードはhttps://github.com/fangwei123456/Parametric-Leaky-Integrate-and-Fire-Spiking-Neuronで入手できます。
Spiking Neural Networks (SNNs) have attracted enormous research interest due to temporal information processing capability, low power consumption, and high biological plausibility. However, the formulation of efficient and high-performance learning algorithms for SNNs is still challenging. Most existing learning methods learn weights only, and require manual tuning of the membrane-related parameters that determine the dynamics of a single spiking neuron. These parameters are typically chosen to be the same for all neurons, which limits the diversity of neurons and thus the expressiveness of the resulting SNNs. In this paper, we take inspiration from the observation that membrane-related parameters are different across brain regions, and propose a training algorithm that is capable of learning not only the synaptic weights but also the membrane time constants of SNNs. We show that incorporating learnable membrane time constants can make the network less sensitive to initial values and can speed up learning. In addition, we reevaluate the pooling methods in SNNs and find that max-pooling will not lead to significant information loss and have the advantage of low computation cost and binary compatibility. We evaluate the proposed method for image classification tasks on both traditional static MNIST, Fashion-MNIST, CIFAR-10 datasets, and neuromorphic N-MNIST, CIFAR10-DVS, DVS128 Gesture datasets. The experiment results show that the proposed method outperforms the state-of-the-art accuracy on nearly all datasets, using fewer time-steps. Our codes are available at https://github.com/fangwei123456/Parametric-Leaky-Integrate-and-Fire-Spiking-Neuron.