中文

正交投影损失

计算机视觉与模式识别 2021-03-26 v1

摘要

深度神经网络在一系列分类任务上取得了显著性能,其中 softmax 交叉熵(CE)损失成为事实上的目标函数。CE 损失鼓励一个类的特征在真实类向量上比负类具有更高的投影分数。然而,这是一个相对约束,并未显式地迫使不同类特征良好分离。受 CE 损失中真值类表示为正交(独热编码向量)这一观察的启发,我们提出了一种称为“正交投影损失”(OPL)的新损失函数,它在特征空间中施加正交性。OPL 增强了 CE 损失的特性,并通过在 mini-batch 级别上施加正交性约束,直接在特征空间中实现类间分离与类内聚类。与 CE 的其他替代方案相比,OPL 具有独特优势,例如无额外可学习参数、不需要精细的负样本挖掘且对批大小不敏感。鉴于 OPL 的即插即用特性,我们在多样化任务上评估它,包括图像识别(CIFAR-100)、大规模分类(ImageNet)、域泛化(PACS)和少样本学习(miniImageNet、CIFAR-FS、tiered-ImageNet 和 Meta-dataset),并展示了其全面的有效性。此外,OPL 对对抗攻击和标签噪声等实际干扰具有更好的鲁棒性。代码见:https://github.com/kahnchana/opl。

关键词

引用

@article{arxiv.2103.14021,
  title  = {Orthogonal Projection Loss},
  author = {Kanchana Ranasinghe and Muzammal Naseer and Munawar Hayat and Salman Khan and Fahad Shahbaz Khan},
  journal= {arXiv preprint arXiv:2103.14021},
  year   = {2021}
}