SLACK: Stable Learning of Augmentations with Cold-start and KL regularization

Juliette Marrie, Michael Arbel, Diane Larlus +1

Data augmentation is known to improve the generalization capabilities of neural networks, provided that the set of transformations is chosen with care, a selection often performed…