Toward Robustness in Multi-label Classification: A Data Augmentation Strategy against Imbalance and Noise
Not on the map. Why?
Abstract (arXiv)
Multi-label classification poses challenges due to imbalanced and noisy labels in training data. We propose a unified data augmentation method, named BalanceMix, to address these challenges. Our approach includes two samplers for imbalanced labels, generating minority-augmented instances with high diversity. It also refines multi-labels at the label-wise granularity, categorizing noisy labels as clean, re-labeled, or ambiguous for robust optimization. Extensive experiments on three benchmark datasets demonstrate that BalanceMix outperforms existing state-of-the-art methods. We release the code at https://github.com/DISL-Lab/BalanceMix.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 23 of its 43 references that are in the map. First posted on arXiv 12 Dec 2023.
Builds on
23 of 43 references.
References · 23
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,053
- 2020DivideMix: Learning with Noisy Labels as Semi-supervised Learningcited by 691
- 2020Asymmetric Loss For Multi-Label Classificationcited by 379
- 2021Distribution Alignment: A Unified Framework for Long-tail Visual Recognitioncited by 198
- 2020General Multi-label Image Classification with Transformerscited by 126
- 2021Query2Label: A Simple Transformer Way to Multi-Label Classificationcited by 126
- 2019Learning a Deep ConvNet for Multi-label Classification with Partial Labelscited by 124
- 2020Distribution-Balanced Loss for Multi-Label Classification in Long-Tailed Datasetscited by 122
- 2021Data Collection and Quality Challenges in Deep Learning: A Data-Centric AI Perspectivecited by 116
11 more
- 2021Understanding and Improving Early Stopping for Learning with Noisy Labelscited by 112
- 2021The Majority Can Help The Minority: Context-rich Minority Oversampling for Long-tailed Classificationcited by 100
- 2020Beyond Class-Conditional Assumption: A Primary Attempt to Combat Instance-Dependent Label Noisecited by 82
- 2021Multi-Label Learning from Single Positive Labelscited by 81
- 2021ML-Decoder: Scalable and Versatile Classification Headcited by 78
- 2021Balanced-MixUp for Highly Imbalanced Medical Image Classificationcited by 46
- 2021Robust Long-Tailed Learning under Label Noisecited by 35
- 2023Fine-Grained Classification with Noisy Labelscited by 15
- 2021Evaluating Multi-label Classifiers with Noisy Labelscited by 9
- 2023SuperDisco: Super-Class Discovery Improves Visual Recognition for the Long-Tailcited by 9
- 2022Multi-class Label Noise Learning via Loss Decomposition and Centroid Estimationcited by 3
Built on by
Citing papers per year: 2024: 4; 2025: 3; 2026: 4.
11 papers.
- 2025Curiosity Meets Cooperation: A Game-Theoretic Approach to Long-Tail Multi-Label Learningcited by 4
- 2024Ensemble Methods for Sequence Classification with Hidden Markov Modelscited by 1
- 2024A Bilevel Optimization Framework for Imbalanced Data Classificationcited by 1
- 2026Model-Agnostic Meta Learning for Class Imbalance Adaptationcited by 1
- 2024Free Performance Gain from Mixing Multiple Partially Labeled Samples in Multi-label Image Classificationcited by 0
- 2024A Survey on Incomplete Multi-label Learning: Recent Advances and Future Trendscited by 0
- 2025Sampling Imbalanced Data with Multi-objective Bilevel Optimizationcited by 0
- 2025On the Performance of Concept Probing: The Influence of the Data (Extended Version)cited by 0
- 2026Noise-Adaptive Regularization for Robust Multi-Label Remote Sensing Image Classificationcited by 0
- 2026A Scalable Curiosity-Driven Game-Theoretic Framework for Long-Tail Multi-Label Learning in Data Miningcited by 0
- 2026Addressing Imbalance in Multi-Label Data via Label-Specific Distance-based Oversamplingcited by 0
- 2026Addressing Imbalance in Multi-Label Data via Label-Specific Distance-based Oversampling4 Jun 2026
- 2026Model-Agnostic Meta Learning for Class Imbalance Adaptation20 Apr 2026
- 2026A Scalable Curiosity-Driven Game-Theoretic Framework for Long-Tail Multi-Label Learning in Data Mining17 Feb 2026
- 2026Noise-Adaptive Regularization for Robust Multi-Label Remote Sensing Image Classification13 Jan 2026
- 2025Curiosity Meets Cooperation: A Game-Theoretic Approach to Long-Tail Multi-Label Learning20 Oct 2025
- 2025On the Performance of Concept Probing: The Influence of the Data (Extended Version)24 Jul 2025
- 2025Sampling Imbalanced Data with Multi-objective Bilevel Optimization12 Jun 2025
- 2024A Bilevel Optimization Framework for Imbalanced Data Classification15 Oct 2024
- 2024Ensemble Methods for Sequence Classification with Hidden Markov Models11 Sep 2024
- 2024A Survey on Incomplete Multi-label Learning: Recent Advances and Future Trends10 Jun 2024
- 2024Free Performance Gain from Mixing Multiple Partially Labeled Samples in Multi-label Image Classification24 May 2024