arXiv reaDer
シングルダーツ:安定したアーキテクチャ検索に向けて
Single-DARTS: Towards Stable Architecture Search
微分可能アーキテクチャ検索(DARTS)は、ニューラルアーキテクチャ検索(NAS)のマイルストーンであり、シンプルさと小さな検索コストを誇っています。ただし、DARTSは依然として頻繁なパフォーマンスの低下に悩まされています。これは、接続のスキップ、ゼロ、プーリングなどの一部の操作がアーキテクチャを支配している場合に発生します。この論文では、この現象が2レベルの最適化に起因することを最初に指摘しました。単一レベルの最適化を使用し、ネットワークの重みとアーキテクチャパラメータを同じデータバッチで同時に更新するSingle-DARTSを提案します。単一レベルの最適化が以前に試みられたとしても、この重要な点について体系的な説明を提供している文献はありません。シングルDARTSは、2レベルの最適化に取って代わり、パフォーマンスの低下を明らかに軽減するだけでなく、アーキテクチャ検索の安定性を高めます。実験結果は、Single-DARTSが主流の検索スペースで最先端のパフォーマンスを達成することを示しています。たとえば、NAS-Benchmark-201では、検索されたアーキテクチャはほぼ最適なアーキテクチャです。また、シングルレベルの最適化フレームワークがバイレベルのフレームワークよりもはるかに安定していることも検証します。このシンプルで効果的な方法が、差分アーキテクチャ検索に関する洞察を与えることを願っています。コードはhttps://github.com/PencilAndBike/Single-DARTS.gitで入手できます。
Differentiable architecture search (DARTS) marks a milestone in Neural Architecture Search (NAS), boasting simplicity and small search costs. However, DARTS still suffers from frequent performance collapse, which happens when some operations, such as skip connections, zeroes and poolings, dominate the architecture. In this paper, we are the first to point out that the phenomenon is attributed to bi-level optimization. We propose Single-DARTS which merely uses single-level optimization, updating network weights and architecture parameters simultaneously with the same data batch. Even single-level optimization has been previously attempted, no literature provides a systematic explanation on this essential point. Replacing the bi-level optimization, Single-DARTS obviously alleviates performance collapse as well as enhances the stability of architecture search. Experiment results show that Single-DARTS achieves state-of-the-art performance on mainstream search spaces. For instance, on NAS-Benchmark-201, the searched architectures are nearly optimal ones. We also validate that the single-level optimization framework is much more stable than the bi-level one. We hope that this simple yet effective method will give some insights on differential architecture search. The code is available at https://github.com/PencilAndBike/Single-DARTS.git.
updated: Wed Aug 18 2021 13:00:39 GMT+0000 (UTC)
published: Wed Aug 18 2021 13:00:39 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アソシエイト