arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.18528cs.CV

梯度重加权:面向不平衡类增量学习

Gradient Reweighting: Towards Imbalanced Class-Incremental Learning

Elmore Family School of Electrical and Computer Engineering, Purdue University(普渡大学埃尔莫尔电气与计算机工程学院)

Jiangpeng He, Fengqing Zhu

AI总结:

针对类增量学习中的阶段间与阶段内双重类别不平衡问题,本文提出梯度重加权方法实现平衡优化和无偏分类器学习,并引入分布感知知识蒸馏损失缓解不平衡遗忘,在多个数据集上取得一致改进。

摘要:

类增量学习(Class-Incremental Learning,CIL)训练模型在非平稳数据中持续识别新类别,同时保留已学知识。将CIL应用于以非均匀分布为特征的真实世界数据时,面临一个主要挑战,即产生双重不平衡问题,包括:(i)旧任务存储样本与新类别数据之间的差异(阶段间不平衡),以及(ii)每个单独任务内部严重的类别不平衡(阶段内不平衡)。我们表明,这种双重不平衡问题会导致全连接(FC)层中带有偏置权重的梯度更新发生偏移,从而在CIL中引发过拟合/欠拟合和灾难性遗忘。我们的方法通过对梯度进行重加权来实现平衡优化和无偏分类器学习,从而解决该问题。此外,我们观察到不平衡遗忘现象:矛盾的是,在CIL过程中,实例丰富的类别反而遭受更高的性能下降,因为在后续学习阶段中更多的训练数据变得不可用。为解决这一问题,我们进一步引入一种分布感知的知识蒸馏损失,通过使输出logits与丢失训练数据的分布成比例地对齐来减轻遗忘。我们在CIFAR-100、ImageNetSubset和Food101上跨多种评估协议验证了我们的方法,并证明了与现有工作相比的一致改进,展示了在真实世界场景中应用CIL的巨大潜力,具有增强的鲁棒性和有效性。

英文摘要:

Class-Incremental Learning (CIL) trains a model to continually recognize new classes from non-stationary data while retaining learned knowledge. A major challenge of CIL arises when applying to real-world data characterized by non-uniform distribution, which introduces a dual imbalance problem involving (i) disparities between stored exemplars of old tasks and new class data (inter-phase imbalance), and (ii) severe class imbalances within each individual task (intra-phase imbalance). We show that this dual imbalance issue causes skewed gradient updates with biased weights in FC layers, thus inducing over/under-fitting and catastrophic forgetting in CIL. Our method addresses it by reweighting the gradients towards balanced optimization and unbiased classifier learning. Additionally, we observe imbalanced forgetting where paradoxically the instance-rich classes suffer higher performance degradation during CIL due to a larger amount of training data becoming unavailable in subsequent learning phases. To tackle this, we further introduce a distribution-aware knowledge distillation loss to mitigate forgetting by aligning output logits proportionally with the distribution of lost training data. We validate our method on CIFAR-100, ImageNetSubset, and Food101 across various evaluation protocols and demonstrate consistent improvements compared to existing works, showing great potential to apply CIL in real-world scenarios with enhanced robustness and effectiveness.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑