arXiv reaDer
組み込みシステムのための効率的でデータにとらわれない画像分類トレーニングパイプラインに向けて
Towards Efficient and Data Agnostic Image Classification Training Pipeline for Embedded Systems
今日、深層学習ベースの方法は、広く一般的に使用されているデータセット(ImageNet、CIFAR、SVHN、Caltech 101、SUN397など)の中で画像分類タスクで目覚ましい進歩を遂げています。上記の各データセットでのSOTAパフォーマンスは、モデルアーキテクチャを注意深く調整し、ターゲットデータのプロパティに応じてトリックをトレーニングすることで得られます。このアプローチでは学術記録を設定できますが、平均的なデータサイエンティストが、実際に遭遇するすべての画像分類タスクに対して高度なトレーニングパイプラインを構築するのに十分なリソースを持っていることは非現実的です。この作業は、画像分類の最新の拡張および正則化方法のレビューと、エポックの総数、初期学習率の値、およびそのスケジュールなど、最も重要なハイパーパラメーターのいくつかを自動的に選択する方法の調査に焦点を当てています。軽量で最新のCNNアーキテクチャ(bileNetV3やEfficientNetなど)を備えたトレーニング手順、十分なレベルの正則化、データ学習レートスケジュールへの適応により、パラメータを手動で調整することなく、さまざまなダウンストリーム画像分類タスクで妥当なパフォーマンスを実現できます。それぞれの特定のタスク。結果として得られるモデルは計算効率が高く、OpenVINOツールキットを使用してCPUに展開できます。ソースコードは、OpenVINO Training Extensions(https://github.com/openvinotoolkit/training_extensions)の一部として入手できます。
Nowadays deep learning-based methods have achieved a remarkable progress at the image classification task among a wide range of commonly used datasets (ImageNet, CIFAR, SVHN, Caltech 101, SUN397, etc.). SOTA performance on each of the mentioned datasets is obtained by careful tuning of the model architecture and training tricks according to the properties of the target data. Although this approach allows setting academic records, it is unrealistic that an average data scientist would have enough resources to build a sophisticated training pipeline for every image classification task he meets in practice. This work is focusing on reviewing the latest augmentation and regularization methods for the image classification and exploring ways to automatically choose some of the most important hyperparameters: total number of epochs, initial learning rate value and it's schedule. Having a training procedure equipped with a lightweight modern CNN architecture (like bileNetV3 or EfficientNet), sufficient level of regularization and adaptive to data learning rate schedule, we can achieve a reasonable performance on a variety of downstream image classification tasks without manual tuning of parameters to each particular task. Resulting models are computationally efficient and can be deployed to CPU using the OpenVINO toolkit. Source code is available as a part of the OpenVINO Training Extensions (https://github.com/openvinotoolkit/training_extensions).
updated: Mon Aug 16 2021 12:38:05 GMT+0000 (UTC)
published: Mon Aug 16 2021 12:38:05 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アソシエイト