arXiv reaDer
BatchQuant:堅牢なQuantizerを使用したすべての量子化アーキテクチャ検索
BatchQuant: Quantized-for-all Architecture Search with Robust Quantizer
エッジデバイスでのディープラーニングモデルのアプリケーションが加速するペースで増加するにつれて、さまざまなリソースの制約があるさまざまなシナリオへの迅速な適応が、モデル展開の重要な側面になりました。その結果、適応構成を使用したモデル最適化戦略がますます一般的になっています。シングルショット量子化ニューラルアーキテクチャ検索は、モデルアーキテクチャと量子化ポリシーの両方で柔軟性を享受しますが、結合された検索スペースには、重み共有スーパーネットをトレーニングするときの不安定さや、指数関数的に増大する検索スペースのナビゲートの難しさなど、多くの課題が伴います。既存の方法は、アーキテクチャの検索スペースをオプションの小さなセットに制限するか、量子化ポリシーの検索スペースを固定精度のポリシーに制限する傾向があります。この目的のために、コンパクト、シングルショット、混合精度、重量共有スーパーネットの高速で安定したトレーニングを可能にする堅牢な量子化器の定式化であるBatchQuantを提案します。 BatchQuantを使用して、以前の方法よりも大幅に短いGPU時間でコンパクトなスーパーネット(10 ^ 76を超える量子化サブネットを提供)をトレーニングします。私たちのアプローチであるQuantized-for-all(QFA)は、ワンショットの重み共有NASスーパーネットをシームレスに拡張して、再トレーニングなしで任意の超低ビット幅の混合精度量子化ポリシーを持つサブネットをサポートする最初のアプローチです。 QFAは、ハードウェア対応の共同ニューラルアーキテクチャ検索と量子化に新しい可能性を開きます。 ImageNetでのメソッドの有効性を示し、複雑度の低い制約(<20 MFLOPs)でSOTATop-1の精度を達成します。コードとモデルはhttps://github.com/bhpfelix/QFAで公開されます。
As the applications of deep learning models on edge devices increase at an accelerating pace, fast adaptation to various scenarios with varying resource constraints has become a crucial aspect of model deployment. As a result, model optimization strategies with adaptive configuration are becoming increasingly popular. While single-shot quantized neural architecture search enjoys flexibility in both model architecture and quantization policy, the combined search space comes with many challenges, including instability when training the weight-sharing supernet and difficulty in navigating the exponentially growing search space. Existing methods tend to either limit the architecture search space to a small set of options or limit the quantization policy search space to fixed precision policies. To this end, we propose BatchQuant, a robust quantizer formulation that allows fast and stable training of a compact, single-shot, mixed-precision, weight-sharing supernet. We employ BatchQuant to train a compact supernet (offering over 10^76 quantized subnets) within substantially fewer GPU hours than previous methods. Our approach, Quantized-for-all (QFA), is the first to seamlessly extend one-shot weight-sharing NAS supernet to support subnets with arbitrary ultra-low bitwidth mixed-precision quantization policies without retraining. QFA opens up new possibilities in joint hardware-aware neural architecture search and quantization. We demonstrate the effectiveness of our method on ImageNet and achieve SOTA Top-1 accuracy under a low complexity constraint (<20 MFLOPs). The code and models will be made publicly available at https://github.com/bhpfelix/QFA.
updated: Wed May 19 2021 06:56:43 GMT+0000 (UTC)
published: Wed May 19 2021 06:56:43 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アソシエイト