arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2211.04625cs.CV

用于图像分类的软增强

Soft Augmentation for Image Classification

Argo AI

Yang Liu, Shen Yan, Laura Leal-Taixé, James Hays, Deva Ramanan

AI总结:

本文提出软增强方法,使学习目标随图像变换强度非线性软化,从而支持更激进增强并提升准确率、遮挡鲁棒性与校准性。

摘要:

现代神经网络参数过多,因此依赖数据增强和权重衰减等强正则化来减少过拟合并提升泛化能力。数据增强的主流形式采用不变变换,即样本的学习目标对于应用于该样本的变换保持不变。我们从人类视觉分类研究中汲取灵感,提出将不变变换增强推广为软增强:学习目标会根据应用于样本的变换程度进行非线性软化,例如,更激进的图像裁剪增强会产生置信度更低的学习目标。我们证明,软目标允许使用更激进的数据增强,提供更稳健的性能提升,可与其他增强策略配合,并且有趣的是,能够产生校准更好的模型(因为模型经过训练,对被激进裁剪/遮挡的样本不那么自信)。结合现有的激进增强策略,软目标:1)在 Cifar-10、Cifar-100、ImageNet-1K 和 ImageNet-V2 上使 top-1 准确率提升翻倍,2)将模型的遮挡性能提升最高至 4 倍,3)使期望校准误差(ECE)减半。最后,我们表明软增强可推广到自监督分类任务。代码可在 https://github.com/youngleox/soft_augmentation 获取。

英文摘要:

Modern neural networks are over-parameterized and thus rely on strong regularization such as data augmentation and weight decay to reduce overfitting and improve generalization. The dominant form of data augmentation applies invariant transforms, where the learning target of a sample is invariant to the transform applied to that sample. We draw inspiration from human visual classification studies and propose generalizing augmentation with invariant transforms to soft augmentation where the learning target softens non-linearly as a function of the degree of the transform applied to the sample: e.g., more aggressive image crop augmentations produce less confident learning targets. We demonstrate that soft targets allow for more aggressive data augmentation, offer more robust performance boosts, work with other augmentation policies, and interestingly, produce better calibrated models (since they are trained to be less confident on aggressively cropped/occluded examples). Combined with existing aggressive augmentation strategies, soft target 1) doubles the top-1 accuracy boost across Cifar-10, Cifar-100, ImageNet-1K, and ImageNet-V2, 2) improves model occlusion performance by up to $4\times$, and 3) halves the expected calibration error (ECE). Finally, we show that soft augmentation generalizes to self-supervised classification tasks. Code available at https://github.com/youngleox/soft_augmentation

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

↑