arXiv reaDer
Pi-NAS:スーパーネットトレーニングの一貫性シフトを削減することによるニューラルアーキテクチャ検索の改善
Pi-NAS: Improving Neural Architecture Search by Reducing Supernet Training Consistency Shift
最近提案されたニューラルアーキテクチャ検索(NAS)メソッドは、スーパーネット内の数十億のアーキテクチャを共同トレーニングし、スーパーネットから切り離されたネットワークの重みを使用してそれらの潜在的な精度を推定します。ただし、アーキテクチャの予測精度と実際の機能の間のランキング相関は正しくないため、既存のNASメソッドのジレンマが発生します。このランキング相関問題は、機能シフトやパラメーターシフトを含むスーパーネットトレーニングの一貫性シフトに起因すると考えられます。特徴シフトは、ランダムパスサンプリングによる隠れ層の動的入力分布として識別されます。入力分布の動的な変化は損失の下降に影響を与え、最終的にアーキテクチャのランク付けに影響を与えます。パラメータシフトは、異なるトレーニングステップの異なるパスにある共有レイヤーの矛盾するパラメータ更新として識別されます。急速に変化するパラメータは、アーキテクチャのランキングを維持できませんでした。 Pi-NASと呼ばれる重要なスーパーネット-Piモデルを使用して、これら2つのシフトに同時に対処します。具体的には、クロスパス学習を含むスーパーネット-Piモデルを採用して、異なるパス間の機能の一貫性のシフトを減らします。一方、パラメータのシフトとモデルの衝突を克服するために、負のサンプルを含む新しい自明でない平均教師を採用します。さらに、Pi-NASは監視なしで実行されるため、より転送可能なアーキテクチャを検索できます。 ImageNetとさまざまなダウンストリームタスク(COCO 2017、ADE20K、Cityscapesなど)での広範な実験により、監視対象NASと比較したPi-NASの有効性と普遍性が実証されています。コードを参照してください:https://github.com/Ernie1/Pi-NAS。
Recently proposed neural architecture search (NAS) methods co-train billions of architectures in a supernet and estimate their potential accuracy using the network weights detached from the supernet. However, the ranking correlation between the architectures' predicted accuracy and their actual capability is incorrect, which causes the existing NAS methods' dilemma. We attribute this ranking correlation problem to the supernet training consistency shift, including feature shift and parameter shift. Feature shift is identified as dynamic input distributions of a hidden layer due to random path sampling. The input distribution dynamic affects the loss descent and finally affects architecture ranking. Parameter shift is identified as contradictory parameter updates for a shared layer lay in different paths in different training steps. The rapidly-changing parameter could not preserve architecture ranking. We address these two shifts simultaneously using a nontrivial supernet-Pi model, called Pi-NAS. Specifically, we employ a supernet-Pi model that contains cross-path learning to reduce the feature consistency shift between different paths. Meanwhile, we adopt a novel nontrivial mean teacher containing negative samples to overcome parameter shift and model collision. Furthermore, our Pi-NAS runs in an unsupervised manner, which can search for more transferable architectures. Extensive experiments on ImageNet and a wide range of downstream tasks (e.g., COCO 2017, ADE20K, and Cityscapes) demonstrate the effectiveness and universality of our Pi-NAS compared to supervised NAS. See Codes: https://github.com/Ernie1/Pi-NAS.
updated: Sun Aug 22 2021 09:08:48 GMT+0000 (UTC)
published: Sun Aug 22 2021 09:08:48 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アソシエイト