arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.08693cs.LGcs.CV
收藏

ParticleAugment:基于采样的数据增强

ParticleAugment: Sampling-Based Data Augmentation

Ulm University(乌尔姆大学)

Alexander Tsaregorodtsev, Vasileios Belagiannis

摘要:

我们提出了一个自动化的数据增强方法的图像分类。我们制定的问题,蒙特卡罗抽样,我们的目标是近似的最佳增强政策。我们提出了一个粒子滤波方案的政策搜索的概率应用一组增强操作形成的过滤器的状态。我们根据参考模型和实际模型之间的损失函数差异来衡量策略性能,然后使用该差异来重新加权粒子并最终更新策略。在我们的实验中,我们证明了我们的自动增强公式在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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑