AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty
Modern deep neural networks can achieve high accuracy when the training distribution and test distribution are identically distributed, but this assumption is frequently violated in practice. When the train and test distributions are mismatched, accuracy can plummet. Currently there are few techniques that improve robustness to unforeseen data shifts encountered during deployment. In this work, we propose a technique to improve the robustness and uncertainty estimates of image classifiers. We propose AugMix, a data processing technique that is simple to implement, adds limited computational overhead, and helps models withstand unforeseen corruptions. AugMix significantly improves robustness and uncertainty measures on challenging image classification benchmarks, closing the gap between previous methods and the best possible performance in some cases by more than half.
Code (15)
Tasks
Domain Generalizationimage-classificationImage ClassificationOut-of-Distribution GeneralizationRobust Object DetectionMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Fourier-Based Augmentations for Improved Robustness and Uncertainty Calibration
Diverse data augmentation strategies are a natural approach to improving robustness in computer vision models against unforeseen shifts in data distribution. However, the ability to tailor such strategies to inoculate a …
Data AugmentationImage AugmentationImproving robustness against common corruptions by covariate shift adaptation
Today's state-of-the-art machine vision models are vulnerable to image corruptions like blurring or compression artefacts, limiting their performance in many real-world applications. We here argue that popular benchmarks…
Domain GeneralizationImage ClassificationOut-of-Distribution GeneralizationUnsupervised Domain AdaptationDiffAug: A Diffuse-and-Denoise Augmentation for Training Robust Classifiers
We introduce DiffAug, a simple and efficient diffusion-based augmentation technique to train image classifiers for the crucial yet challenging goal of improved classifier robustness. Applying DiffAug to a given example c…
DenoisingImage GenerationOut-of-Distribution DetectionAugMixCloak: A Defense against Membership Inference Attacks via Image Transformation
Traditional machine learning (ML) raises serious privacy concerns, while federated learning (FL) mitigates the risk of data leakage by keeping data on local devices. However, the training process of FL can still leak sen…
Data AugmentationFederated LearningInference AttackMembership Inference AttackAugMax: Adversarial Composition of Random Augmentations for Robust Training
Data augmentation is a simple yet effective way to improve the robustness of deep neural networks (DNNs). Diversity and hardness are two complementary dimensions of data augmentation to achieve robustness. For example, A…
Data AugmentationDiversity