Puzzle Mix: Exploiting Saliency and Local Statistics for Optimal Mixup
Not on the map. Why?
Abstract (arXiv)
While deep neural networks achieve great performance on fitting the training distribution, the learned networks are prone to overfitting and are susceptible to adversarial attacks. In this regard, a number of mixup based augmentation methods have been recently proposed. However, these approaches mainly focus on creating previously unseen virtual examples and can sometimes provide misleading supervisory signal to the network. To this end, we propose Puzzle Mix, a mixup method for explicitly utilizing the saliency information and the underlying statistics of the natural examples. This leads to an interesting optimization problem alternating between the multi-label objective for optimal mixing mask and saliency discounted optimal transport objective. Our experiments show Puzzle Mix achieves the state of the art generalization and the adversarial robustness results compared to other mixup methods on CIFAR-100, Tiny-ImageNet, and ImageNet datasets. The source code is available at https://github.com/snu-mllab/PuzzleMix.
Key facts. Cited by 260 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Sep 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 41; 2022: 62; 2023: 58; 2024: 58; 2025: 27; 2026: 14.
260 papers.
Sort:
By area:
- 2021Rainbow Memory: Continual Learning with a Memory of Diverse Samplescited by 273
- 2022Dataset Condensation via Efficient Synthetic-Data Parameterizationcited by 187
- 2022LIFT: Language-Interfaced Fine-Tuning for Non-Language Machine Learning Taskscited by 146
- 2021Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversitycited by 127
- 2021Mix3D: Out-of-Context Data Augmentation for 3D Scenescited by 126
- 2021Interpretable Deep Learning: Interpretation, Interpretability, Trustworthiness, and Beyondcited by 116
- 2023Diversify Your Vision Datasets with Automatic Diffusion-Based Augmentationcited by 100
- 2022A Comprehensive Survey of Image Augmentation Techniques for Deep Learningcited by 95
- 2023MedViT: A Robust Vision Transformer for Generalized Medical Image Classificationcited by 89
- 2021TransMix: Attend to Mix for Vision Transformerscited by 79
- 2024DiffuseMix: Label-Preserving Data Augmentation with Diffusion Modelscited by 78
- 2021SSUL: Semantic Segmentation with Unknown Label for Exemplar-based Class-Incremental Learningcited by 71
38 more
- 2021AutoMix: Unveiling the Power of Mixup for Stronger Classifierscited by 69
- 2022CycleMix: A Holistic Strategy for Medical Image Segmentation from Scribble Supervisioncited by 64
- 2022MixGen: A New Multi-Modal Data Augmentationcited by 61
- 2021Regularization Strategy for Point Cloud via Rigidly Mixed Samplecited by 59
- 2021When and How Mixup Improves Calibrationcited by 59
- 2022C-Mixup: Improving Generalization in Regressioncited by 57
- 2021SmoothMix: Training Confidence-calibrated Smoothed Classifiers for Certified Robustnesscited by 49
- 2021Meta-FDMixup: Cross-Domain Few-Shot Learning Guided by Labeled Target Datacited by 48
- 2021A Systematic Review of Robustness in Deep Learning for Computer Vision: Mind the gap?cited by 48
- 2022TokenMix: Rethinking Image Mixing for Data Augmentation in Vision Transformerscited by 48
- 2023Credal Bayesian Deep Learningcited by 45
- 2022Rethinking Data Augmentation for Single-source Domain Generalization in Medical Image Segmentationcited by 44
- 2021SSMix: Saliency-Based Span Mixup for Text Classificationcited by 39
- 2024A Comprehensive Survey on Data Augmentationcited by 38
- 2021VideoLT: Large-scale Long-tailed Video Recognitioncited by 35
- 2021Object-aware Contrastive Learning for Debiased Scene Representationcited by 34
- 2021AlignMixup: Improving Representations By Interpolating Aligned Featurescited by 33
- 2023ExcelFormer: A neural network surpassing GBDTs on tabular datacited by 33
- 2021MixMo: Mixing Multiple Inputs for Multiple Outputs via Deep Subnetworkscited by 32
- 2021Graph Transplant: Node Saliency-Guided Graph Mixup with Local Structure Preservationcited by 30
- 2022A Survey of Mix-based Data Augmentation: Taxonomy, Methods, Applications, and Explainabilitycited by 30
- 2022Semantic-Aware Representation Blending for Multi-Label Image Recognition with Partial Labelscited by 29
- 2022Geodesic Multi-Modal Mixup for Robust Fine-Tuningcited by 29
- 2022SelecMix: Debiased Learning by Contradicting-pair Samplingcited by 29
- 2021Noisy Feature Mixupcited by 27
- 2022A Simple Data Mixing Prior for Improving Self-Supervised Learningcited by 27
- 2022A Unified Analysis of Mixed Sample Data Augmentation: A Loss Function Perspectivecited by 27
- 2022TokenMixup: Efficient Attention-guided Token-level Data Augmentation for Transformerscited by 26
- 2021Survey: Image Mixing and Deleting for Data Augmentationcited by 25
- 2024ScribFormer: Transformer Makes CNN Work Better for Scribble-based Medical Image Segmentationcited by 24
- 2021It Takes Two to Tango: Mixup for Deep Metric Learningcited by 23
- 2023Image Data Augmentation Approaches: A Comprehensive Survey and Future directionscited by 23
- 2023GuidedMixup: An Efficient Mixup Strategy Guided by Saliency Mapscited by 23
- 2023ScribbleVC: Scribble-supervised Medical Image Segmentation with Vision-Class Embeddingcited by 23
- 2021Learning Representational Invariances for Data-Efficient Action Recognitioncited by 22
- 2021Boosting Discriminative Visual Representation Learning with Scenario-Agnostic Mixupcited by 22
- 2021MUM : Mix Image Tiles and UnMix Feature Tiles for Semi-Supervised Object Detectioncited by 21
- 2022Model-Agnostic Augmentation for Accurate Graph Classificationcited by 21
50 most cited of 260 All 260 (JSON)
- 2026OFBD: Object-Focused Background Debiasing for Long-Tailed Learning29 Sep 2026
- 2026SlideMix: Enhancing Whole Slide Image Analysis via Multimodal Shuffling31 Aug 2026
- 2026DisMix: Order-Aware Mixup for Medical Imaging via Disentangling Ordinal and Non-Ordinal Features5 Aug 2026
- 2026InstructMixup: Instruction-Guided Salient Patch Editing for Robust Data Augmentation21 Jul 2026
- 2026AGVBench: A Reliability-Oriented Benchmark of Data Augmentation for Vein Recognition2 Jul 2026
- 2026S2-FracMix: Label-Preserving Self-Saliency Mixup Augmentation24 Jun 2026
- 2026SNR-ST-Mix: Sample-specific Neighborhood Regression Mixup for Augmented Spatial Transcriptomics Imputation with Deep Neural Network7 Jun 2026
- 2026OA-CutMix: Correcting the Label Bias of CutMix3 Jun 2026
- 2026ZScribbleSeg: A comprehensive segmentation framework with modeling of efficient annotation and maximization of scribble supervision7 May 2026
- 2026Learning from Synthetic Data via Provenance-Based Input Gradient Guidance3 Apr 2026
- 2026PLESS: Pseudo-Label Enhancement with Spreading Scribbles for Weakly Supervised Segmentation12 Feb 2026
- 2026NeuroShield: A Neuro-Symbolic Framework for Adversarial Robustness19 Jan 2026
38 more
- 2026oculomix: Hierarchical Sampling for Retinal-Based Systemic Disease Prediction16 Jan 2026
- 2026Unsupervised Modular Adaptive Region Growing and RegionMix Classification for Wind Turbine Segmentation7 Jan 2026
- 2025Credal and Interval Deep Evidential Classifications5 Dec 2025
- 2025Local and Global Context-and-Object-part-Aware Superpixel-based Data Augmentation for Deep Visual Recognition28 Nov 2025
- 2025Diffusion-Guided Mask-Consistent Paired Mixing for Endoscopic Image Segmentation5 Nov 2025
- 2025MergeMix: A Unified Augmentation Paradigm for Visual and Multi-Modal Understanding27 Oct 2025
- 2025MS-Mix: Sentiment-Guided Adaptive Augmentation for Multimodal Sentiment Analysis13 Oct 2025
- 2025On-the-Fly Data Augmentation via Gradient-Guided and Sample-Aware Influence Estimation1 Oct 2025
- 2025IPF-RDA: An Information-Preserving Framework for Robust Data Augmentation20 Sep 2025
- 2025SelectMix: Enhancing Label Noise Robustness through Targeted Sample Mixing14 Sep 2025
- 2025PQ-DAF: Pose-driven Quality-controlled Data Augmentation for Data-scarce Driver Distraction Detection14 Aug 2025
- 2025The Power of Many: Synergistic Unification of Diverse Augmentations for Efficient Adversarial Robustness5 Aug 2025
- 2025TABNet: A Triplet Augmentation Self-Recovery Framework with Boundary-Aware Pseudo-Labels for Medical Image Segmentation3 Jul 2025
- 2025Multi-Source COVID-19 Detection via Variance Risk Extrapolation29 Jun 2025
- 2025Topology-Aware Modeling for Unsupervised Simulation-to-Reality Point Cloud Recognition26 Jun 2025
- 2025Enhancing Ambiguous Dynamic Facial Expression Recognition with Soft Label-based Data Augmentation25 Jun 2025
- 2025Scan-and-Print: Patch-level Data Summarization and Augmentation for Content-aware Layout Generation in Poster Design27 May 2025
- 2025SGD-Mix: Enhancing Domain-Specific Image Classification with Label-Preserving Data Augmentation17 May 2025
- 2025Language-Driven Dual Style Mixing for Single-Domain Generalized Object Detection12 May 2025
- 2025Beyond One-Hot Labels: Semantic Mixing for Model Calibration18 Apr 2025
- 2025MDIT: A Model-free Data Interpolation Method for Diverse Instruction Tuning9 Apr 2025
- 2025MedCL: Learning Consistent Anatomy Distribution for Scribble-supervised Medical Image Segmentation28 Mar 2025
- 2025CAARMA: Class Augmentation with Adversarial Mixup Regularization20 Mar 2025
- 2025Narrowing Class-Wise Robustness Gaps in Adversarial Training20 Mar 2025
- 2025Context-guided Responsible Data Augmentation with Diffusion Models12 Mar 2025
- 2025Class-Aware PillarMix: Can Mixed Sample Data Augmentation Enhance 3D Object Detection with Radar Point Clouds?4 Mar 2025
- 2025MIDAS: Mixing Ambiguous Data with Soft Labels for Dynamic Facial Expression Recognition28 Feb 2025
- 2025QMaxViT-Unet+: A Query-Based MaxViT-Unet with Edge Enhancement for Scribble-Supervised Segmentation of Medical Images14 Feb 2025
- 2025LayerMix: Enhanced Data Augmentation through Fractal Integration for Robust Deep Learning8 Jan 2025
- 2024Lungmix: A Mixup-Based Strategy for Generalization in Respiratory Sound Classification29 Dec 2024
- 2024HELPNet: Hierarchical Perturbations Consistency and Entropy-guided Ensemble for Scribble Supervised Medical Image Segmentation25 Dec 2024
- 2024RESQUE: Quantifying Estimator to Task and Distribution Shift for Sustainable Model Reusability20 Dec 2024
- 2024GenMix: Effective Data Augmentation with Generative Diffusion Model Image Editing3 Dec 2024
- 2024BGM: Background Mixup for X-ray Prohibited Items Detection30 Nov 2024
- 2024ScribbleVS: Scribble-Supervised Medical Image Segmentation via Dynamic Competitive Pseudo Label Selection15 Nov 2024
- 2024Mix from Failure: Confusion-Pairing Mixup for Long-Tailed Recognition12 Nov 2024
- 2024An Open API Architecture to Discover the Trustworthy Explanation of Cloud AI Services5 Nov 2024
- 2024Provable Benefit of Cutout and CutMix for Feature Learning31 Oct 2024
50 newest of 260 All 260 (JSON)
- 2024DiffuseMix: Label-Preserving Data Augmentation with Diffusion Modelscited by 78
- 2024GenMix: Effective Data Augmentation with Generative Diffusion Model Image Editingcited by 14
- 2025Context-guided Responsible Data Augmentation with Diffusion Modelscited by 9
- 2023GeNIe: Generative Hard Negative Images Through Diffusioncited by 8
- 2025SGD-Mix: Enhancing Domain-Specific Image Classification with Label-Preserving Data Augmentationcited by 3
- 2025SGD-Mix: Enhancing Domain-Specific Image Classification with Label-Preserving Data Augmentation17 May 2025
- 2025Context-guided Responsible Data Augmentation with Diffusion Models12 Mar 2025
- 2024GenMix: Effective Data Augmentation with Generative Diffusion Model Image Editing3 Dec 2024
- 2024DiffuseMix: Label-Preserving Data Augmentation with Diffusion Models5 Apr 2024
- 2023GeNIe: Generative Hard Negative Images Through Diffusion5 Dec 2023
- 2023Revisiting Single-gated Mixtures of Expertscited by 10
- 2023Revisiting Single-gated Mixtures of Experts11 Apr 2023
- 2021Rainbow Memory: Continual Learning with a Memory of Diverse Samplescited by 273
- 2022Dataset Condensation via Efficient Synthetic-Data Parameterizationcited by 187
- 2022LIFT: Language-Interfaced Fine-Tuning for Non-Language Machine Learning Taskscited by 146
- 2021Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversitycited by 127
- 2021Mix3D: Out-of-Context Data Augmentation for 3D Scenescited by 126
- 2021Interpretable Deep Learning: Interpretation, Interpretability, Trustworthiness, and Beyondcited by 116
- 2023Diversify Your Vision Datasets with Automatic Diffusion-Based Augmentationcited by 100
- 2022A Comprehensive Survey of Image Augmentation Techniques for Deep Learningcited by 95
- 2023MedViT: A Robust Vision Transformer for Generalized Medical Image Classificationcited by 89
- 2021TransMix: Attend to Mix for Vision Transformerscited by 79
- 2021SSUL: Semantic Segmentation with Unknown Label for Exemplar-based Class-Incremental Learningcited by 71
- 2021AutoMix: Unveiling the Power of Mixup for Stronger Classifierscited by 69
12 most cited of 252 All 252 (JSON)
- 2026OFBD: Object-Focused Background Debiasing for Long-Tailed Learning29 Sep 2026
- 2026SlideMix: Enhancing Whole Slide Image Analysis via Multimodal Shuffling31 Aug 2026
- 2026DisMix: Order-Aware Mixup for Medical Imaging via Disentangling Ordinal and Non-Ordinal Features5 Aug 2026
- 2026InstructMixup: Instruction-Guided Salient Patch Editing for Robust Data Augmentation21 Jul 2026
- 2026AGVBench: A Reliability-Oriented Benchmark of Data Augmentation for Vein Recognition2 Jul 2026
- 2026S2-FracMix: Label-Preserving Self-Saliency Mixup Augmentation24 Jun 2026
- 2026SNR-ST-Mix: Sample-specific Neighborhood Regression Mixup for Augmented Spatial Transcriptomics Imputation with Deep Neural Network7 Jun 2026
- 2026OA-CutMix: Correcting the Label Bias of CutMix3 Jun 2026
- 2026ZScribbleSeg: A comprehensive segmentation framework with modeling of efficient annotation and maximization of scribble supervision7 May 2026
- 2026Learning from Synthetic Data via Provenance-Based Input Gradient Guidance3 Apr 2026
- 2026PLESS: Pseudo-Label Enhancement with Spreading Scribbles for Weakly Supervised Segmentation12 Feb 2026
- 2026NeuroShield: A Neuro-Symbolic Framework for Adversarial Robustness19 Jan 2026
12 newest of 252 All 252 (JSON)