arXiv reaDer
SPIDE: フィードバック スパイキング ニューラル ネットワークをトレーニングするための純粋なスパイク ベースの方法
SPIDE: A Purely Spike-based Method for Training Feedback Spiking Neural Networks
イベントベースの計算を使用したスパイキング ニューラル ネットワーク (SNN) は、ニューロモルフィック ハードウェアでのエネルギー効率の高いアプリケーションのための有望な脳にヒントを得たモデルです。ただし、人工ニューラル ネットワークからの変換やサロゲート勾配を使用した直接トレーニングなど、ほとんどの教師あり SNN トレーニング方法では、トレーニング中にニューロンをスパイクするスパイク ベースの操作ではなく、複雑な計算が必要です。この論文では、純粋なスパイクベースの計算による教師あり学習のために、最近提案されたトレーニング方法である平衡状態の暗黙の微分 (IDE) を拡張する、平衡状態のスパイクベースの暗黙の微分 (SPIDE) を研究します。 SNN のエネルギー効率の高いトレーニング用。具体的には、3 値スパイキング ニューロン カップルを導入し、暗黙的な微分がこの設計に基づくスパイクによって解決できることを証明します。したがって、フォワード パスとバックワード パスの両方を含むトレーニング手順全体がイベント駆動型のスパイク計算として作成され、重みがローカルで更新されます。 2段階の平均発砲率。次に、スパイクの近似誤差を減らすためにリセット膜電位を変更することを提案します。これらの主要なコンポーネントを使用すると、柔軟な構造を持つ SNN を少数の時間ステップで訓練し、訓練中にスパース性を発火させることができます。エネルギー コストの理論的推定は、高効率の可能性を示しています。一方、実験では、これらの制約があっても、トレーニング済みのモデルは MNIST、CIFAR-10、CIFAR-100、および CIFAR10-DVS で競争力のある結果を達成できることが示されています。コードは https://github.com/pkuxmq/SPIDE-FSNN で入手できます。
Spiking neural networks (SNNs) with event-based computation are promising brain-inspired models for energy-efficient applications on neuromorphic hardware. However, most supervised SNN training methods, such as conversion from artificial neural networks or direct training with surrogate gradients, require complex computation rather than spike-based operations of spiking neurons during training. In this paper, we study spike-based implicit differentiation on the equilibrium state (SPIDE) that extends the recently proposed training method, implicit differentiation on the equilibrium state (IDE), for supervised learning with purely spike-based computation, which demonstrates the potential for energy-efficient training of SNNs. Specifically, we introduce ternary spiking neuron couples and prove that implicit differentiation can be solved by spikes based on this design, so the whole training procedure, including both forward and backward passes, is made as event-driven spike computation, and weights are updated locally with two-stage average firing rates. Then we propose to modify the reset membrane potential to reduce the approximation error of spikes. With these key components, we can train SNNs with flexible structures in a small number of time steps and with firing sparsity during training, and the theoretical estimation of energy costs demonstrates the potential for high efficiency. Meanwhile, experiments show that even with these constraints, our trained models can still achieve competitive results on MNIST, CIFAR-10, CIFAR-100, and CIFAR10-DVS. Our code is available at https://github.com/pkuxmq/SPIDE-FSNN.
updated: Wed Feb 01 2023 04:22:59 GMT+0000 (UTC)
published: Wed Feb 01 2023 04:22:59 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アソシエイト