AI宝库

AI宝库首页 › AI 资讯 › 见LEANN:一个开源矢量指数,Shrinks a 60M-Chunk RAG指

见LEANN:一个开源矢量指数,Shrinks a 60M-Chunk RAG指数从201GB到6GB

来源:MarkTechPost · 发布时间:10-28 · 由 AI宝库每小时自动聚合

UC Berkeley的LEANN矢量指数存储了被打压的图表,并在查询时间重算嵌入。 使用 4 GB 而不是 188 来匹配 HNSW 的精确度:

UC Berkeley的天空计算实验室建造了LEANN,这是一个开源向量指数,几乎不存储嵌入。 LEANN向量索引在查询时重算嵌入,而不是保存到磁盘上。 Repo 基准测试显示一个6000万块的维基百科指数从201GB下降到6GB. 研究论文在MLSys 2026上获得了"最佳研究论文"奖。 因此,它带来语义搜索和RAG在个人数据之上,在一台笔记本电脑上。

TL; DR 数据

- 大小:6 GB 对于 60M 块( repo); 4 GB 在纸张的76 GB 维基百科设置中。 默认嵌入器:Contriever (110M params). 它是一个指数,而不是LLM,所以没有适用上下文长度。

电脑和工作站论文试制了一台装有32GB RAM和一款M1 Ultra Mac的RTX 4090盒。 只有CPU的 Linux 安装存在,没有发布 CPU 时空。

- 性能:使用不到5%的原始数据存储量,以90%的回收率与HNSW下游精度相匹配。

- 最佳:25.5%的NQ精度,与HNSW完全相同,使用4GB而不是188GB.

- 最糟糕的是:热点QA检索需要44.80 s在Mac上,占管道总时间的50.1%.

什么是LEANN吗?

LEANN是一个由FAISS上方所建的低存储相近相邻(ANN)指数。 它针对一个问题:矢量指数往往大于其指数数据。 研究论文显示 a 76 GB被体需要173 GB的嵌入加15 GB HNSW图。 超过翻倍的原始数据。

LEANN保存了被刷卡的图,放下了完整的向量,并按需要重建。 代码是麻省理工学院许可的,并安装有uv pip安装lein.

LEANN如何在不拖延搜索的情况下重新计算嵌入?

HNSW 这样的图形索引访问每个查询只访问一小组节点。 LEANN利用这个。 它存储了图表外加一个紧凑的产品量化(PQ)表。 在查询时间,它只重置了搜索路径上的节点。

3种技术保持这种实用性:

- 两级搜索:便宜的PQ距离将邻居排在首位。 只有顶级候选人才能得到准确的重评。

- 动态批量:将几个跳板的考生分组,例如一次64个,提高GPU的效率。

- 高度保护分流:大多数节点保持很少边缘,而前2%的 " 接 " 节点保持较多。

在平面上,双层搜索给出了平均1.4x速度。 将批量融资加到1.8x. 嵌入重算仍然占了查询延迟的约76%.

LEANN在基准测试上表现如何?.

研究团队对一款60M-passage RPJ-Wiki的数据存储器进行了NQ,TriviaQA,GPQA和HotpotQA的测试。 延迟度测量为90%回召@3. 以下所有数字在研究论文中自报。

在NQ上,用一台RTX 4090上的Qwen3-4B,LEANN和HNSW都达到了下游精度的25.5%. LEANN使用4GB的存储。 HNSW使用188 GB. BM25的分数为 18.3%,PQ压缩分数为 17.9%.

代价是搜索时间。 LEANN检索在NQ上需要2.48 s,而HNSW则需要0.05 s. 世代需要大约20 s,所以端到端的耐性从20.95 s上升到23.34 s.

在整个RTX 4090上的数据集,LEANN检索范围从1.12 s(GPQA)到7.12 s(HotpotQA). 由EdgeRAG所启发的以IVF为主的重算基线,速度可达200x慢。 LEANN比BM25提高11.8%的精确比分。

GitHub repo在个人数据上报告了类似的节省。 400K聊天信息从1.8GB收缩到64MB. 38K浏览器条目从130MB收缩到6.4MB.

LEANN 对其它向量索引

储存、延迟和准确数字取自LEANN文件v2、表1和表2以及第7.2节(自报)。 许可证和设计行链接到每个项目自己的来源。

开发者在哪里可以使用LEANN?

Repo船准备应用文件,Apple Mail,Chrome历史,WeChat,iMessage,以及ChatGPT或Claude出口。 MCP读者从Slack和Twitter书签中提取活数据。 LEANN还作为语义搜索MCP服务器为Claude Code运行。

30 SWE-Bench Pro任务来自Context Bench,团队报告24.2%的初始相关代码召回,而BM25则11.4%. agent也使用了8.4%的代币。

HNSW是默认后端,DiskANN用于更快的搜索。 世代与Ollama,Hugging Face,Anthropic或任何OpenAI相容的API合作。 它支持Ubuntu,Arch,WSL,macOS和Windows. 一些应用程序,如Apple邮件搜索,只是macOS.

关键外卖

- LEANN存储一个已刷卡的图表,并在查询时重算嵌入。

- 60M块:201GB下至回波基准测试的6GB.

- 与HNSW精度相匹配(25.5%在NQ上),恢复90%。

- 搜索费用秒:2.48s对0.05s,NQ上的HNSW。

- 麻省理工学院执照和一份2026年最佳研究论文

请检查access-date=中的日期值 (帮助) Paper, GitHub Repo和PyPI包。 并加入我们的通讯。 等等,你在电报里吗?现在你也可以加入我们的电报。

需要和我们合作推广您的 GitHub Repo OR Hugging Face Page OR Product Release OR Webinar等吗?联系我们

Asif Razzaq是Marktechpost AI Media Inc.的首席执行官。 作为一个有远见的企业家和工程师,阿西夫致力于利用人工智能的潜力为社会造福。 他最近的努力是人工智能媒体平台的发布,该平台以深入报道机器学习和深层学习新闻而出名,这些新闻在技术上既可靠又为广大观众所容易理解。 该平台每月有超过200万次观看,表明其受观众欢迎。

本文由 AI宝库自动抓取并翻译,版权归原作者所有;如需查看原文请点下方链接。

相关 AI 工具

相关免费羊毛

同源资讯

阅读原文(marktechpost.com)

标题、摘要与正文均来自该信源的公开页面,版权归原作者所有;本站自动聚合与翻译,仅作出处跳转。

← 返回 AI 资讯列表AI工具图鉴首页

返回 AI宝库首页 · 提交工具 · 商务合作 · 关于本站 · 595 款 AI 工具分类导航 · 每日 AI 热门项目与动态