论文标题
基于双距离中心损失的车辆重新识别
Vehicle Re-identification Based on Dual Distance Center Loss
论文作者
论文摘要
最近,深度学习已被广泛用于车辆重新识别领域。当训练深层模型时,软马克斯损失通常被用作监督工具。但是,SoftMax损失在封闭设置的任务中表现良好,但对于开放式任务来说不是很好。在本文中,我们总结了中心损失的五个缺点,并通过提出双距离中心损失(DDCL)解决了所有这些缺点。尤其是我们解决了中心损失必须与SoftMax损失相结合的缺点,以监督训练模型,这为我们提供了检查中心损失的新观点。此外,我们还验证了特征空间中提出的DDCL和软马克斯损失之间的不一致,这使得中心损失不再受到删除SoftMax损失后功能空间中软疗法的损失的限制。具体来说,我们根据欧几里得距离到同一中心的距离增加了皮尔逊距离,这使同一类的所有特征局限于特征空间中的超孔和超立方体的相交。提出的皮尔逊距离增强了中心损失的类内部紧凑性,并增强了中心损失的概括能力。此外,通过在所有中心对之间设计一个欧几里得距离阈值,这不仅可以增强中心损失的类间可分离性,而且还可以使中心损失(或DDCL)效果很好,而无需结合SoftMax损失。我们将DDCL应用于名为VERI-776数据集和车辆数据集的车辆重新识别领域。为了验证其良好的概括能力,我们还将其在名为MSMT17数据集和Market1501数据集的人员重新识别领域常用的两个数据集中验证。
Recently, deep learning has been widely used in the field of vehicle re-identification. When training a deep model, softmax loss is usually used as a supervision tool. However, the softmax loss performs well for closed-set tasks, but not very well for open-set tasks. In this paper, we sum up five shortcomings of center loss and solved all of them by proposing a dual distance center loss (DDCL). Especially we solve the shortcoming that center loss must combine with the softmax loss to supervise training the model, which provides us with a new perspective to examine the center loss. In addition, we verify the inconsistency between the proposed DDCL and softmax loss in the feature space, which makes the center loss no longer be limited by the softmax loss in the feature space after removing the softmax loss. To be specifically, we add the Pearson distance on the basis of the Euclidean distance to the same center, which makes all features of the same class be confined to the intersection of a hypersphere and a hypercube in the feature space. The proposed Pearson distance strengthens the intra-class compactness of the center loss and enhances the generalization ability of center loss. Moreover, by designing a Euclidean distance threshold between all center pairs, which not only strengthens the inter-class separability of center loss, but also makes the center loss (or DDCL) works well without the combination of softmax loss. We apply DDCL in the field of vehicle re-identification named VeRi-776 dataset and VehicleID dataset. And in order to verify its good generalization ability, we also verify it in two datasets commonly used in the field of person re-identification named MSMT17 dataset and Market1501 dataset.