arXiv reaDer
リソース適応スーパーネット トレーニングのための優先サブネット サンプリング
Prioritized Subnet Sampling for Resource-Adaptive Supernet Training
リソース適応スーパーネットは、動的に利用可能なリソースに適合するように、推論のためにサブネットを調整します。このホワイト ペーパーでは、PSS-Net と呼ばれるリソース適応型スーパーネットをトレーニングするために、優先順位付けされたサブネット サンプリングを提案します。私たちは複数のサブネット プールを維持しており、それぞれに同様のリソース消費量を持つ実質的なサブネットの情報が保存されています。リソースの制約を考慮して、このリソースの制約で調整されたサブネットは、事前定義されたサブネット構造空間からサンプリングされ、高品質のサブネットが対応するサブネット プールに挿入されます。その後、サンプリングは徐々にサブネット プールからサブネットをサンプリングする傾向があります。さらに、サンプリングがサブネット プールからのものである場合、より優れたパフォーマンス メトリックを持つものには、PSS-Net をトレーニングする優先度が高く割り当てられます。トレーニングの最後に、PSS-Net は各プールで最適なサブネットを保持し、利用可能なリソースが異なる場合に、推論のために高品質のサブネットを高速に切り替える資格を与えます。 MobileNet-V1/V2 および ResNet-50 を使用した ImageNet での実験では、PSS-Net が最先端のリソース適応スーパーネットよりも優れたパフォーマンスを発揮できることが示されています。私たちのプロジェクトは、https://github.com/chenbong/PSS-Net で公開されています。
A resource-adaptive supernet adjusts its subnets for inference to fit the dynamically available resources. In this paper, we propose prioritized subnet sampling to train a resource-adaptive supernet, termed PSS-Net. We maintain multiple subnet pools, each of which stores the information of substantial subnets with similar resource consumption. Considering a resource constraint, subnets conditioned on this resource constraint are sampled from a pre-defined subnet structure space and high-quality ones will be inserted into the corresponding subnet pool. Then, the sampling will gradually be prone to sampling subnets from the subnet pools. Moreover, the one with a better performance metric is assigned with higher priority to train our PSS-Net, if sampling is from a subnet pool. At the end of training, our PSS-Net retains the best subnet in each pool to entitle a fast switch of high-quality subnets for inference when the available resources vary. Experiments on ImageNet using MobileNet-V1/V2 and ResNet-50 show that our PSS-Net can well outperform state-of-the-art resource-adaptive supernets. Our project is publicly available at https://github.com/chenbong/PSS-Net.
updated: Mon Feb 13 2023 10:36:50 GMT+0000 (UTC)
published: Sun Sep 12 2021 04:43:51 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アソシエイト