Boosting Discriminative Visual Representation Learning with Scenario-Agnostic Mixup

30 Nov 2021  ·  Siyuan Li, Zicheng Liu, Zedong Wang, Di wu, Zihan Liu, Stan Z. Li ·

Mixup is a well-known data-dependent augmentation technique for DNNs, consisting of two sub-tasks: mixup generation and classification. However, the recent dominant online training method confines mixup to supervised learning (SL), and the objective of the generation sub-task is limited to selected sample pairs instead of the whole data manifold, which might cause trivial solutions. To overcome such limitations, we comprehensively study the objective of mixup generation and propose \textbf{S}cenario-\textbf{A}gnostic \textbf{Mix}up (SAMix) for both SL and Self-supervised Learning (SSL) scenarios. Specifically, we hypothesize and verify the objective function of mixup generation as optimizing local smoothness between two mixed classes subject to global discrimination from other classes. Accordingly, we propose $\eta$-balanced mixup loss for complementary learning of the two sub-objectives. Meanwhile, a label-free generation sub-network is designed, which effectively provides non-trivial mixup samples and improves transferable abilities. Moreover, to reduce the computational cost of online training, we further introduce a pre-trained version, SAMix$^\mathcal{P}$, achieving more favorable efficiency and generalizability. Extensive experiments on nine SL and SSL benchmarks demonstrate the consistent superiority and versatility of SAMix compared with existing methods.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Image Classification CIFAR-100 WRN-28-8 +SAMix Percentage correct 85.50 # 62
Image Classification CIFAR-100 ResNeXt-50(32x4d) + SAMix Percentage correct 84.42 # 76
Image Classification ImageNet ResNet-18 (SAMix) Top 1 Accuracy 72.33% # 925
Number of params 11.7M # 492
Image Classification ImageNet ResNet-50 (SAMix) Top 1 Accuracy 79.41% # 694
Number of params 25.6M # 601
Image Classification ImageNet ResNet-101 (SAMix) Top 1 Accuracy 81.08% # 613
Number of params 44.6M # 703
Image Classification ImageNet ResNet-34 (SAMix) Top 1 Accuracy 76.35% # 846
Number of params 21.8M # 553
Image Classification iNaturalist 2018 ResNet-50 (SAMix) Top-1 Accuracy 64.84% # 43
Image Classification iNaturalist 2018 ResNeXt-101 (SAMix) Top-1 Accuracy 70.54% # 31
Image Classification Places205 SAMix (ResNet-50 Supervised) Top 1 Accuracy 64.3 # 7
Image Classification Tiny ImageNet Classification ResNeXt-50 (SAMix) Validation Acc 72.18% # 12
Image Classification Tiny ImageNet Classification ResNet18 (SAMix) Validation Acc 68.89% # 16

Methods