中文

通过超球面均匀性差距泛化与解耦神经崩溃

机器学习 2023-04-18 v2 计算机视觉与模式识别 机器学习

摘要

神经崩溃(NC)现象描述了深度神经网络的一种底层几何对称性,其中深度学习的特征与分类器均收敛到单纯形等角紧框架。已有研究表明,交叉熵损失与均方误差均可证明地导致 NC。我们去除 NC 关于特征维度与类别数的关键假设,进而提出一个泛化神经崩溃(GNC)假设,其有效包含了原始 NC。受 NC 刻画神经网络训练目标方式的启发,我们将 GNC 解耦为两个目标:最小类内变异性与最大类间可分性。我们随后使用超球面均匀性(刻画单位超球面上均匀程度)作为统一框架来量化这两个目标。最后,我们提出一个通用目标——超球面均匀性差距(HUG),其由类间与类内超球面均匀性之差定义。HUG 不仅可证明收敛到 GNC,还将 GNC 解耦为两个独立目标。与将类内紧致性与类间可分性耦合的交叉熵损失不同,HUG 具有更高灵活性,并作为良好的替代损失函数。实证结果表明,HUG 在泛化性与鲁棒性方面表现良好。

关键词

引用

@article{arxiv.2303.06484,
  title  = {Generalizing and Decoupling Neural Collapse via Hyperspherical Uniformity Gap},
  author = {Weiyang Liu and Longhui Yu and Adrian Weller and Bernhard Schölkopf},
  journal= {arXiv preprint arXiv:2303.06484},
  year   = {2023}
}

备注

ICLR 2023 (v2: fixed typos)