arXiv reaDer
プロキシ データによるニューラル アーキテクチャ検索の高速化
Accelerating Neural Architecture Search via Proxy Data
ニューラル アーキテクチャ検索 (NAS) への関心が高まっているにもかかわらず、NAS のかなりの計算コストは研究者にとって障害となっています。したがって、検索パフォーマンスを犠牲にすることなく、プロキシ データ、つまりターゲット データの代表的なサブセットを使用して NAS のコストを削減することを提案します。データ選択は様々な分野で利用されていますが、NAS-Bench-1shot1が提供するNASアルゴリズムの既存の選択方法を評価すると、それらが必ずしもNASに適しているとは限らず、新しい選択方法が必要であることがわかります。様々な選択手法を用いて構築されたプロキシデータをデータエントロピーにより分析することで、NASに合わせた新しいプロキシデータ選択手法を提案します。有効性を経験的に実証するために、さまざまなデータセット、検索スペース、NAS アルゴリズムにわたって徹底的な実験を行います。その結果、提案された選択を備えた NAS アルゴリズムは、データセット全体を使用して取得されたアーキテクチャと競合するアーキテクチャを発見します。これにより、検索コストが大幅に削減されます。提案された選択で DARTS を実行するには、CIFAR-10 では 40 分、ImageNet では 1 つの GPU を使用して 7.5 時間しかかかりません。さらに、提案された選択を使用して ImageNet で検索されたアーキテクチャを CIFAR-10 に逆転送すると、2.4% の最先端のテスト エラーが発生します。私たちのコードは https://github.com/nabk89/NAS-with-Proxy-data で入手できます。
Despite the increasing interest in neural architecture search (NAS), the significant computational cost of NAS is a hindrance to researchers. Hence, we propose to reduce the cost of NAS using proxy data, i.e., a representative subset of the target data, without sacrificing search performance. Even though data selection has been used across various fields, our evaluation of existing selection methods for NAS algorithms offered by NAS-Bench-1shot1 reveals that they are not always appropriate for NAS and a new selection method is necessary. By analyzing proxy data constructed using various selection methods through data entropy, we propose a novel proxy data selection method tailored for NAS. To empirically demonstrate the effectiveness, we conduct thorough experiments across diverse datasets, search spaces, and NAS algorithms. Consequently, NAS algorithms with the proposed selection discover architectures that are competitive with those obtained using the entire dataset. It significantly reduces the search cost: executing DARTS with the proposed selection requires only 40 minutes on CIFAR-10 and 7.5 hours on ImageNet with a single GPU. Additionally, when the architecture searched on ImageNet using the proposed selection is inversely transferred to CIFAR-10, a state-of-the-art test error of 2.4% is yielded. Our code is available at https://github.com/nabk89/NAS-with-Proxy-data.
updated: Wed Jun 09 2021 03:08:53 GMT+0000 (UTC)
published: Wed Jun 09 2021 03:08:53 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アソシエイト