A simple way to make neural networks robust against diverse image corruptions
About
The human visual system is remarkably robust against a wide range of naturally occurring variations and corruptions like rain or snow. In contrast, the performance of modern image recognition models strongly degrades when evaluated on previously unseen corruptions. Here, we demonstrate that a simple but properly tuned training with additive Gaussian and Speckle noise generalizes surprisingly well to unseen corruptions, easily reaching the previous state of the art on the corruption benchmark ImageNet-C (with ResNet50) and on MNIST-C. We build on top of these strong baseline results and show that an adversarial training of the recognition model against uncorrelated worst-case noise distributions leads to an additional increase in performance. This regularization can be combined with previously proposed defense methods for further improvement.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | ImageNet 1k (test) | Top-1 Accuracy76.1 | 798 | |
| Image Classification | ImageNet-A (test) | Top-1 Acc1.1 | 154 | |
| Image Classification | ImageNet-C (test) | -- | 110 | |
| Image Classification | ImageNet-R (test) | -- | 105 | |
| Robustness to Corruptions | ImageNet-C (test) | mCE63 | 56 | |
| Image Classification | ImageNet-SK (test) | Top-1 Accuracy26.3 | 34 | |
| Adversarial Robustness | ImageNet 1k (test) | FGSM Robustness17.8 | 34 | |
| Image Classification | ImageNet-200 (test) | Top-1 Error Rate8.1 | 28 | |
| Image Classification | Real Blurry Images (test) | Error Rate56.9 | 10 | |
| Image Classification | Tiny ImageNet-C noise-free (test) | Relative Accuracy35.3 | 4 |