arXiv reaDer
伸縮性のあるセルは、DARTSの検索を改善します
Stretchable Cells Help DARTS Search Better
微分可能ニューラルアーキテクチャ検索(DARTS)は、柔軟で多様な細胞型の発見に大きな成功を収めています。評価ギャップを減らすために、スーパーネットはターゲットネットワークと同一のレイヤーを持つことが期待されます。ただし、この一貫した検索でも、特にレイヤーの少ないスーパーネットでは、検索されたセルのパフォーマンスが低下することがよくあります。これは、現在のDARTSメソッドではセルが広く浅い傾向があり、このトポロジの崩壊により、検索されたセルが最適ではなくなるためです。この論文では、セルに明示的な伸縮性を与えることでこの問題を軽減し、操作タイプとトポロジの両方で同時に伸縮性セルに検索を直接実装できるようにします。具体的には、トポロジ変数のセットと組み合わせ確率分布を導入して、ターゲットトポロジを明示的にモデル化します。より多様で複雑なトポロジーにより、私たちの方法はさまざまなレイヤー番号にうまく適応します。 CIFAR-10とImageNetでの広範な実験は、伸縮性セルがより少ないレイヤーとパラメーターでより優れたパフォーマンスを実現することを示しています。たとえば、私たちの方法では、CIFAR-10データセットの精度を0.28%向上させ、パラメーターを45%削減したり、ImageNetデータセットの同様のFLOPで2.9%向上させたりできます。
Differentiable neural architecture search (DARTS) has gained much success in discovering flexible and diverse cell types. To reduce the evaluation gap, the supernet is expected to have identical layers with the target network. However, even for this consistent search, the searched cells often suffer from poor performance, especially for the supernet with fewer layers, as current DARTS methods are prone to wide and shallow cells, and this topology collapse induces sub-optimal searched cells. In this paper, we alleviate this issue by endowing the cells with explicit stretchability, so the search can be directly implemented on our stretchable cells for both operation type and topology simultaneously. Concretely, we introduce a set of topological variables and a combinatorial probabilistic distribution to explicitly model the target topology. With more diverse and complex topologies, our method adapts well for various layer numbers. Extensive experiments on CIFAR-10 and ImageNet show that our stretchable cells obtain better performance with fewer layers and parameters. For example, our method can improve DARTS by 0.28% accuracy on CIFAR-10 dataset with 45% parameters reduced or 2.9% with similar FLOPs on ImageNet dataset.
updated: Mon Apr 18 2022 13:43:38 GMT+0000 (UTC)
published: Wed Nov 18 2020 14:15:51 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アソシエイト