中文

MODE:面向 RAG 的文档专家混合模型

人工智能 2025-09-03 v1

摘要

检索增强生成(RAG)常依赖大型向量数据库和针对大规模语料库调优的交叉编码器,这对于小型、领域特定的集合而言可能过于繁重。我们提出 MODE(Mixture of Document Experts),一种轻量级替代方案,该方案将细粒度最近邻搜索替换为聚类路由检索。文档被嵌入、分组到语义连贯的聚类中,并由缓存的质心表示。查询时,我们路由到顶级质心(或质心),仅在这些聚类内部检索上下文,消除外部向量数据库和重排序的需求,同时保持延迟低。在 HotpotQA 和 SQuAD 语料库中进行 100-500 个块的实验后,MODE 在答案质量上匹配或超过密集检索基线,同时显著降低了端到端检索时间。消融实验表明,聚类粒度和多聚类路由控制召回/精确度的权衡,紧凑聚类有助于提高下游准确性。MODE 为在简单性、速度和主题聚焦方面都很重要的小型和中型语料库提供了实用方法。

关键词

引用

@article{arxiv.2509.00100,
  title  = {MODE: Mixture of Document Experts for RAG},
  author = {Rahul Anand},
  journal= {arXiv preprint arXiv:2509.00100},
  year   = {2025}
}