PRIME: A few primitives can boost robustness to common corruptions

Despite their impressive performance on image classification tasks, deep networks have a hard time generalizing to unforeseen corruptions of their data. To fix this vulnerability, prior works have built complex data augmentation strategies, combining multiple methods to enrich the training data. However, introducing intricate design choices or heuristics makes it hard to understand which elements of these methods are indeed crucial for improving robustness. In this work, we take a step back and follow a principled approach to achieve robustness to common corruptions. We propose PRIME, a general data augmentation scheme that relies on simple yet rich families of max-entropy image transformations. PRIME outperforms the prior art in terms of corruption robustness, while its simplicity and plug-and-play nature enable combination with other methods to further boost their robustness. We analyze PRIME to shed light on the importance of the mixing strategy on synthesizing corrupted images, and to reveal the robustness-accuracy trade-offs arising in the context of common corruptions. Finally, we show that the computational efficiency of our method allows it to be easily used in both on-line and off-line data augmentation schemes.

PDF Abstract

Results from the Paper

Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Domain Generalization ImageNet-C PRIME + DeepAugment (ResNet-50) mean Corruption Error (mCE) 51.3 # 28
Top 1 Accuracy 59.9 # 5
Domain Generalization ImageNet-C PRIME with JSD (ResNet-50) mean Corruption Error (mCE) 55.5 # 31
Top 1 Accuracy 56.4 # 7
Domain Generalization ImageNet-C PRIME (ResNet-50) mean Corruption Error (mCE) 57.5 # 33
Top 1 Accuracy 55.0 # 8
Domain Generalization ImageNet-R PRIME with JSD (ResNet-50) Top-1 Error Rate 53.7 # 29
Domain Generalization ImageNet-R PRIME (ResNet-50) Top-1 Error Rate 57.1 # 31


No methods listed for this paper. Add relevant methods here