arXiv reaDer
フルスパイク変分オートエンコーダ
Fully Spiking Variational Autoencoder
スパイキングニューラルネットワーク(SNN)は、バイナリおよびイベント駆動型の性質があるため、超高速かつ超低エネルギー消費のニューロモルフィックデバイスで実行できます。したがって、SNNには、高品質の画像を作成するためにエッジデバイスで実行される生成モデルなど、さまざまなアプリケーションが期待されます。この研究では、画像生成を可能にするために、SNNを使用して変分オートエンコーダー(VAE)を構築します。 VAEは、生成モデル間の安定性で知られています。最近、その品質が向上しました。バニラVAEでは、潜在空間は正規分布として表され、サンプリングには浮動小数点計算が必要です。ただし、すべての機能はバイナリ時系列データである必要があるため、SNNではこれは不可能です。したがって、自己回帰SNNモデルを使用して潜在空間を構築し、その出力からランダムにサンプルを選択して潜在変数をサンプリングしました。これにより、潜在変数がベルヌーイ過程に従うことができ、変分学習が可能になります。したがって、すべてのモジュールがSNNで構築される完全スパイク変分オートエンコーダを構築します。私たちの知る限り、SNNレイヤーのみでVAEを構築したのは私たちが初めてです。いくつかのデータセットを実験し、従来のANNと同等またはそれ以上の品質の画像を生成できることを確認しました。コードはまもなく利用可能になります。
Spiking neural networks (SNNs) can be run on neuromorphic devices with ultra-high speed and ultra-low energy consumption because of their binary and event-driven nature. Therefore, SNNs are expected to have various applications, including as generative models being running on edge devices to create high-quality images. In this study, we build a variational autoencoder (VAE) with SNN to enable image generation. VAE is known for its stability among generative models; recently, its quality advanced. In vanilla VAE, the latent space is represented as a normal distribution, and floating-point calculations are required in sampling. However, this is not possible in SNNs because all features must be binary time series data. Therefore, we constructed the latent space with an autoregressive SNN model, and randomly selected samples from its output to sample the latent variables. This allows the latent variables to follow the Bernoulli process and allows variational learning. Thus, we build the Fully Spiking Variational Autoencoder where all modules are constructed with SNN. To the best of our knowledge, we are the first to build a VAE only with SNN layers. We experimented with several datasets, and confirmed that it can generate images with the same or better quality compared to conventional ANNs. The code will be available soon.
updated: Sun Sep 26 2021 06:10:14 GMT+0000 (UTC)
published: Sun Sep 26 2021 06:10:14 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アソシエイト