山西大学学报(自然科学版)2024,Vol.47Issue(5):973-981,9.DOI:10.13451/j.sxu.ns.2023106
基于样本旋转的生成困难样本的深度度量学习方法
Sample Rotation-based Hard Sample-Generating Methods for Deep Metric Learning
摘要
Abstract
Existing deep metric learning methods guide efficient training of the model by constructing hard sample generation meth-ods.The hard sample generation methods based on algebraic computation have the advantages of simplicity and efficiency.Howev-er,such methods lack consideration of the overall data distribution,resulting in strong randomness of the generated samples and slow convergence of the model.To address this problem,we propose a hard sample generation method based on sample rotation by rotating positive samples in a triad to the reverse extension of the line connecting the anchor point and the class center on the axis of the class to which they belong,and give a new loss function to construct a deep metric learning model(RHS-DML)for generating hard samples based on sample rotation,effectively improving the training efficiency of the model.Experiments on image retrieval were conducted on the Cars196,CUB200-2011,and Stanford Online Products datasets,and compared with the symmetric sample generation method in algebraic computing.The results showed that the retrieval performance of the algorithm proposed was 2.4%,0.7%,and 1.4%higher than the symmetric sample generation cost method on the three datasets,respectively.关键词
深度度量学习/困难样本生成/多类N元组损失/代数计算Key words
deep metric learning/hard sample generation/multi-class n-pair loss/algebraic calculations分类
信息技术与安全科学引用本文复制引用
张鸽,闫京,魏巍,梁吉业..基于样本旋转的生成困难样本的深度度量学习方法[J].山西大学学报(自然科学版),2024,47(5):973-981,9.基金项目
国家自然科学基金(61976184) (61976184)