arXiv reaDer
U-Boost NAS:使用率が向上した微分可能なニューラルアーキテクチャ検索
U-Boost NAS: Utilization-Boosted Differentiable Neural Architecture Search
ターゲットプラットフォームでのリソース使用率の最適化は、DNN推論中に高いパフォーマンスを達成するための鍵です。推論の待ち時間、メモリフットプリント、およびエネルギー消費の最適化が提案されていますが、以前のハードウェア対応のニューラルアーキテクチャ検索(NAS)メソッドではリソースの使用が省略されており、DNNがターゲットの推論プラットフォームを最大限に活用できません。リソース使用率を効率的かつ正確にモデル化することは、特にGoogleTPUなどの広く使用されているアレイベースの推論アクセラレーターにとっては困難です。この作業では、タスクの精度と推論の待ち時間だけでなく、リソースの使用率も最適化する、新しいハードウェア対応のNASフレームワークを提案します。また、推論アクセラレータでのリソース使用率の新しい計算モデルを提案し、検証します。提案されたNASフレームワークと提案されたリソース使用率モデルを使用することにより、CIFAR-10およびImagenet-100データセットの画像分類で同等または改善された精度を達成しながら、以前のハードウェア対応NASメソッドと比較してDNN推論の2.8〜4倍の高速化を達成します。
Optimizing resource utilization in target platforms is key to achieving high performance during DNN inference. While optimizations have been proposed for inference latency, memory footprint, and energy consumption, prior hardware-aware neural architecture search (NAS) methods have omitted resource utilization, preventing DNNs to take full advantage of the target inference platforms. Modeling resource utilization efficiently and accurately is challenging, especially for widely-used array-based inference accelerators such as Google TPU. In this work, we propose a novel hardware-aware NAS framework that does not only optimize for task accuracy and inference latency, but also for resource utilization. We also propose and validate a new computational model for resource utilization in inference accelerators. By using the proposed NAS framework and the proposed resource utilization model, we achieve 2.8 - 4x speedup for DNN inference compared to prior hardware-aware NAS methods while attaining similar or improved accuracy in image classification on CIFAR-10 and Imagenet-100 datasets.
updated: Wed Mar 23 2022 13:44:15 GMT+0000 (UTC)
published: Wed Mar 23 2022 13:44:15 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アソシエイト