通过混淆样本配对增强数据,提升少数类识别准确率
Mix from Failure: Confusion-Pairing Mixup for Long-Tailed Recognition
- 基于模型混淆分布实时生成混淆样本对进行混合增强
- 在多个长尾数据集上显著提升少数类分类性能
- 适合处理类别不平衡的图像识别任务
长尾图像识别是更贴近真实世界分布的计算机视觉问题。现有方法通常通过调整损失函数、解耦分类器学习或设计多头专家架构来应对。本文提出一种新思路:通过增强训练数据来提升少数类样本多样性。所提方法Confusion-Pairing Mixup(CP-Mix)实时估计模型混淆分布,利用混淆样本对进行在线数据增强,使模型针对性地学习区分常被误判的类别对。同时,采用新型Mixup形式缓解由数据不均衡引起的决策边界偏移。大量实验表明,CP-Mix在多个长尾数据集上优于现有方法,有效缓解了分类器的混淆问题。
原文摘要 · Abstract (English)
Long-tailed image recognition is a computer vision problem considering a real-world class distribution rather than an artificial uniform. Existing methods typically detour the problem by i) adjusting a loss function, ii) decoupling classifier learning, or iii) proposing a new multi-head architecture called experts. In this paper, we tackle the problem from a different perspective to augment a training dataset to enhance the sample diversity of minority classes. Specifically, our method, namely Confusion-Pairing Mixup (CP-Mix), estimates the confusion distribution of the model and handles the data deficiency problem by augmenting samples from confusion pairs in real-time. In this way, CP-Mix trains the model to mitigate its weakness and distinguish a pair of classes it frequently misclassifies. In addition, CP-Mix utilizes a novel mixup formulation to handle the bias in decision boundaries that originated from the imbalanced dataset. Extensive experiments demonstrate that CP-Mix outperforms existing methods for long-tailed image recognition and successfully relieves the confusion of the classifier.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。