arXiv reaDer
行列積演算子によるディープニューラルネットワークの圧縮
Compressing deep neural networks by matrix product operators
ディープニューラルネットワークは、信号のマルチレイヤーマッピングをパラメーター化したもので、線形変換と非線形変換が交互に配置されています。線形変換は、完全に接続された層と畳み込み層で一般的に使用されており、トレーニングおよび保存されるほとんどの変分パラメーターを含みます。ディープニューラルネットワークを圧縮して、変動パラメーターの数を減らし、予測力を減らすことは、これらのパラメーターを効率的にトレーニングし、過剰適合のリスクを下げる最適化されたスキームの確立に向けた重要ですが、挑戦的な問題です。ここでは、この問題が、物理学で最初に提案されたテンソルネットワークである行列積演算子(MPO)で線形変換を表現することによって効果的に解決できることを示します。これは、1次元量子状態の短距離エンタングルメントを特徴付けるためです。私たちは、FC2、LeNet-5、VGG、ResNet、およびDenseNetを含む5つの典型的なニューラルネットワークで、MNISTとCIFAR-10の2つの広く使用されているデータセットでこのアプローチをテストし、このMPO表現が実際に忠実に設定されていることを発見しましたまた、入力信号と出力信号の間の効率的なマッピング。これにより、パラメーターの数が大幅に削減され、予測精度を維持または向上させることができます。私たちの方法は、ディープラーニングでの表現を大幅に簡素化し、よりシンプルで安価であるがより効率的な現代のニューラルネットワークのフレームワークを確立するための可能なルートを開きます。
A deep neural network is a parametrization of a multilayer mapping of signals in terms of many alternatively arranged linear and nonlinear transformations. The linear transformations, which are generally used in the fully connected as well as convolutional layers, contain most of the variational parameters that are trained and stored. Compressing a deep neural network to reduce its number of variational parameters but not its prediction power is an important but challenging problem toward the establishment of an optimized scheme in training efficiently these parameters and in lowering the risk of overfitting. Here we show that this problem can be effectively solved by representing linear transformations with matrix product operators (MPOs), which is a tensor network originally proposed in physics to characterize the short-range entanglement in one-dimensional quantum states. We have tested this approach in five typical neural networks, including FC2, LeNet-5, VGG, ResNet, and DenseNet on two widely used data sets, namely, MNIST and CIFAR-10, and found that this MPO representation indeed sets up a faithful and efficient mapping between input and output signals, which can keep or even improve the prediction accuracy with a dramatically reduced number of parameters. Our method greatly simplifies the representations in deep learning, and opens a possible route toward establishing a framework of modern neural networks which might be simpler and cheaper, but more efficient.
updated: Wed Jun 10 2020 03:26:01 GMT+0000 (UTC)
published: Thu Apr 11 2019 17:59:00 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アソシエイト