arXiv reaDer
ヘッブ主成分分析による畳み込みニューラルネットワークのトレーニング
Training Convolutional Neural Networks With Hebbian Principal Component Analysis
最近の研究では、深い畳み込みニューラルネットワークをトレーニングするときに、生物学的にもっともらしいヘッブ学習をバックプロパゲーション学習(backprop)と統合できることが示されています。特に、ヘッブ学習はニューラルネットワークの下位層または上位層のトレーニングに使用できることが示されています。たとえば、ヘッブ学習は、事前にトレーニングされたディープニューラルネットワークの上位層を再トレーニングするのに効果的であり、SGDと同等の精度を達成すると同時に、必要なトレーニングエポックを少なくして、転移学習の潜在的なアプリケーションを示唆します。このホワイトペーパーでは、これらの結果に基づいて、前の作業で使用したヘッブの勝者がすべてを取る(HWTA)戦略の代わりに、非線形ヘッブ主成分分析(HPCA)学習ルールを使用して、これらの設定でのヘッブ学習をさらに改善します。このアプローチは、コンピュータービジョンのコンテキストでテストします。特に、HPCAルールは、CIFAR-10画像データセットから関連する特徴を抽出するために畳み込みニューラルネットワークをトレーニングするために使用されます。私たちが調査するHPCAバリアントは、以前の結果をさらに改善し、生物学的にもっともらしい学習アルゴリズムへのさらなる関心を動機付けます。
Recent work has shown that biologically plausible Hebbian learning can be integrated with backpropagation learning (backprop), when training deep convolutional neural networks. In particular, it has been shown that Hebbian learning can be used for training the lower or the higher layers of a neural network. For instance, Hebbian learning is effective for re-training the higher layers of a pre-trained deep neural network, achieving comparable accuracy w.r.t. SGD, while requiring fewer training epochs, suggesting potential applications for transfer learning. In this paper we build on these results and we further improve Hebbian learning in these settings, by using a nonlinear Hebbian Principal Component Analysis (HPCA) learning rule, in place of the Hebbian Winner Takes All (HWTA) strategy used in previous work. We test this approach in the context of computer vision. In particular, the HPCA rule is used to train Convolutional Neural Networks in order to extract relevant features from the CIFAR-10 image dataset. The HPCA variant that we explore further improves the previous results, motivating further interest towards biologically plausible learning algorithms.
updated: Tue Dec 22 2020 18:17:46 GMT+0000 (UTC)
published: Tue Dec 22 2020 18:17:46 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アソシエイト