中文

监督对比原型学习:无增强的鲁棒神经网络

机器学习 2022-11-29 v1 人工智能

摘要

深度神经网络(DNN)输入空间中的变换会导致特征空间中非预期的改变。几乎感知上相同的输入(如对抗样本)可能具有显著疏远的特征表示。相反,分布外(OOD)样本可能与训练集样本具有高度相似的特征表示。我们对使用分类头训练的DNN的理论分析表明,受分类问题规模限制的僵化logit空间是缺乏鲁棒性\textit{鲁棒性}的根本原因之一。我们的第二个观察是,DNN会过拟合训练增强技术,而无法学习细微不变\textit{细微不变}表示。受近期原型与对比学习框架在提升鲁棒性和学习细微不变表示方面成功的启发,我们提出一种训练框架,监督对比原型学习\textbf{监督对比原型学习}(SCPL)。我们使用同类和异类原型的N-pair对比损失,并以原型分类头\textbf{原型分类头}(PCH)替代分类头。我们的方法样本高效\textit{样本高效},不需要样本挖掘\textit{样本挖掘},可在任何现有DNN上实现且无需修改其架构,并可与其它训练增强技术结合。我们从经验上评估了我们的方法在分布外与对抗样本上的干净\textbf{干净}鲁棒性。我们的框架在鲁棒性\textit{鲁棒性}上优于其它最先进的对比与原型学习方法。

关键词

引用

@article{arxiv.2211.14424,
  title  = {Supervised Contrastive Prototype Learning: Augmentation Free Robust Neural Network},
  author = {Iordanis Fostiropoulos and Laurent Itti},
  journal= {arXiv preprint arXiv:2211.14424},
  year   = {2022}
}