arXiv reaDer
DHA:データ拡張ポリシー、ハイパーパラメータ、アーキテクチャのエンドツーエンドの共同最適化
DHA: End-to-End Joint Optimization of Data Augmentation Policy, Hyper-parameter and Architecture
自動機械学習(AutoML)には通常、データ拡張(DA)ポリシー、ハイパーパラメーター最適化(HPO)、ニューラルアーキテクチャ検索(NAS)などのいくつかの重要なコンポーネントが含まれます。これらのコンポーネントを分離して自動化するための多くの戦略が開発されていますが、検索ディメンションが大幅に増加し、各コンポーネントの入力タイプが異なるため、これらのコンポーネントの共同最適化は依然として困難です。一方、これらのコンポーネントを順番に実行するには、人間の専門家による注意深い調整が必要になることが多く、最適な結果が得られない可能性があります。これと並行して、NASに導入する前に、最初に最適なアーキテクチャを検索してから再トレーニングするという一般的な方法では、検索段階と再トレーニング段階の間のパフォーマンスの相関が低いことがよくあります。 AutoMLコンポーネントを統合し、検索の最後にすぐに使用できるモデルを返すエンドツーエンドのソリューションが望まれます。これらを考慮して、データ拡張ポリシー、ハイパーパラメータ、アーキテクチャの共同最適化を実現するDHAを提案します。具体的には、エンドツーエンドのNASは、圧縮された低次元の特徴空間を最適化することで差別化可能な方法で実現され、DAポリシーとHPOは同時に動的に更新されます。実験によると、DHAはさまざまなデータセットで最先端の(SOTA)結果を達成し、特にセルベースの検索スペースを備えたImageNetで77.4%の精度を達成します。これは、現在のSOTAよりも0.5%高くなっています。私たちの知る限りでは、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. Meanwhile, conducting these components in a sequence often requires careful coordination by human experts and may lead to sub-optimal results. 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 search 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 updated dynamically at the same time. Experiments show that DHA achieves state-of-the-art (SOTA) results on various datasets, especially 77.4% accuracy on ImageNet with cell based search space, which is higher than current SOTA by 0.5%. 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: Mon Sep 13 2021 08:12:50 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アソシエイト