首页 | 本学科首页   官方微博 | 高级检索  
     

优化三元组损失的深度距离度量学习方法
引用本文:李子龙,周勇,鲍蓉,王洪栋. 优化三元组损失的深度距离度量学习方法[J]. 计算机应用, 2021, 41(12): 3480-3484. DOI: 10.11772/j.issn.1001-9081.2021061107
作者姓名:李子龙  周勇  鲍蓉  王洪栋
作者单位:徐州工程学院 信息工程学院,江苏 徐州 221018
中国矿业大学 计算机科学与技术学院,江苏 徐州 221116
基金项目:国家自然科学基金资助项目(61806206);江苏省建设系统科技项目(2018ZD077);徐州工程学院校级科研项目(XKY2019107);江苏省高校自然科学研究项目(20KJB170023)
摘    要:针对基于三元组损失的单一深度距离度量在多样化数据集环境下适应性差,且容易造成过拟合的问题,提出了一种优化三元组损失的深度距离度量学习方法。首先,对经过神经网络映射的三元组训练样本的相对距离进行阈值化处理,并使用线性分段函数作为相对距离的评价函数;然后,将评价函数作为一个弱分类器加入到Boosting算法中生成一个强分类器;最后,采用交替优化的方法来学习弱分类器和神经网络的参数。通过在图像检索任务中对各种深度距离度量学习方法进行评估,可以看到所提方法在CUB-200-2011、Cars-196和SOP数据集上的Recall@1值比之前最好的成绩分别提高了4.2、3.2和0.6。实验结果表明,所提方法的性能优于对比方法,同时在一定程度上避免了过拟合。

关 键 词:深度距离度量  深度学习  三元组损失  卷积神经网络  Boosting  
收稿时间:2021-05-12
修稿时间:2021-07-26

Deep distance metric learning method based on optimized triplet loss
LI Zilong,ZHOU Yong,BAO Rong,WANG Hongdong. Deep distance metric learning method based on optimized triplet loss[J]. Journal of Computer Applications, 2021, 41(12): 3480-3484. DOI: 10.11772/j.issn.1001-9081.2021061107
Authors:LI Zilong  ZHOU Yong  BAO Rong  WANG Hongdong
Affiliation:School of Information Engineering,Xuzhou University of Technology,Xuzhou Jiangsu 221018 China
School of Computer Science and Technology,China University of Mining and Technology,Xuzhou Jiangsu 221116,China
Abstract:Focused on the issues that the single deep distance metric based on triplet loss has poor adaptability to the diversified datasets and easily leads to overfitting, a deep distance metric learning method based on optimized triplet loss was proposed. Firstly, by thresholding the relative distance of triplet training samples mapped by neural network, and a piecewise linear function was used as the evaluation function of relative distance. Secondly, the evaluation function was added to the Boosting algorithm as a weak classifier to generate a strong classifier. Finally, an alternating optimization method was used to learn the parameters of the weak classifier and neural network. Through the evaluation of various deep distance metric learning methods in the image retrieval task, it can be seen that the Recall@1 of the proposed method is 4.2, 3.2 and 0.6 higher than that of the previous best score on CUB-200-2011, Cars-196 and SOP datasets respectively. Experimental results show that the proposed method outperforms the comparison methods, while avoiding overfitting to a certain extent.
Keywords:deep distance metric  deep learning  triplet loss  Convolutional Neural Network (CNN)  Boosting  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号