中文

深度神经网络分类中被良好分类样本被低估

机器学习 2023-03-17 v6 人工智能 计算与语言 计算机视觉与模式识别

摘要

学习深度分类模型的常规思路是关注分类错误的样本,而忽略远离决策边界且已被良好分类的样本。例如,在使用交叉熵损失进行训练时,具有较高似然(即被良好分类的样本)在反向传播中贡献的梯度更小。然而,我们从理论上表明,这种常见做法阻碍了表示学习、能量优化和间隔增长。为抵消这一缺陷,我们提出通过附加奖励来激励被良好分类的样本,以恢复它们对学习过程的贡献。这一反例在理论上解决了这三个问题。我们通过直接验证理论结果,或在图像分类、图分类和机器翻译等多种任务上借助我们的反例取得显著性能提升,来为上述论断提供实证支持。此外,本文表明我们可以处理复杂场景,如不平衡分类、OOD 检测和对抗攻击下的应用,因为我们的思路能够解决这三个问题。代码见:https://github.com/lancopku/well-classified-examples-are-underestimated。

关键词

引用

@article{arxiv.2110.06537,
  title  = {Well-classified Examples are Underestimated in Classification with Deep Neural Networks},
  author = {Guangxiang Zhao and Wenkai Yang and Xuancheng Ren and Lei Li and Yunfang Wu and Xu Sun},
  journal= {arXiv preprint arXiv:2110.06537},
  year   = {2023}
}

备注

Accepted by AAAI 2022; 18 pages, 11 figures, 13 tables