arXiv reaDer
ClickTrain:きめ細かいアーキテクチャを介した効率的で正確なエンドツーエンドのディープラーニングトレーニング-剪定を維持する
ClickTrain: Efficient and Accurate End-to-End Deep Learning Training via Fine-Grained Architecture-Preserving Pruning
畳み込みニューラルネットワーク(CNN)は、予測精度と分析品質に対する需要の高まりにより、ますます深く、広く、非線形になっています。ただし、広くて深いCNNは、大量のコンピューティングリソースと処理時間を必要とします。これまでの多くの研究では、推論のパフォーマンスを向上させるためにモデルの剪定が研究されてきましたが、トレーニングコストを効果的に削減するための研究はほとんど行われていません。このホワイトペーパーでは、ClickTrainを提案します。これは、CNNの効率的で正確なエンドツーエンドのトレーニングおよびプルーニングフレームワークです。 ClickTrainは、既存のトレーニング中の剪定作業とは異なり、きめ細かいアーキテクチャを維持する剪定により、より高いモデル精度と圧縮率を提供します。 ClickTrainは、提案された新しい正確な重みの重要度の推定、動的なパターンの生成と選択、およびコンパイラ支援の計算の最適化でパターンベースのプルーニングを活用することにより、ベースラインと比較して、余分な時間のオーバーヘッドなしに直接展開するための非常に正確で高速なプルーニングされたCNNモデルを生成しますトレーニング。 ClickTrainはまた、同等の精度と圧縮率で、トレーニング後の剪定方法のエンドツーエンドの時間コストを最大2.3倍削減します。さらに、トレーニング中の最先端の剪定アプローチと比較して、ClickTrainは、同様の限られたトレーニング時間の下で、テストされたCNNモデルとデータセットの精度と圧縮率の両方を大幅に改善します。
Convolutional neural networks (CNNs) are becoming increasingly deeper, wider, and non-linear because of the growing demand on prediction accuracy and analysis quality. The wide and deep CNNs, however, require a large amount of computing resources and processing time. Many previous works have studied model pruning to improve inference performance, but little work has been done for effectively reducing training cost. In this paper, we propose ClickTrain: an efficient and accurate end-to-end training and pruning framework for CNNs. Different from the existing pruning-during-training work, ClickTrain provides higher model accuracy and compression ratio via fine-grained architecture-preserving pruning. By leveraging pattern-based pruning with our proposed novel accurate weight importance estimation, dynamic pattern generation and selection, and compiler-assisted computation optimizations, ClickTrain generates highly accurate and fast pruned CNN models for direct deployment without any extra time overhead, compared with the baseline training. ClickTrain also reduces the end-to-end time cost of the pruning-after-training method by up to 2.3X with comparable accuracy and compression ratio. Moreover, compared with the state-of-the-art pruning-during-training approach, ClickTrain provides significant improvements both accuracy and compression ratio on the tested CNN models and datasets, under similar limited training time.
updated: Fri Apr 02 2021 18:55:15 GMT+0000 (UTC)
published: Fri Nov 20 2020 01:46:56 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アソシエイト