Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Boosting Discriminative Visual Representation Learning with Scenario-Agnostic Mixup

About

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.

Siyuan Li, Zicheng Liu, Zedong Wang, Di Wu, Zihan Liu, Stan Z. Li• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionPASCAL VOC 2007 (test)
mAP57.8
821
Object DetectionMS-COCO 2017 (val)--
237
Image ClassificationFGVC Aircraft
Top-1 Accuracy86.8
185
Image ClassificationTiny-ImageNet
Top-1 Accuracy72.18
143
Image ClassificationSTL-10
Top-1 Accuracy92.45
128
Image ClassificationiNaturalist 2018 (val)
Top-1 Accuracy70.56
116
Image ClassificationCUB-200
Accuracy86.83
92
Image ClassificationPlaces205 (val)
Top-1 Accuracy64.35
68
Linear ClassificationImageNet-1K 1.0 (val)
Top-1 Accuracy73.8
48
Image ClassificationImageNet-1K linear evaluation 1.0
Top-1 Acc68.86
40
Showing 10 of 12 rows

Other info

Code

Follow for update