arXiv:2510.14252cs.CL2025-10

让小模型像专家一样主动构建文档记忆,提升检索生成效果

MoM: Mixtures of Scenario-Aware Document Memories for Retrieval-Augmented Generation Systems

  • 用多路径采样和多视角评估选出最优文档记忆
  • 通过反向推理训练小模型掌握专家思维路径
  • 在三个领域验证,显著提升小模型的文本理解能力

传统RAG系统依赖被动分块理解文本,限制了知识深度与推理能力。本文提出场景感知文档记忆混合框架(MoM),将文档处理从被动分块转向主动理解,模拟人类阅读认知过程。首先通过大模型生成领域专家逻辑大纲,指导结构化分块与核心内容提取;采用多路径采样与多维度评估机制,设计清晰度与完整性指标筛选最优记忆;引入反向推理策略,从高质量结果中还原专家思维路径以训练小语言模型(SLMs);最后基于概率建模理论构建三层文档记忆检索机制。在三个不同领域的实验表明,该框架不仅解决了现有RAG系统的分块难题,为大模型提供语义完整的文档记忆,还使小模型具备类人智能文本处理能力。

原文摘要 · Abstract (English)

The traditional RAG paradigm, which typically engages in the comprehension of relevant text chunks in response to received queries, inherently restricts both the depth of knowledge internalization and reasoning capabilities. To address this limitation, our research transforms the text processing in RAG from passive chunking to proactive understanding, defining this process as document memory extraction with the objective of simulating human cognitive processes during reading. Building upon this, we propose the Mixtures of scenario-aware document Memories (MoM) framework, engineered to efficiently handle documents from multiple domains and train small language models (SLMs) to acquire the ability to proactively explore and construct document memories. The MoM initially instructs large language models (LLMs) to simulate domain experts in generating document logical outlines, thereby directing structured chunking and core content extraction. It employs a multi-path sampling and multi-perspective evaluation mechanism, specifically designing comprehensive metrics that represent chunk clarity and extraction completeness to select the optimal document memories. Additionally, to infuse deeper human-like reading abilities during the training of SLMs, we incorporate a reverse reasoning strategy, which deduces refined expert thinking paths from high-quality outcomes. Finally, leveraging diverse forms of content generated by MoM, we develop a three-layer document memory retrieval mechanism, which is grounded in our theoretical proof from the perspective of probabilistic modeling. Extensive experimental results across three distinct domains demonstrate that the MoM framework not only resolves text chunking challenges in existing RAG systems, providing LLMs with semantically complete document memories, but also paves the way for SLMs to achieve human-centric intelligent text processing.

RAG小模型文档记忆反向推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。