套娃嵌入建层次索引:MatRAG 多跳问答比 FAISS 快 47%

A Matryoshka Hierarchical RAG for Efficient Multi-Hop Question Answering

Gianluca Bonifazi, Christopher Buratti, Michele Marchetti, Federica Parlapiano, Giulia Quaglieri, Davide Traini, Domenico Ursino, Luca Virgili

cs.CL, cs.IR

2026-10-01

Matryoshka 前缀对齐聚类 DAG 层级,三基准 EM/F1 全第一,响应比 FAISS 快 47%,索引省三个数量级。

这篇在解决什么

多跳问答要把分散在多篇文档里的证据串成链,单轮稠密检索覆盖不了这种跨文档路径:一个查询和一篇文档的相似度,说不清下一跳在哪。现有解法分两派。图方法(GraphRAG、LightRAG、HippoRAG2)离线建知识图谱,查询时在图上游走;层次方法(RAPTOR)把语料组织成树,中间节点靠 LLM 写摘要。质量都不差,代价都高:建图要做实体链接和关系抽取,建树要递归调 LLM,查询时还得靠 LLM 规划下一跳。

代价有数:HotpotQA 语料(9,811 篇文档)上,GraphRAG 建索引花 661,392 秒,LightRAG 花 728,471 秒,合七天半到八天半;HippoRAG2 答一题平均 76.84 秒。

方法

MatRAG 把 MRL(Matryoshka Representation Learning,套娃表征学习,指向量的任意长度前缀截出来都是可用表征)的嵌套结构,对齐到聚类层次的粗细上。

离线建索引:

对齐的理由:粗层装的是语义混杂的大簇,在簇内排候选需要的判别信息本来就粗,短前缀装得下;全维度只留给最终给单篇文档排序的叶子层。

在线迭代检索:

结果

设置:三个基准、7 个基线、统一生成模型 gemma3:27b,每个基准抽 5 组各 1,000 题取均值。端到端问答(节选最强基线):

方法HotpotQA EM/F12Wiki EM/F1MuSiQue EM/F1
MatRAG60.50 / 73.6652.70 / 63.4227.40 / 37.89
NaiveRAG(FAISS 平铺)53.05 / 64.5044.10 / 48.1520.65 / 29.55
HippoRAG252.40 / 64.1049.85 / 55.2026.20 / 33.55
LightRAG52.15 / 66.5545.35 / 50.8021.85 / 31.80

六个指标全部第一。检索侧 R@5:HotpotQA 89.30(NaiveRAG 84.10)、2Wiki 79.27(HippoRAG2 77.80)、MuSiQue 57.08(NaiveRAG 50.35)。唯一丢的检索指标是 2Wiki 的 R@2,63.08 对 NaiveRAG 的 63.40。MuSiQue 专为反捷径检索设计,MatRAG 的 R@2 在这里领先 NaiveRAG 5.4 个百分点,是层次遍历有效最直接的证据。

指标(HotpotQA)MatRAG对照
响应时间1.85sNaiveRAG 3.52s;HippoRAG2 76.84s
索引时间316sGraphRAG 661,392s;NaiveRAG 98s
送进 LLM 的 token788GraphRAG 3,812;LightRAG 5,841

响应时间在三个基准上比 NaiveRAG 快 47.4%、35.8%、35.1%,比 HippoRAG2 快 96% 以上;索引比 NaiveRAG 慢(316 秒对 98 秒),换来的是检索质量。消融补了两个验证:截断前缀聚出的簇在 DB 指数和 Silhouette 分数上不降反微升,省算力没牺牲簇质量;自适应锚定优于两个固定权重变体,HotpotQA F1 73.66,对 72.25(只用原查询)和 71.24(只用扩展查询)。

为什么重要

「便宜的平铺检索」和「昂贵的结构化检索」之间缺一个中间档,MatRAG 补上了:结构化的质量拿到手,KG 构建和 LLM 规划的钱省下。更可迁移的是 MRL 的用法:此前都是部署时选一个折中维度、整条流水线统一用,按索引层级分配维度是新用法,不需要额外训练,nomic 这类现成模型直接支持。

适合证据分散的多跳问答、延迟敏感的在线检索、建不起图谱的领域;单跳事实查询用平铺稠密检索已经够。检索质量相对最强基线的提升多在 15 个百分点,属于渐进改进;查询延迟压到 FAISS 平铺的一半左右、索引成本比图方法低约三个数量级,这部分是实打实的工程收益。

局限与存疑

作者自述三条:GLiNER 在领域术语特殊的语料上抽实体不可靠,错误会传进预算控制和重排;语义分布太均匀的语料聚不出有判别力的簇;贪心自顶向下遍历没有回溯,粗层选错分支且正确父簇不在 p=2 冗余范围内时,相关文档就进不了候选集。

读下来另有几点存疑。语料只有 6,119 到 11,656 篇,离生产级 RAG 差四五个数量级,FAISS 平铺在万篇规模显不出规模优势,「快过 FAISS」放到百万篇语料是否成立没有验证,HDBSCAN 在那个量级的成本也未知。基线超参全部沿用各自原论文的设置,没在这些语料上重调,图方法的成绩可能被低估。「实体计数替代 LLM 规划」只和 ToG、KGP 这类图上 agent 比过,没有对照 IRCoT、FLARE 这类稠密检索上的迭代规划方法。

术语

原文与代码

社区讨论

相关论文

全部论文解读