arXiv reaDer
AlphaNet:Alpha-Divergenceを使用したスーパーネットのトレーニングの改善
AlphaNet: Improved Training of Supernets with Alpha-Divergence
重み共有ニューラルアーキテクチャ検索(NAS)は、効率的なニューラルアーキテクチャ設計を自動化するための効果的な手法です。重み共有NASは、すべてのアーキテクチャをサブネットワークとして組み立て、スーパーネットをサブネットワークと共同でトレーニングするスーパーネットを構築します。ウェイトシェアリングNASの成功は、スーパーネットの知識をサブネットワークに抽出することに大きく依存しています。ただし、広く使用されている蒸留ダイバージェンス、つまり KL ダイバージェンスは、教師のスーパーネットの不確実性を過大評価または過小評価する学生サブネットワークにつながり、サブネットワークのパフォーマンスが低下する可能性があることがわかりました。この作業では、より一般化されたアルファ ダイバージェンスでスーパーネット トレーニングを改善することを提案します。アルファ発散を適応的に選択することにより、教師モデルの不確実性の過大評価または過小評価を同時に防止します。提案されたアルファダイバージェンスベースのスーパーネットトレーニングをスリミング可能なニューラルネットワークと重み共有NASの両方に適用し、大幅な改善を示します。具体的には、発見されたモデルファミリであるAlphaNetは、BigNAS、Once-for-Allネットワーク、AttentiveNASなどの幅広いFLOPレジームで従来技術のモデルよりも優れています。わずか4億4400万フロップスで80.0%のImageNetトップ1精度を達成します。コードと事前トレーニング済みモデルは、https://github.com/facebookresearch/AlphaNetで入手できます。
Weight-sharing neural architecture search (NAS) is an effective technique for automating efficient neural architecture design. Weight-sharing NAS builds a supernet that assembles all the architectures as its sub-networks and jointly trains the supernet with the sub-networks. The success of weight-sharing NAS heavily relies on distilling the knowledge of the supernet to the sub-networks. However, we find that the widely used distillation divergence, i.e., KL divergence, may lead to student sub-networks that over-estimate or under-estimate the uncertainty of the teacher supernet, leading to inferior performance of the sub-networks. In this work, we propose to improve the supernet training with a more generalized alpha-divergence. By adaptively selecting the alpha-divergence, we simultaneously prevent the over-estimation or under-estimation of the uncertainty of the teacher model. We apply the proposed alpha-divergence based supernets training to both slimmable neural networks and weight-sharing NAS, and demonstrate significant improvements. Specifically, our discovered model family, AlphaNet, outperforms prior-art models on a wide range of FLOPs regimes, including BigNAS, Once-for-All networks, and AttentiveNAS. We achieve ImageNet top-1 accuracy of 80.0% with only 444M FLOPs. Our code and pretrained models are available at https://github.com/facebookresearch/AlphaNet.
updated: Thu Jun 10 2021 17:19:59 GMT+0000 (UTC)
published: Tue Feb 16 2021 04:23:55 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アソシエイト