arXiv reaDer
1時間で83%のImageNet精度
83% ImageNet Accuracy in One Hour
EfficientNetsは、効率的にスケーリングされた畳み込みニューラルネットワークに基づく最先端の画像分類モデルのファミリーです。現在、EfficientNetsはトレーニングに数日かかる場合があります。たとえば、EfficientNet-B0モデルのトレーニングには、Cloud TPUv2-8ノードで23時間かかります。このホワイトペーパーでは、2048コアのTPU-v3ポッドでEfficientNetsのトレーニングをスケールアップする手法を検討します。このようなスケールでトレーニングを行うと、スピードアップを実現できます。大規模なバッチオプティマイザーの選択やレートスケジュールの学習、分散評価やバッチ正規化手法の利用など、1024TPU-v3コアでトレーニングをバッチサイズ65536にスケーリングするために必要な最適化について説明します。さらに、EfficientNetの動作を大規模に分析するために、ImageNetデータセットでトレーニングされたEfficientNetモデルのタイミングとパフォーマンスのベンチマークを示します。最適化により、ImageNetでEfficientNetを1時間4分で83%の精度でトレーニングできます。
EfficientNets are a family of state-of-the-art image classification models based on efficiently scaled convolutional neural networks. Currently, EfficientNets can take on the order of days to train; for example, training an EfficientNet-B0 model takes 23 hours on a Cloud TPU v2-8 node. In this paper, we explore techniques to scale up the training of EfficientNets on TPU-v3 Pods with 2048 cores, motivated by speedups that can be achieved when training at such scales. We discuss optimizations required to scale training to a batch size of 65536 on 1024 TPU-v3 cores, such as selecting large batch optimizers and learning rate schedules as well as utilizing distributed evaluation and batch normalization techniques. Additionally, we present timing and performance benchmarks for EfficientNet models trained on the ImageNet dataset in order to analyze the behavior of EfficientNets at scale. With our optimizations, we are able to train EfficientNet on ImageNet to an accuracy of 83% in 1 hour and 4 minutes.
updated: Fri Oct 30 2020 19:27:11 GMT+0000 (UTC)
published: Fri Oct 30 2020 19:27:11 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アソシエイト