4 个月前

基于不完整数据学习的大规模图像检索迭代流形嵌入层

基于不完整数据学习的大规模图像检索迭代流形嵌入层

摘要

现有的流形学习方法并不适合图像检索任务,因为大多数方法无法处理查询图像,并且在大规模数据库中存在较高的额外计算成本。因此,我们提出了一种迭代流形嵌入(Iterative Manifold Embedding, IME)层,其权重通过无监督策略离线学习,以利用不完整数据探索内在流形。在包含27000张图像的大规模数据库中,IME层在查询时嵌入原始表示的速度比其他流形学习方法快120倍以上。在离线学习阶段,我们将位于高维空间流形上的数据库图像的原始描述符逐步嵌入到基于流形的表示中,生成IME表示。根据数据库图像的原始描述符和IME表示,我们通过岭回归估计IME层的权重。在在线检索阶段,我们使用IME层将查询图像的原始表示映射为IME表示,所需时间成本可忽略不计(2毫秒)。我们在五个公开的标准数据集上进行了图像检索实验。所提出的IME层显著优于相关的降维方法和流形学习方法。无需后处理的情况下,我们的IME层在大多数数据集上实现了对现有最先进的图像检索方法(带有后处理)性能的提升,并且需要更低的计算成本。

代码仓库

基准测试

基准方法指标
image-retrieval-on-instreIME layer
MAP: 82.4
image-retrieval-on-oxf105kSIFT+IME layer
MAP: 31.3%
image-retrieval-on-oxf105kCNN+IME layer
MAP: 87.2%
image-retrieval-on-oxf5kIsoMap [32]
MAP: 77.9%
image-retrieval-on-oxf5kPCA [51]
MAP: 82.6%
image-retrieval-on-oxf5kSIFT+IME layer
MAP: 62.2%
image-retrieval-on-oxf5kLLE [33]
MAP: 51.7%
image-retrieval-on-oxf5kIME
MAP: 83.5%
image-retrieval-on-oxf5kCNN+IME layer
MAP: 92%
image-retrieval-on-paris6kIME layer
mAP: 96.6

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于不完整数据学习的大规模图像检索迭代流形嵌入层 | 论文 | HyperAI超神经