Diffusemix: Label-Preserving Data Augmentation with Diffusion Models
Khawar Islam, Muhammad Zaigham Zaheer, Arif Mahmood, Karthik Nandakumar
Abstract
Target (T) CutMix (S+T) Mixup (S+T) GridMix (S+T) ResizeMix (S+T) DiffuseMix (S) AugMix (T) PixMix (T) IPMix (T) PuzzleMix (S+T) DiffuseMix (S) SmoothMix (S+T) DiffuseMix (T) AdaAutoMix (S+T) DiffuseMix (T) Figure 1. Top row: existing mixup methods interpolate two different training images [22, 49]. Bottom row: label-preserving methods. For each input image, DIFFUSEMIX employs conditional prompts to obtain generated images. The input image is then concatenated with a generated image to obtain a hybrid image. Each hybrid image is blended with a random fractal to obtain the final training image.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 265f5240-6a7d-4456-b681-2990e665be50Cited by top-tier papers25
- Distribution-Aware Data Expansion with Diffusion ModelsHaowei Zhu, Ling Yang, Jun-Hai Yong, Hongzhi Yin et al.NeurIPS 2024 · 27 citations
- MergeMix: A Unified Augmentation Paradigm for Visual and Multi-Modal UnderstandingXin Jin, Siyuan Li, Siyong Jian, Kai Yu et al.ICLR 2026 · 14 citations
- Just Add $100 More: Augmenting Pseudo-LiDAR Point Cloud for Resolving Class-imbalance ProblemMincheol Chang, Siyeong Lee, Jinkyu Kim, Namil KimNeurIPS 2024 · 4 citations
- Diffusion Guided Adaptive Augmentation for Generalization in Visual Reinforcement LearningJeong Woon Lee, Hyoseok HwangICCV 2025 · 3 citations
- Generate, Refine, and Encode: Leveraging Synthesized Novel Samples for On-the-Fly Fine-Grained Category DiscoveryXiao Liu, Nan Pu, Haiyang Zheng, Wenjing Li et al.ICCV 2025 · 3 citations
Builds on22
- Adding Conditional Control to Text-to-Image Diffusion ModelsLvmin Zhang, Anyi Rao, Maneesh AgrawalaICCV 2023 · 6,759 citations
- CutMix: Regularization Strategy to Train Strong Classifiers With Localizable FeaturesSangdoo Yun, Dongyoon Han, Sanghyuk Chun, Seong Joon Oh et al.ICCV 2019 · 5,843 citations
- RandAugment: Practical Automated Data Augmentation with a Reduced Search SpaceEkin Dogus Cubuk, Barret Zoph, Jonathon Shlens, Quoc LeNeurIPS 2020 · 4,453 citations
- Palette: Image-to-Image Diffusion ModelsChitwan Saharia, William Chan, Huiwen Chang, Chris A. Lee et al.SIGGRAPH 2022 · 1,638 citations
- AugMix: A Simple Data Processing Method to Improve Robustness and UncertaintyDan Hendrycks, Norman Mu, Ekin Dogus Cubuk, Barret Zoph et al.ICLR 2020 · 1,572 citations
Related papers
- TdAttenMix: Top-Down Attention Guided MixupZhiming Wang, Lin Gu, Feng LuAAAI 2025 · 3 citations
- ReMix: Towards Image-to-Image Translation With Limited DataJie Cao, Luanxuan Hou, Ming-Hsuan Yang, Ran He et al.CVPR 2021
- StyleMix: Separating Content and Style for Enhanced Data AugmentationMinui Hong, Jinwoo Choi, Gunhee KimCVPR 2021
- GenLabel: Mixup Relabeling using Generative ModelsJy-yong Sohn, Liang Shang, Hongxu Chen, Jaekyun Moon et al.ICML 2022 · 15 citations
- Conditional Score Guidance for Text-Driven Image-to-Image TranslationHyunsoo Lee, Minsoo Kang, Bohyung HanNeurIPS 2023 · 23 citations
