arXiv reaDer
段階的剪定による効果的なモデル圧縮
Effective Model Compression via Stage-wise Pruning
自動機械学習(Auto-ML)の剪定方法は、剪定戦略を自動的に検索して、深い畳み込みニューラルネットワーク(深いCNN)の計算の複雑さを軽減することを目的としています。ただし、以前のいくつかの作業では、多くのAuto-MLプルーニング方法の結果が、均一プルーニング方法の結果を超えることさえできないことがわかりました。この論文では、スーパーネットの不完全で不公平なトレーニングによって引き起こされるAuto-MLプルーニングの非効率性が示されています。ディープスーパーネットは、候補が多すぎるため、トレーニングが不十分になります。不完全なトレーニングを克服するために、段階的剪定(SWP)法が提案されます。これは、深いスーパーネットをいくつかの段階的スーパーネットに分割して候補数を減らし、インプレース蒸留を利用して段階的トレーニングを監視します。さらに、各チャネルのサンプリング確率が等しくないため、ワイドスーパーネットは不公平なトレーニングに見舞われます。したがって、フルネットとタイニーネットは各トレーニング反復でサンプリングされ、各チャネルがオーバートレーニングされる可能性があります。注目すべきことに、SWPでトレーニングされたサブネットのプロキシパフォーマンスは、以前のほとんどのAuto-MLプルーニング作業のパフォーマンスよりも実際のパフォーマンスに近くなっています。実験によると、SWPはモバイル設定でCIFAR-10とImageNetの両方で最先端を実現しています。
Automated Machine Learning(Auto-ML) pruning methods aim at searching a pruning strategy automatically to reduce the computational complexity of deep Convolutional Neural Networks(deep CNNs). However, some previous work found that the results of many Auto-ML pruning methods cannot even surpass the results of the uniformly pruning method. In this paper, the ineffectiveness of Auto-ML pruning which is caused by unfull and unfair training of the supernet is shown. A deep supernet suffers from unfull training because it contains too many candidates. To overcome the unfull training, a stage-wise pruning(SWP) method is proposed, which splits a deep supernet into several stage-wise supernets to reduce the candidate number and utilize inplace distillation to supervise the stage training. Besides, A wide supernet is hit by unfair training since the sampling probability of each channel is unequal. Therefore, the fullnet and the tinynet are sampled in each training iteration to ensure each channel can be overtrained. Remarkably, the proxy performance of the subnets trained with SWP is closer to the actual performance than that of most of the previous Auto-ML pruning work. Experiments show that SWP achieves the state-of-the-art on both CIFAR-10 and ImageNet under the mobile setting.
updated: Wed Sep 22 2021 09:30:27 GMT+0000 (UTC)
published: Tue Nov 10 2020 05:19:05 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アソシエイト