arXiv reaDer
PR-DARTS:剪定ベースの微分可能なアーキテクチャ検索
PR-DARTS: Pruning-Based Differentiable Architecture Search
エッジデバイスへの畳み込みニューラルネットワーク(CNN)の展開は、パフォーマンス要件と利用可能な処理能力の間の実質的なギャップによって妨げられます。最近の研究では、CNNの計算オーバーヘッドを削減するためのネットワークプルーニング手法の開発が大きく進歩しましたが、特に高いプルーニング比では、かなりの精度の低下が残っています。非剪定ネットワーク用に設計されたアーキテクチャが剪定ネットワークには効果的でない可能性があることを疑問視し、新しい検索スペースと新しい検索目的を定義することにより、剪定方法のアーキテクチャを検索することを提案します。剪定されたネットワークの一般化を改善するために、2つの新しいPrunedConvおよびPrunedLinear操作を提案します。具体的には、これらの操作は、剪定されたネットワークの目的関数を正規化することにより、不安定な勾配の問題を軽減します。提案された検索目的により、剪定された重み要素に関するアーキテクチャパラメータをトレーニングできます。定量分析は、検索されたアーキテクチャが、CIFAR-10およびImageNetの最先端のプルーニングネットワークで使用されているアーキテクチャよりも優れていることを示しています。ハードウェアの有効性に関して、PR-DARTSはMobileNet-v2の精度を73.44%から81.35%(+ 7.91%の改善)に向上させ、3.87倍高速に実行します。
The deployment of Convolutional Neural Networks (CNNs) on edge devices is hindered by the substantial gap between performance requirements and available processing power. While recent research has made large strides in developing network pruning methods for reducing the computing overhead of CNNs, there remains considerable accuracy loss, especially at high pruning ratios. Questioning that the architectures designed for non-pruned networks might not be effective for pruned networks, we propose to search architectures for pruning methods by defining a new search space and a novel search objective. To improve the generalization of the pruned networks, we propose two novel PrunedConv and PrunedLinear operations. Specifically, these operations mitigate the problem of unstable gradients by regularizing the objective function of the pruned networks. The proposed search objective enables us to train architecture parameters regarding the pruned weight elements. Quantitative analyses demonstrate that our searched architectures outperform those used in the state-of-the-art pruning networks on CIFAR-10 and ImageNet. In terms of hardware effectiveness, PR-DARTS increases MobileNet-v2's accuracy from 73.44% to 81.35% (+7.91% improvement) and runs 3.87× faster.
updated: Thu Jul 14 2022 14:53:50 GMT+0000 (UTC)
published: Thu Jul 14 2022 14:53:50 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アソシエイト