arXiv reaDer
EfficientNet:畳み込みニューラルネットワークのモデルスケーリングの再考
EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
たたみ込みニューラルネットワーク(ConvNets)は、通常、固定のリソースバジェットで開発され、より多くのリソースが利用可能な場合は、精度を高めるためにスケールアップされます。このホワイトペーパーでは、モデルのスケーリングを体系的に調査し、ネットワークの深さ、幅、および解像度を慎重にバランスさせることでパフォーマンスが向上することを確認します。この観察に基づいて、シンプルながら非常に効果的な複合係数を使用して、深さ/幅/解像度のすべての次元を均一にスケーリングする新しいスケーリング方法を提案します。 MobileNetとResNetのスケールアップにおけるこの方法の有効性を示します。さらに進むために、ニューラルアーキテクチャ検索を使用して新しいベースラインネットワークを設計し、それをスケールアップして、EfficientNetsと呼ばれるモデルのファミリを取得します。特に、当社のEfficientNet-B7は、ImageNetで最先端の84.3%のトップ1の精度を達成し、既存の最高のConvNetよりも8.4倍小さく、推論で6.1倍高速です。私たちのEfficientNetsは、CIFAR-100(91.7%)、Flowers(98.8%)、およびその他の3つの転送学習データセットで、十分に転送し、最先端の精度を達成しています。ソースコードはhttps://github.com/tensorflow/tpu/tree/master/models/official/efficientnetにあります。
Convolutional Neural Networks (ConvNets) are commonly developed at a fixed resource budget, and then scaled up for better accuracy if more resources are available. In this paper, we systematically study model scaling and identify that carefully balancing network depth, width, and resolution can lead to better performance. Based on this observation, we propose a new scaling method that uniformly scales all dimensions of depth/width/resolution using a simple yet highly effective compound coefficient. We demonstrate the effectiveness of this method on scaling up MobileNets and ResNet. To go even further, we use neural architecture search to design a new baseline network and scale it up to obtain a family of models, called EfficientNets, which achieve much better accuracy and efficiency than previous ConvNets. In particular, our EfficientNet-B7 achieves state-of-the-art 84.3% top-1 accuracy on ImageNet, while being 8.4x smaller and 6.1x faster on inference than the best existing ConvNet. Our EfficientNets also transfer well and achieve state-of-the-art accuracy on CIFAR-100 (91.7%), Flowers (98.8%), and 3 other transfer learning datasets, with an order of magnitude fewer parameters. Source code is at https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet.
updated: Fri Sep 11 2020 05:08:01 GMT+0000 (UTC)
published: Tue May 28 2019 17:05:32 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アソシエイト