画像分類のための自動化されたデータ拡張アプローチを提示します。問題をモンテカルロサンプリングとして定式化します。ここでの目標は、最適な拡張ポリシーを概算することです。モデルトレーニング中に最適な増強ポリシーとそのスケジュールを見つけるために、粒子フィルタリングの定式化を提案します。ポリシー移行モデルはガウス事前分布とオプションの拡張速度パラメーターに依存しますが、パフォーマンス測定手順はトレーニングセットの検証サブセットに依存します。私たちの実験では、自動拡張の定式化が、この問題の標準ネットワークアーキテクチャを使用して、CIFAR-10、CIFAR-100、およびImageNetデータセットで有望な結果に達することを示しています。関連する作業と比較することにより、私たちの方法がポリシー検索の計算コストとモデルのパフォーマンスのバランスに達していることも示しています。
We present an automated data augmentation approach for image classification. We formulate the problem as Monte Carlo sampling where our goal is to approximate the optimal augmentation policies. We propose a particle filtering formulation to find optimal augmentation policies and their schedules during model training. Our performance measurement procedure relies on a validation subset of our training set, while the policy transition model depends on a Gaussian prior and an optional augmentation velocity parameter. In our experiments, we show that our formulation for automated augmentation reaches promising results on CIFAR-10, CIFAR-100, and ImageNet datasets using the standard network architectures for this problem. By comparing with the related work, we also show that our method reaches a balance between the computational cost of policy search and the model performance.