arXiv reaDer
マルチラベル分類のためのきめ細かい自動拡張
Fine-Grained AutoAugmentation for Multi-label Classification
データ拡張は、深層学習モデルの一般化を改善するために一般的に使用されるアプローチです。最近の研究によると、学習したデータ拡張ポリシーは、手作りのポリシーよりも優れた一般化を実現できます。ただし、これらの作業のほとんどは、データセット内のすべてのサンプルに統一された拡張ポリシーを使用します。これは、マルチラベル分類タスクのすべてのラベルに必ずしも有益ではないことが観察されます。つまり、一部のポリシーは、他のラベルに利益をもたらす一方で、一部のラベルに悪影響を与える可能性があります。この問題に取り組むために、我々は、マルチラベルシナリオのための新しいラベルベースの自動拡張(LB-Aug)方法を提案します。この方法では、拡張ポリシーネットワークによってラベルに関して拡張ポリシーが生成されます。ポリシーは、ポリシー勾配法を使用した強化学習によって学習され、インスタンスラベルから最適な拡張ポリシーへのマッピングを提供します。数値実験は、私たちのLB-Augが、画像とビデオの分類に関する複数のベンチマークで、以前の最先端の拡張方法を大幅に上回っていることを示しています。
Data augmentation is a commonly used approach to improving the generalization of deep learning models. Recent works show that learned data augmentation policies can achieve better generalization than hand-crafted ones. However, most of these works use unified augmentation policies for all samples in a dataset, which is observed not necessarily beneficial for all labels in multi-label classification tasks, i.e., some policies may have negative impacts on some labels while benefitting the others. To tackle this problem, we propose a novel Label-Based AutoAugmentation (LB-Aug) method for multi-label scenarios, where augmentation policies are generated with respect to labels by an augmentation-policy network. The policies are learned via reinforcement learning using policy gradient methods, providing a mapping from instance labels to their optimal augmentation policies. Numerical experiments show that our LB-Aug outperforms previous state-of-the-art augmentation methods by large margins in multiple benchmarks on image and video classification.
updated: Mon Jul 12 2021 12:47:16 GMT+0000 (UTC)
published: Mon Jul 12 2021 12:47:16 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アソシエイト