I NSTANCE –S PECIFIC A UGMENTATION : C APTURING L OCAL I NVARIANCES
I NSTANCE –S PECIFIC A UGMENTATION : C APTURING L OCAL I NVARIANCES
复制标题
DOI:
--
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
Ning Miao;Tom Rainforth;Emile Mathieu;Yann Dubois;Y. Teh;Adam Foster;Hyunjik Kim
中科院分区:
文献类型:
--
作者:
Ning Miao;Tom Rainforth;Emile Mathieu;Yann Dubois;Y. Teh;Adam Foster;Hyunjik Kim
We introduce InstaAug, a method for automatically learning input-specific augmentations from data. Previous data augmentation methods have generally assumed independence between the original input and the transformation applied to that input. This can be highly restrictive, as the invariances that the augmentations are based on are themselves often highly input dependent; e.g., we can change a leaf from green to yellow while maintaining its label, but not a lime. InstaAug instead allows for input dependency by introducing an invariance module that maps inputs to tailored transformation distributions. It can be simultaneously trained alongside the downstream model in a fully end-to-end manner, or separately learned for a pre-trained model. We empirically demonstrate that InstaAug learns meaningful input-dependent augmentations for a wide range of transformation classes, which in turn provides better performance on both supervised and self-supervised tasks. Codes are available at https://github.com/NingMiao/InstaAug.