面向长尾识别的混淆感知谱正则化器
Confusion-Aware Spectral Regularizer for Long-Tailed Recognition
把混淆矩阵的谱范数当作可微正则项来压低类间混淆,为长尾分类中「最差类精度」这一可靠性指标提供了有理论上界支撑的训练手段,可直接嵌入现有长尾训练流程。
原文依据 minimizes the spectral norm of the confusion matrix during training to reduce inter-class confusion
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论长尾图像分类长期是一个难题,因为真实世界数据通常服从高度不平衡的分布:少数头部类别占据主导,而许多尾部类别仅含有限样本。这种不平衡使特征学习偏向头部类别,并导致稀有类别性能显著下降。尽管近期研究提出了重采样、重加权与解耦学习等策略,但相较于整体精度,在最欠代表类别上的提升仍然有限。在本工作中,我们从以混淆为中心的视角研究长尾识别,显式关注最差类泛化。我们首先建立了一个新的类特定误差分析理论框架,表明最差类误差可以被频率加权混淆矩阵的谱范数与一个依赖模型的复杂度项紧密地上界约束。受此启发,我们提出混淆感知谱正则化器(CAR),在训练过程中最小化混淆矩阵的谱范数,以降低类间混淆并增强尾部类泛化。为实现稳定高效的优化,CAR 集成了可微混淆矩阵代理与基于 EMA 的混淆估计器,以在小批量之间维持平滑且低方差的估计。在多个长尾基准上的大量实验表明,CAR 显著提升了最差类精度与整体性能。当与 ConCutMix 增强结合时,CAR 在从零训练设置(提升 2.37% 至 4.83%)与预训练微调设置(提升 2.42% 至 4.17%)下,于 ImageNet-LT、CIFAR100-LT 和 iNaturalist 数据集上均持续超越现有的最优长尾学习方法。
英文原文摘要 1836 字符
Long-tailed image classification remains a long-standing challenge, as real-world data typically follow highly imbalanced distributions where a few head classes dominate and many tail classes contain only limited samples. This imbalance biases feature learning toward head categories and leads to significant degradation on rare classes. Although recent studies have proposed re-sampling, re-weighting, and decoupled learning strategies, the improvement on the most underrepresented classes still remains marginal compared with overall accuracy. In this work, we present a confusion-centric perspective for long-tailed recognition that explicitly focuses on worst-class generalization. We first establish a new theoretical framework of class-specific error analysis, which shows that the worst-class error can be tightly upper-bounded by the spectral norm of the frequency-weighted confusion matrix and a model-dependent complexity term. Guided by this insight, we propose the Confusion-Aware Spectral Regularizer (CAR) that minimizes the spectral norm of the confusion matrix during training to reduce inter-class confusion and enhance tail-class generalization. To enable stable and efficient optimization, CAR integrates a Differentiable Confusion Matrix Surrogate and an EMA-based Confusion Estimator to maintain smooth and low-variance estimates across mini-batches. Extensive experiments across multiple long-tailed benchmarks demonstrates that CAR substantially improves both worst-class accuracy and overall performance. When combined with ConCutMix augmentation, CAR consistently surpasses exisiting state-of-the-art long-tailed learning methods under both the training-from-scratch setting (by 2.37% 4.83%) and the fine-tuning-from-pretrained setting (by 2.42% 4.17%) across ImageNet-LT, CIFAR100-LT, and iNaturalist datasets.