3 个月前

MES-Loss:互等距分离度量学习损失函数

MES-Loss:互等距分离度量学习损失函数

摘要

近年来,深度度量学习(Deep Metric Learning, DML)因其在聚类、图像检索等领域的广泛应用而受到广泛关注。得益于深度学习(Deep Learning, DL)的成功,众多DML方法相继被提出。神经网络(Neural Networks, NNs)通过采用DML损失函数,学习一个映射函数,将样本映射至高判别性的低维特征空间,从而便于在该流形结构中度量样本对之间的相似性。现有大多数方法通常通过增强高层特征空间中的类内紧凑性来提升神经网络的判别能力,但未能显式地施加约束以优化类间分离性。本文提出了一种新的复合型DML损失函数,该函数在保持类内紧凑性的基础上,显式引入约束机制,通过使各类中心相互等距分布,以实现最优的类间分离。实验结果表明,所提出的DML损失函数在两个真实世界数据集上的聚类与图像检索任务中均取得了当前最优的性能表现。

基准测试

基准方法指标
image-clustering-on-cars196MES-Loss
NMI: 74.82
image-clustering-on-cub-200-2011MES-Loss
NMI: 73.35
image-retrieval-on-cars196MES-Loss
R@1: 87.89
R@8: 97.62
image-retrieval-on-cub-200-2011MES-Loss
R@1: 67.03
R@2: 75.67
R@4: 84.75
R@8: 91.50

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
MES-Loss:互等距分离度量学习损失函数 | 论文 | HyperAI超神经