arXiv reaDer
LB-CNN:ChainerとCupyを使用したライトバイナリ畳み込みニューラルネットワークの高速トレーニングのためのオープンソースフレームワーク
LB-CNN: An Open Source Framework for Fast Training of Light Binary Convolutional Neural Networks using Chainer and Cupy
ライトバイナリ畳み込みニューラルネットワーク(LB-CNN)は、多くの産業用アプリケーションで必要とされる低エネルギーコンピューティングプラットフォームに実装する場合に特に役立ちます。ここでは、コンパクトなLB-CNNを最適化するためのフレームワークを紹介し、その有効性を評価します。フレームワークは無料で利用でき、無料アクセスのクラウドプラットフォームで実行できるため、大きな投資は必要ありません。最適化されたモデルは、標準化された.h5形式で保存され、特定のテクノロジーにさらに展開するための専用ツールへの入力として使用できるため、さまざまなインテリジェントイメージセンサーの迅速な開発が可能になります。モデルの最適化、特にバイナリ畳み込みカーネルの選択を加速する主な要素は、出力層をエクストリームラーニングマシンとしてトレーニングするための大幅な高速化を提供するChainer / Cupy機械学習ライブラリです。 Keras / Tensorflowを使用した出力レイヤーの追加トレーニングが含まれているため、精度が向上します。 MNIST、GTSRB、ORL、VGGなど、広く使用されているデータセットの結果は、精度と複雑さの間で非常に優れた妥協点を示しています。特に、顔認識の問題については、慎重に最適化されたLB-CNNモデルが最大100%の精度を提供します。このようなTinyMLソリューションは、低エネルギー消費で画像認識を必要とする産業用アプリケーションに最適です。
Light binary convolutional neural networks (LB-CNN) are particularly useful when implemented in low-energy computing platforms as required in many industrial applications. Herein, a framework for optimizing compact LB-CNN is introduced and its effectiveness is evaluated. The framework is freely available and may run on free-access cloud platforms, thus requiring no major investments. The optimized model is saved in the standardized .h5 format and can be used as input to specialized tools for further deployment into specific technologies, thus enabling the rapid development of various intelligent image sensors. The main ingredient in accelerating the optimization of our model, particularly the selection of binary convolution kernels, is the Chainer/Cupy machine learning library offering significant speed-ups for training the output layer as an extreme-learning machine. Additional training of the output layer using Keras/Tensorflow is included, as it allows an increase in accuracy. Results for widely used datasets including MNIST, GTSRB, ORL, VGG show very good compromise between accuracy and complexity. Particularly, for face recognition problems a carefully optimized LB-CNN model provides up to 100% accuracies. Such TinyML solutions are well suited for industrial applications requiring image recognition with low energy consumption.
updated: Fri Jun 25 2021 09:40:04 GMT+0000 (UTC)
published: Fri Jun 25 2021 09:40:04 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アソシエイト