arXiv reaDer
畳み込みニューラルネットワークの剪定に対する事前トレーニングの影響の実験的研究
An Experimental Study of the Impact of Pre-training on the Pruning of a Convolutional Neural Network
近年、ディープニューラルネットワークはさまざまなアプリケーションドメインで幅広い成功を収めています。ただし、重要な計算リソースとメモリリソースが必要であるため、特にモバイルデバイスやリアルタイムアプリケーションでの展開が大幅に妨げられます。ニューラルネットワークには通常、ネットワークの重みに対応する多数のパラメーターが含まれます。トレーニングプロセスの助けを借りて取得されたこのようなパラメータは、ネットワークのパフォーマンスを決定します。ただし、それらは非常に冗長でもあります。剪定方法は、特に、無関係な重みを識別して削除することにより、パラメーターセットのサイズを縮小しようとします。この論文では、剪定効率に対するトレーニング戦略の影響を調べます。 2つのトレーニングモダリティが検討および比較されます:(1)微調整および(2)ゼロから。 4つのデータセット(CIFAR10、CIFAR100、SVHN、Caltech101)と2つの異なるCNN(VGG16とMobileNet)で得られた実験結果は、大規模なコーパス(ImageNetなど)で事前トレーニングされた後、特定のデータセットは、最初からトレーニングされた同じネットワークよりもはるかに効率的に(パラメーター削減の最大80%)プルーニングできます。
In recent years, deep neural networks have known a wide success in various application domains. However, they require important computational and memory resources, which severely hinders their deployment, notably on mobile devices or for real-time applications. Neural networks usually involve a large number of parameters, which correspond to the weights of the network. Such parameters, obtained with the help of a training process, are determinant for the performance of the network. However, they are also highly redundant. The pruning methods notably attempt to reduce the size of the parameter set, by identifying and removing the irrelevant weights. In this paper, we examine the impact of the training strategy on the pruning efficiency. Two training modalities are considered and compared: (1) fine-tuned and (2) from scratch. The experimental results obtained on four datasets (CIFAR10, CIFAR100, SVHN and Caltech101) and for two different CNNs (VGG16 and MobileNet) demonstrate that a network that has been pre-trained on a large corpus (e.g. ImageNet) and then fine-tuned on a particular dataset can be pruned much more efficiently (up to 80% of parameter reduction) than the same network trained from scratch.
updated: Wed Dec 15 2021 16:02:15 GMT+0000 (UTC)
published: Wed Dec 15 2021 16:02:15 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アソシエイト