arXiv reaDer
ParticleAugment:サンプリングベースのデータ拡張
ParticleAugment: Sampling-Based Data Augmentation
画像分類のための自動化されたデータ拡張アプローチを提示します。問題をモンテカルロサンプリングとして定式化します。ここでの目標は、最適な拡張ポリシーを概算することです。一連の拡張操作を適用する確率がフィルターの状態を形成する、ポリシー検索用の粒子フィルタリングスキームを提案します。参照モデルと実際のモデルの損失関数の差に基づいてポリシーのパフォーマンスを測定します。その後、これを使用してパーティクルの重みを再設定し、最終的にポリシーを更新します。私たちの実験では、自動拡張の定式化が、この問題の標準ネットワークアーキテクチャを使用して、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 scheme for the policy search where the probability of applying a set of augmentation operations forms the state of the filter. We measure the policy performance based on the loss function difference between a reference and the actual model, which we afterwards use to re-weight the particles and finally update the policy. 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, our method reaches a balance between the computational cost of policy search and the model performance. Our code will be made publicly available.
updated: Tue Aug 24 2021 10:58:54 GMT+0000 (UTC)
published: Wed Jun 16 2021 10:56:02 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アソシエイト