论文标题
用球形嵌入深度度量学习
Deep Metric Learning with Spherical Embedding
论文作者
论文摘要
近年来,由于无缝将距离度量学习和深度神经网络相结合,深度度量学习引起了很多关注。许多努力都致力于设计不同的基于成对的角损耗函数,从而使嵌入向量的幅度和方向信息解脱,并确保训练和测试测量的一致性。但是,这些传统的角损耗无法保证所有样品嵌入在训练阶段的表面上,这将导致批处理优化的不稳定梯度,并可能影响嵌入学习的快速收敛。在本文中,我们首先研究了嵌入规范对具有角度距离的深度度量学习的影响,然后提出一个球形嵌入约束(SEC)以正规化规范的分布。 SEC自适应地调整了嵌入,以落在相同的Hypersphere上,并执行更平衡的方向更新。关于深度度量学习,面部识别和对比的自学学习的广泛实验表明,基于SEC的角度空间学习策略可显着提高最先进的表现。
Deep metric learning has attracted much attention in recent years, due to seamlessly combining the distance metric learning and deep neural network. Many endeavors are devoted to design different pair-based angular loss functions, which decouple the magnitude and direction information for embedding vectors and ensure the training and testing measure consistency. However, these traditional angular losses cannot guarantee that all the sample embeddings are on the surface of the same hypersphere during the training stage, which would result in unstable gradient in batch optimization and may influence the quick convergence of the embedding learning. In this paper, we first investigate the effect of the embedding norm for deep metric learning with angular distance, and then propose a spherical embedding constraint (SEC) to regularize the distribution of the norms. SEC adaptively adjusts the embeddings to fall on the same hypersphere and performs more balanced direction update. Extensive experiments on deep metric learning, face recognition, and contrastive self-supervised learning show that the SEC-based angular space learning strategy significantly improves the performance of the state-of-the-art.