arXiv reaDer
生成モデルによる微分可能なアーキテクチャ検索の改善
Improving Differentiable Architecture Search with a Generative Model
DARTSのような微分可能ニューラルアーキテクチャ検索(NAS)アルゴリズムでは、モデルの重みを更新するために使用されるトレーニングセットと、モデルアーキテクチャを更新するために使用される検証セットは、同じデータ分布からサンプリングされます。したがって、データセット内の珍しい機能は、トレーニング中に十分な注意を引くことができません。このホワイトペーパーでは、より複雑なNASアルゴリズムを導入する代わりに、高品質の合成データセットをトレーニングに追加することで、分類モデルがその弱点を特定し、認識精度を向上させることができるという考えを探ります。 「生成モデル(DASGM)を使用した微分可能アーキテクチャ検索」と呼ばれるトレーニング戦略を紹介します。DASGMでは、トレーニングセットを使用して分類モデルの重みを更新し、合成データセットを使用してそのアーキテクチャをトレーニングします。トレーニングセットとは異なる分布。これは、分類モデルがその弱点を特定するためのより優れた機能を学習するのに役立ちます。DASGMをマルチレベルの最適化フレームワークに定式化し、それを解決するための効果的なアルゴリズムを開発します。CIFAR-10、CIFAR-100、およびImageNetはDASGMの有効性を実証しました。コードが利用可能になります。
In differentiable neural architecture search (NAS) algorithms like DARTS, the training set used to update model weight and the validation set used to update model architectures are sampled from the same data distribution. Thus, the uncommon features in the dataset fail to receive enough attention during training. In this paper, instead of introducing more complex NAS algorithms, we explore the idea that adding quality synthesized datasets into training can help the classification model identify its weakness and improve recognition accuracy. We introduce a training strategy called ``Differentiable Architecture Search with a Generative Model(DASGM)." In DASGM, the training set is used to update the classification model weight, while a synthesized dataset is used to train its architecture. The generated images have different distributions from the training set, which can help the classification model learn better features to identify its weakness. We formulate DASGM into a multi-level optimization framework and develop an effective algorithm to solve it. Experiments on CIFAR-10, CIFAR-100, and ImageNet have demonstrated the effectiveness of DASGM. Code will be made available.
updated: Tue Nov 30 2021 23:28:02 GMT+0000 (UTC)
published: Tue Nov 30 2021 23:28:02 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アソシエイト