arXiv reaDer
低遅延スパイキングニューラルネットワークの最適化された潜在的な初期化
Optimized Potential Initialization for Low-latency Spiking Neural Networks
スパイキングニューラルネットワーク(SNN)は、低消費電力、生物学的妥当性、および敵対的な堅牢性という独特の特性により、非常に重要視されてきました。ディープSNNをトレーニングする最も効果的な方法は、ANNからSNNへの変換を使用することです。これにより、ディープネットワーク構造と大規模データセットで最高のパフォーマンスが得られます。ただし、精度と遅延の間にはトレードオフがあります。オリジナルのANNとしての高精度を実現するためには、スパイキングニューロンの発火率とアナログニューロンの活性化値を一致させるために長いシミュレーション時間が必要であり、SNNの実用化を妨げています。この論文では、非常に低いレイテンシ(32タイムステップ未満)で高性能の変換されたSNNを実現することを目指しています。 ANNからSNNへの変換を理論的に分析することから始め、しきい値のスケーリングが重みの正規化と同様の役割を果たすことを示します。モデル容量を犠牲にしてANNからSNNへの変換を容易にする制約を導入する代わりに、各層の変換損失を減らすために初期膜電位を最適化することにより、より直接的な方法を適用しました。さらに、膜電位の最適な初期化により、予想されるエラーのないANNからSNNへの変換を実装できることを示します。 CIFAR-10、CIFAR-100、およびImageNetデータセットでアルゴリズムを評価し、より少ないタイムステップで最先端の精度を実現します。たとえば、16のタイムステップでCIFAR-10のトップ1の精度は93.38%に達します。さらに、私たちの方法は他のANN-SNN変換方法に適用でき、タイムステップが小さい場合にパフォーマンスを大幅に向上させます。
Spiking Neural Networks (SNNs) have been attached great importance due to the distinctive properties of low power consumption, biological plausibility, and adversarial robustness. The most effective way to train deep SNNs is through ANN-to-SNN conversion, which have yielded the best performance in deep network structure and large-scale datasets. However, there is a trade-off between accuracy and latency. In order to achieve high precision as original ANNs, a long simulation time is needed to match the firing rate of a spiking neuron with the activation value of an analog neuron, which impedes the practical application of SNN. In this paper, we aim to achieve high-performance converted SNNs with extremely low latency (fewer than 32 time-steps). We start by theoretically analyzing ANN-to-SNN conversion and show that scaling the thresholds does play a similar role as weight normalization. Instead of introducing constraints that facilitate ANN-to-SNN conversion at the cost of model capacity, we applied a more direct way by optimizing the initial membrane potential to reduce the conversion loss in each layer. Besides, we demonstrate that optimal initialization of membrane potentials can implement expected error-free ANN-to-SNN conversion. We evaluate our algorithm on the CIFAR-10, CIFAR-100 and ImageNet datasets and achieve state-of-the-art accuracy, using fewer time-steps. For example, we reach top-1 accuracy of 93.38% on CIFAR-10 with 16 time-steps. Moreover, our method can be applied to other ANN-SNN conversion methodologies and remarkably promote performance when the time-steps is small.
updated: Thu Feb 03 2022 07:15:43 GMT+0000 (UTC)
published: Thu Feb 03 2022 07:15:43 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アソシエイト