十分かつ多様なトレーニングサンプルを作成することを目指して、データ増強は、ディープモデルのトレーニングでの有効性が実証されています。最高の増強の基準を定義するのは難しいことに関して、本稿では、DeepAugNetと呼ばれる新しい学習ベースの増強方法を紹介します。これは、最終的に増強されたデータをいくつかの順次増強されたサブセットの集合として定式化します。具体的には、現在の拡張サブセットは、深層強化学習を使用して決定論的な拡張ポリシーを学習することにより、最後の拡張サブセットと比較してパフォーマンスの改善を最大化するために必要です。統一された最適化の目標を導入することにより、DeepAugNetは、データ増強とディープモデルトレーニングを、ディープQ学習アルゴリズムと代理ディープモデルの決闘のハイブリッドアーキテクチャを同時にトレーニングすることで実現されるエンドツーエンドのトレーニング方法で組み合わせる予定です。 Fashion MNIST、CUB、CIFAR-100、WebCaricatureなどのさまざまなベンチマークデータセットで、提案されているDeepAugNetを広範囲に評価しました。現在の最先端技術と比較して、この方法は小規模データセットで大幅な改善を達成し、大規模データセットで同等のパフォーマンスを達成できます。コードはまもなく利用可能になります。
Aiming to produce sufficient and diverse training samples, data augmentation has been demonstrated for its effectiveness in training deep models. Regarding that the criterion of the best augmentation is challenging to define, we in this paper present a novel learning-based augmentation method termed as DeepAugNet, which formulates the final augmented data as a collection of several sequentially augmented subsets. Specifically, the current augmented subset is required to maximize the performance improvement compared with the last augmented subset by learning the deterministic augmentation policy using deep reinforcement learning. By introducing an unified optimization goal, DeepAugNet intends to combine the data augmentation and the deep model training in an end-to-end training manner which is realized by simultaneously training a hybrid architecture of dueling deep Q-learning algorithm and a surrogate deep model. We extensively evaluated our proposed DeepAugNet on various benchmark datasets including Fashion MNIST, CUB, CIFAR-100 and WebCaricature. Compared with the current state-of-the-arts, our method can achieve a significant improvement in small-scale datasets, and a comparable performance in large-scale datasets. Code will be available soon.