arXiv reaDer
リソース適応型スーパーネットトレーニングのための優先サブネットサンプリング
Prioritized Subnet Sampling for Resource-Adaptive Supernet Training
リソース適応型スーパーネットは、動的に利用可能なリソースに合うように推論のためにサブネットを調整します。この論文では、PSS-Netと呼ばれるリソース適応型スーパーネットをトレーニングするための優先サブネットサンプリングを提案します。複数のサブネットプールを維持しており、各プールには、同様のリソース消費を持つ実質的なサブネットの情報が格納されています。リソースの制約を考慮して、このリソースの制約を条件とするサブネットは、事前定義されたサブネット構造スペースからサンプリングされ、高品質のサブネットが対応するサブネットプールに挿入されます。その後、サンプリングは徐々にサブネットプールからサブネットをサンプリングする傾向があります。さらに、サンプリングがサブネットプールからのものである場合、パフォーマンスメトリックが優れているものには、PSS-Netをトレーニングするための優先度が高く割り当てられます。トレーニングの最後に、PSS-Netは各プールで最適なサブネットを保持し、利用可能なリソースが変化したときに推論するための高品質のサブネットの高速切り替えを可能にします。 MobileNetV1 / V2を使用した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 MobileNetV1/V2 show that our PSS-Net can well outperform state-of-the-art resource-adaptive supernets. Our project is at https://github.com/chenbong/PSS-Net.
updated: Sun Sep 12 2021 04:43:51 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アソシエイト