尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
Funnel Retrieval 深度拆解:MRL 五级重排级联如何用 8 维向量检索对齐 2048 维精度
Funnel Retrieval 深度拆解MRL 五级重排级联如何用 8 维向量检索对齐 2048 维精度【免费下载链接】MRLCode repository for the paper - Matryoshka Representation Learning项目地址: https://gitcode.com/gh_mirrors/mrl/MRLMRL 套娃表示为什么向量可以一层层剥开MRLMatryoshka Representation Learning套娃表示学习解决了一个经典矛盾图像检索要高精度就得用 2048 维的完整向量要省算力就得砍到低维——传统做法二选一。MRL 的思路是让一个 2048 维向量的前 8 维、前 16 维……前 2048 维各自都是可用的完整表示像套娃一样层层嵌套。实现上只需把分类头换成MRL_Linear_Layer它对嵌套尺寸[8, 16, 32, 64, 128, 256, 512, 1024, 2048]分别施加交叉熵损失再求和核心代码见 MRL.py。这样训出来的 ResNet50 只需一次前向就能输出任意尺寸的截断表示。如上图ImageNet-1K 检索实验MRL 在所有维度下都显著超过 SVD、SlimeNet、随机投影等传统降维基线——8 维表示就能拿到 56% 以上的 mAP10这是漏斗检索敢于从 8 维起步的前提。Funnel Retrieval五级重排级联的完整流程Funnel Retrieval漏斗检索的目标是用远低于 2048 维的计算量达到接近 2048 维直接检索的精度。整个流程分1 次粗检索 5 级重排候选集像漏斗一样逐级收窄阶段向量维度候选集长度操作① 粗检索FAISS82048对全库约 128 万张图做精确 L2 k-NN② 第 1 级重排16200截取前 200按 16 维距离重排③ 第 2 级重排32100截取前 100按 32 维距离重排④ 第 3 级重排6450截取前 50按 64 维距离重排⑤ 第 4 级重排12825截取前 25按 128 维距离重排⑥ 第 5 级重排204810截取前 10按完整 2048 维距离定序两条级联序列重排维度rerank_dim递增 候选集shortlist递减在 retrieval/reranking.ipynb 中定义rerank_dim [16, 32, 64, 128, 2048] # rerank cascade shortlist_set [[200, 100, 50, 25, 10], ...] # shortlist cascade它的巧妙之处在于重排只发生在候选集内部第 5 级虽然用 2048 维计算距离但只对 10 个候选做而非全库 128 万次比较。每一级都是维度升一档、候选砍一半最终输出 Top-10。相比需要人工挑D_r/D_s组合的单次自适应检索漏斗策略免除了对两个超参的调优且几乎不损失精度。效果如何128× 理论加速精度几乎无损上图给出 ImageNet-1K 与分布外数据集 ImageNet-4K 上 mAP10 对 MFLOPs/Query 的权衡曲线。Funnel 点红色漏斗标记紧贴 2048 维基线的精度水平ImageNet-1K约128× 理论加速真实部署中14× 加速ImageNet-4KOOD 场景6× 真实加速、32× 理论加速精度依然对齐也就是说一个 8 维起步的检索请求经过五级级联后拿到了与 2048 维全精度检索几乎相同的 Top-1 表现而计算量只剩原来的零头。相关文件导读如何在项目里复现嵌套分类层与损失函数MRL.py检索流水线说明retrieval/README.md生成库/查询向量inference/pytorch_inference.py加--retrieval --mrl参数FAISS 建库与 8 维粗检索retrieval/faiss_nn.ipynb漏斗重排核心实现retrieval/reranking.ipynbmAPk / Recallk 指标计算retrieval/compute_metrics.ipynb分布外评测集 ImageNet-4K 说明imagenet-4k/README.md总结Funnel Retrieval 是 MRL 嵌套表示的最佳打开方式8 维粗筛全库 → 五级重排逐级提精度 → 2048 维只给最终 10 个候选算距离。它以 128× 的理论计算节省对齐了满精度检索效果为大规模图像检索、以图搜图等场景提供了一条低成本、高精度的落地路径。【免费下载链接】MRLCode repository for the paper - Matryoshka Representation Learning项目地址: https://gitcode.com/gh_mirrors/mrl/MRL创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED

相关推荐

一文读懂PhoneProfilesPlus核心概念:3分钟搞懂Profile与Event的奥秘

一文读懂PhoneProfilesPlus核心概念:3分钟搞懂Profile与Event的奥秘

一文读懂PhoneProfilesPlus核心概念:3分钟搞懂Profile与Event的奥秘 【免费下载链接】PhoneProfilesPlus Profile manager for Android - event triggered 项目地址: https://gitcode.com/gh_mirrors/ph/PhoneProfilesPlus PhoneProfilesPlus(简称…

📅 2026/9/30 7:47:30
视频缩略图集神器vcsi:一条命令把整部电影变成16宫格速览图

视频缩略图集神器vcsi:一条命令把整部电影变成16宫格速览图

视频缩略图集神器vcsi:一条命令把整部电影变成16宫格速览图 【免费下载链接】vcsi Create video contact sheets, thumbnails 项目地址: https://gitcode.com/gh_mirrors/vc/vcsi vcsi 是一款开源免费的命令行工具,可以一键把任意视频文件&#x…

📅 2026/9/30 6:42:39
Laravel活动日志包laravel-logger完全解析:20+开箱即用功能节省90%审计开发时间

Laravel活动日志包laravel-logger完全解析:20+开箱即用功能节省90%审计开发时间

Laravel活动日志包laravel-logger完全解析:20开箱即用功能节省90%审计开发时间 【免费下载链接】laravel-logger An out the box activity logger for your Laravel or Lumen application. Laravel logger is an activity event logger for your laravel applicatio…

📅 2026/10/8 19:59:36
MORE NEWS

更多资讯

📰

纯Java自研Agent Harness平台:设计取舍与企业级落地复盘

2024 年下半年,我们团队开始认真琢磨怎么把大模型 Agent 能力接进手上那些 Java 存量业务系统。试了一圈当时流行的方案之后,我发现一个尴尬的事实:生态里做得顺手的 Agent 框架几乎都长在 Python 或 TypeScript 上,而我们的生产环…

📰

CAP理论在PHP工程中的落地:缓存、消息队列与主从架构的一致性实践

开头CAP定理这几年在分布式系统的讨论里几乎成了必考题。但说句实话,我在做PHP的这些年里,真正把CAP当成架构设计工具的团队并不多。大多数情况是:缓存不一致了,临时加一个延迟双删;消息重复消费了,临时加一…

📰

纯Java构建企业级Agent Harness平台:治理设计与实践

做企业级 Agent Harness 平台这件事,用纯 Java 来扛,很多人第一反应是不划算。但如果你真的在企业里做过 AI 应用,你会明白我说的“企业级”到底卡在哪儿:不是模型有多聪明,而是权限、审计、隔离、降级、成本控制这些破…

📰

Hadoop图书推荐系统实战:从伪分布式搭建到ItemCF算法落地

简介:这是一份基于Hadoop实现的图书推荐系统完整项目工程,面向大数据方向初学者与需要完成课程设计、毕业设计的开发者,可用于学习分布式数据处理与推荐算法落地。压缩包共346个文件、6.57MB,前端资源(js、css、png、j…

📰

社区智能诊疗系统Java毕设全复盘:SpringBoot+Vue从设计到部署

社区医疗的信息化水平,坦白说,比大多数人想象中要落后不少。很多社区卫生服务中心至今还在用纸质登记本管理挂号,用Excel表维护医生排班,电子病历更是一笔糊涂账。这个小场景,恰好是Java毕设的一块绝佳试验田——业务链…

📰

ComfyUI本地部署完整指南:从环境配置到工作流搭建

1. 为什么我劝你趁早把 ComfyUI 搬到本地先说结论:如果你打算长期玩 AI 绘图,本地部署 ComfyUI 这件事,早晚都得做,早做早省心。我自己是从在线版一路用过来的,最开始图省事,觉得网页打开就能跑&#xff0c…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

读完文章,想聊聊您的网站?

告诉我们您的行业与需求,资深顾问一对一梳理方案与报价,全程免费。

📞 💬