arXiv reaDer
DHA: データ拡張ポリシー、ハイパーパラメータ、およびアーキテクチャのエンドツーエンドの共同最適化
DHA: End-to-End Joint Optimization of Data Augmentation Policy, Hyper-parameter and Architecture
通常、自動機械学習 (AutoML) には、Data Augmentation (DA) ポリシー、Hyper-Parameter Optimization (HPO)、Neural Architecture Search (NAS) など、いくつかの重要なコンポーネントが含まれます。これらのコンポーネントを個別に自動化するための多くの戦略が開発されていますが、検索次元が大幅に増加し、各コンポーネントのバリアント入力タイプが多いため、これらのコンポーネントの共同最適化は依然として困難です。これと並行して、最初に最適なアーキテクチャを検索し、NAS に展開する前にそれを再トレーニングするという一般的な方法では、検索段階と再トレーニング段階の間のパフォーマンスの相関関係が低くなることがよくあります。 AutoML コンポーネントを統合し、検索の最後にすぐに使用できるモデルを返すエンドツーエンドのソリューションが望ましいです。これらを踏まえ、Data Augmentation Policy、Hyper-parameter、Architectureの連携最適化を実現するDHAを提案します。具体的には、圧縮された低次元の特徴空間を最適化することにより、エンド ツー エンドの NAS が微分可能な方法で実現されます。一方、DA ポリシーと HPO は、ネットワーク パラメータとネットワーク アーキテクチャの更新に同時に適応する動的スケジューラと見なされます。時間。実験は、DHA がさまざまなデータセットと検索空間で最先端 (SOTA) の結果を達成することを示しています。私たちの知る限り、私たちは再トレーニングなしでエンドツーエンドの方法で DA ポリシー、NAS、および HPO を効率的かつ共同で最適化した最初の企業です。
Automated machine learning (AutoML) usually involves several crucial components, such as Data Augmentation (DA) policy, Hyper-Parameter Optimization (HPO), and Neural Architecture Search (NAS). Although many strategies have been developed for automating these components in separation, joint optimization of these components remains challenging due to the largely increased search dimension and the variant input types of each component. In parallel to this, the common practice of searching for the optimal architecture first and then retraining it before deployment in NAS often suffers from low performance correlation between the searching and retraining stages. An end-to-end solution that integrates the AutoML components and returns a ready-to-use model at the end of the search is desirable. In view of these, we propose DHA, which achieves joint optimization of Data augmentation policy, Hyper-parameter and Architecture. Specifically, end-to-end NAS is achieved in a differentiable manner by optimizing a compressed lower-dimensional feature space, while DA policy and HPO are regarded as dynamic schedulers, which adapt themselves to the update of network parameters and network architecture at the same time. Experiments show that DHA achieves state-of-the-art (SOTA) results on various datasets and search spaces. To the best of our knowledge, we are the first to efficiently and jointly optimize DA policy, NAS, and HPO in an end-to-end manner without retraining.
updated: Thu Nov 03 2022 12:33:19 GMT+0000 (UTC)
published: Mon Sep 13 2021 08:12:50 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アソシエイト