arXiv reaDer
LA3: 効率的なラベル対応 AutoAugment
LA3: Efficient Label-Aware AutoAugment
自動拡張は、データ拡張ポリシーを検索してディープ ニューラル ネットワーク トレーニングの一般化可能性を向上させるための新しい効果的な手法です。既存の作業のほとんどは、サンプルやクラスのバリエーションを考慮せずに、特定のデータセット内のすべてのデータ サンプルに適用できる統一されたポリシーを構築することに重点を置いています。この論文では、Label-Aware AutoAugment (LA3) という名前の新しい 2 段階のデータ拡張アルゴリズムを提案します。これは、ラベル情報を利用し、異なるラベルのサンプルに対して個別に拡張ポリシーを学習します。 LA3 は 2 つの学習段階で構成され、最初の段階では、個々の拡張方法が評価され、ニューラル予測子によって支援されるベイジアン最適化によってラベルごとにランク付けされます。これにより、低検索コストで各ラベルの効果的な拡張手法を特定できます。第 2 段階では、効果的な拡張と補完的な拡張の選択から複合拡張ポリシーが構築されます。これにより、パフォーマンスが大幅に向上し、通常のモデル トレーニングに簡単に展開できます。広範な実験により、LA3 が CIFAR-10 および CIFAR-100 で既存の方法に匹敵する、またはそれを上回る優れたパフォーマンスを達成し、自動拡張方法の中で ResNet-50 で 79.97% の新しい最先端の ImageNet 精度を達成することが実証されています。低計算コスト。
Automated augmentation is an emerging and effective technique to search for data augmentation policies to improve generalizability of deep neural network training. Most existing work focuses on constructing a unified policy applicable to all data samples in a given dataset, without considering sample or class variations. In this paper, we propose a novel two-stage data augmentation algorithm, named Label-Aware AutoAugment (LA3), which takes advantage of the label information, and learns augmentation policies separately for samples of different labels. LA3 consists of two learning stages, where in the first stage, individual augmentation methods are evaluated and ranked for each label via Bayesian Optimization aided by a neural predictor, which allows us to identify effective augmentation techniques for each label under a low search cost. And in the second stage, a composite augmentation policy is constructed out of a selection of effective as well as complementary augmentations, which produces significant performance boost and can be easily deployed in typical model training. Extensive experiments demonstrate that LA3 achieves excellent performance matching or surpassing existing methods on CIFAR-10 and CIFAR-100, and achieves a new state-of-the-art ImageNet accuracy of 79.97% on ResNet-50 among auto-augmentation methods, while maintaining a low computational cost.
updated: Thu Apr 20 2023 13:42:18 GMT+0000 (UTC)
published: Thu Apr 20 2023 13:42:18 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アソシエイト