arXiv reaDer
KLIF: サロゲート勾配勾配と膜電位を調整するための最適化されたスパイキング ニューロン ユニット
KLIF: An optimized spiking neuron unit for tuning surrogate gradient slope and membrane potential
スパイキング ニューラル ネットワーク (SNN) は、時間情報を処理する能力、低消費電力、およびより高い生物学的妥当性により、多くの注目を集めています。ただし、SNN 用の効率的で高性能な学習アルゴリズムを開発することは依然として困難です。人工ニューラル ネットワーク (ANN) から SNN への変換などの方法では、わずかなパフォーマンスの低下で ANN を SNN に変換できますが、レート コーディングを概算するには長いシミュレーションが必要です。サロゲート勾配近似などのスパイクベースのバックプロパゲーション (BP) による SNN の直接トレーニングは、より柔軟です。しかし現在、SNN のパフォーマンスは ANN に比べて競争力がありません。この論文では、SNN の学習能力を向上させるために、新しい k ベースの漏洩した Integrate-and-Fire (KLIF) ニューロン モデルを提案します。一般的な漏洩統合発火 (LIF) モデルと比較して、KLIF は学習可能なスケーリング係数を追加して、トレーニング中に代理勾配曲線の勾配と幅を動的に更新し、膜電位をスパイク発火と選択的に提供する ReLU 活性化関数を組み込みます。リセット中。提案されたスパイキング ユニットは、静的 MNIST、Fashion-MNIST、CIFAR-10 データセット、およびニューロモルフィック N-MNIST、CIFAR10-DVS、および DVS128-Gesture データセットの両方で評価されます。実験では、追加の計算コストを導入することなく、KLIF が LIF よりもはるかに優れたパフォーマンスを発揮し、これらのデータセットで最先端のパフォーマンスをわずかな時間ステップで実現することが示されています。また、KLIF は LIF よりも生物学的にもっともらしいと考えられています。 KLIF の優れたパフォーマンスにより、さまざまなタスクで SNN の LIF の役割を完全に置き換えることができます。
Spiking neural networks (SNNs) have attracted much attention due to their ability to process temporal information, low power consumption, and higher biological plausibility. However, it is still challenging to develop efficient and high-performing learning algorithms for SNNs. Methods like artificial neural network (ANN)-to-SNN conversion can transform ANNs to SNNs with slight performance loss, but it needs a long simulation to approximate the rate coding. Directly training SNN by spike-based backpropagation (BP) such as surrogate gradient approximation is more flexible. Yet now, the performance of SNNs is not competitive compared with ANNs. In this paper, we propose a novel k-based leaky Integrate-and-Fire (KLIF) neuron model to improve the learning ability of SNNs. Compared with the popular leaky integrate-and-fire (LIF) model, KLIF adds a learnable scaling factor to dynamically update the slope and width of the surrogate gradient curve during training and incorporates a ReLU activation function that selectively delivers membrane potential to spike firing and resetting. The proposed spiking unit is evaluated on both static MNIST, Fashion-MNIST, CIFAR-10 datasets, as well as neuromorphic N-MNIST, CIFAR10-DVS, and DVS128-Gesture datasets. Experiments indicate that KLIF performs much better than LIF without introducing additional computational cost and achieves state-of-the-art performance on these datasets with few time steps. Also, KLIF is believed to be more biological plausible than LIF. The good performance of KLIF can make it completely replace the role of LIF in SNN for various tasks.
updated: Sat Feb 18 2023 05:18:18 GMT+0000 (UTC)
published: Sat Feb 18 2023 05:18:18 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト