arXiv reaDer
SNN2ANN:スパイキングニューラルネットワーク用の高速でメモリ効率の高いトレーニングフレームワーク
SNN2ANN: A Fast and Memory-Efficient Training Framework for Spiking Neural Networks
スパイキングニューラルネットワークは、低電力環境向けの効率的な計算モデルです。スパイクベースのBPアルゴリズムとANNからSNN(ANN2SNN)への変換は、SNNトレーニングの成功した手法です。それにもかかわらず、スパイクベースのBPトレーニングは遅く、大きなメモリコストを必要とします。 ANN2NNはSNNをトレーニングするための低コストの方法を提供しますが、良好なパフォーマンスを得るために十分にトレーニングされたANNを模倣するために多くの推論ステップが必要です。この論文では、SNNを高速かつメモリ効率の高い方法でトレーニングするためのSNN-to-ANN(SNN2ANN)フレームワークを提案します。 SNN2ANNは、2つのコンポーネントで構成されています。a)ANNとSNN間の重み共有アーキテクチャ、およびb)スパイクマッピングユニット。まず、アーキテクチャはANNブランチで重み共有パラメータをトレーニングするため、SNNのトレーニングが高速になりメモリコストが低くなります。第二に、スパイクマッピングユニットは、ANNのアクティベーション値がスパイク機能であることを保証します。その結果、SNNの分類エラーは、ANNブランチをトレーニングすることによって最適化できます。さらに、ノイズの多いスパイクの問題に対処するために、適応しきい値調整(ATA)アルゴリズムを設計します。実験結果は、SNN2ANNベースのモデルがベンチマークデータセット(CIFAR10、CIFAR100、およびTiny-ImageNet)で良好に機能することを示しています。さらに、SNN2ANNは、スパイクベースのBPモデルの0.625倍のタイムステップ、0.377倍のトレーニング時間、0.27倍のGPUメモリコスト、および0.33倍のスパイクアクティビティで同等の精度を達成できます。
Spiking neural networks are efficient computation models for low-power environments. Spike-based BP algorithms and ANN-to-SNN (ANN2SNN) conversions are successful techniques for SNN training. Nevertheless, the spike-base BP training is slow and requires large memory costs. Though ANN2NN provides a low-cost way to train SNNs, it requires many inference steps to mimic the well-trained ANN for good performance. In this paper, we propose a SNN-to-ANN (SNN2ANN) framework to train the SNN in a fast and memory-efficient way. The SNN2ANN consists of 2 components: a) a weight sharing architecture between ANN and SNN and b) spiking mapping units. Firstly, the architecture trains the weight-sharing parameters on the ANN branch, resulting in fast training and low memory costs for SNN. Secondly, the spiking mapping units ensure that the activation values of the ANN are the spiking features. As a result, the classification error of the SNN can be optimized by training the ANN branch. Besides, we design an adaptive threshold adjustment (ATA) algorithm to address the noisy spike problem. Experiment results show that our SNN2ANN-based models perform well on the benchmark datasets (CIFAR10, CIFAR100, and Tiny-ImageNet). Moreover, the SNN2ANN can achieve comparable accuracy under 0.625x time steps, 0.377x training time, 0.27x GPU memory costs, and 0.33x spike activities of the Spike-based BP model.
updated: Sun Jun 19 2022 16:52:56 GMT+0000 (UTC)
published: Sun Jun 19 2022 16:52:56 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アソシエイト