• •    

CCML2021+15: 优化三元组损失的深度距离度量学习方法

李子龙1,周勇2,鲍蓉1,王洪栋1   

  1. 1. 徐州工程学院
    2. 中国矿业大学
  • 收稿日期:2021-06-28 修回日期:2021-07-26 发布日期:2021-07-26
  • 通讯作者: 李子龙

CCML2021+15: Deep Distance Metric Learning Method Based on Optimizing Triplet Loss

  • Received:2021-06-28 Revised:2021-07-26 Online:2021-07-26

摘要: 针对基于三元组损失的单一深度距离度量在多样化数据集环境下适应性差,且容易造成过拟合的问题,提出了一种优化三元组损失的深度距离度量学习方法。首先,对经过神经网络映射的三元组训练样本的相对距离进行阈值化处理,并使用线性分段函数作为相对距离的评价函数;然后,将评价函数作为一个弱分类器加入到Boosting算法中生成一个强分类器;最后,采用交替优化的方法来学习弱分类器和神经网络的参数。通过在图像检索任务中对各种深度距离度量学习进行评估,实验结果表明,该方法在CUB-200-2011、 CARS-196和SOP数据集上的性能优于其他方法,同时在一定程度上还避免了过拟合。

关键词: 深度距离度量, 深度学习, 三元组损失, 卷积神经网络, boosting

Abstract: Focused on the issue that the single deep distance metric based on triplet loss is inappropriate for handling heterogeneous data and easily lead to overfiting, a deep distance metric learning method based on optimizing triplet loss was proposed. Firstly, by thresholding the relative distance of triplet training samples mapped by neural network, and a piecewise linear discriminant function wasused as evaluation function of relative distance; Secondly, the evaluation function was added to the Boosting algorithm as weak classifier to generate a strong classifier; Finally, In order to learn the parameters of the weak classifier and neural network, an alternating optimization method wasused to obtain the optimal values. The experimental results show that the performance of proposed mehtodon the CUB-200-2011、 CARS-196 and SOP data sets is better than other methods. At the same time, the proposed method also avoids overfitting to a certain extent.

Key words: Keywords: deep distance metric, deep learning, triplet loss, convolutional neural network, boosting