arXiv reaDer
自己類似性マトリックス ベースの CNN フィルター プルーニング
Self Similarity Matrix based CNN Filter Pruning
近年、ディープラーニング ソリューションのほとんどは、モバイル デバイスへの展開を対象としています。これにより、軽量モデルの開発の必要性がますます差し迫っています。もう 1 つの解決策は、通常の深層学習モデルを最適化して刈り込むことです。このホワイト ペーパーでは、2D CNN フィルターから計算された自己相似行列 (SSM) を使用して、CNN モデルの枝刈りの問題に取り組みます。同様の活性化マップを出力に寄与する冗長フィルターをランク付けして除去するための 2 つの新しいアルゴリズムを提案します。この方法の重要な特徴の 1 つは、モデルのトレーニング後に微調整する必要がないことです。トレーニングとプルーニングの両方のプロセスが同時に完了します。最も人気のある 2 つの CNN モデル (ResNet と VGG) でメソッドのベンチマークを行い、CIFAR-10 データセットでそのパフォーマンスを記録します。
In recent years, most of the deep learning solutions are targeted to be deployed in mobile devices. This makes the need for development of lightweight models all the more imminent. Another solution is to optimize and prune regular deep learning models. In this paper, we tackle the problem of CNN model pruning with the help of Self-Similarity Matrix (SSM) computed from the 2D CNN filters. We propose two novel algorithms to rank and prune redundant filters which contribute similar activation maps to the output. One of the key features of our method is that there is no need of finetuning after training the model. Both the training and pruning process is completed simultaneously. We benchmark our method on two of the most popular CNN models - ResNet and VGG and record their performance on the CIFAR-10 dataset.
updated: Thu Nov 03 2022 13:47:44 GMT+0000 (UTC)
published: Thu Nov 03 2022 13:47:44 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アソシエイト