arXiv:2505.20285cs.CL2025-05被引 21

让大模型学会像人一样搜索,通过预训练提升智能体检索能力。

MaskSearch: A Universal Pre-Training Framework to Enhance Agentic Search Capability

  • 用掩码预测任务训练模型使用搜索工具补全内容,习得通用检索与推理能力。
  • 在多跳问答任务中,对域内和域外任务均显著提升性能,超越现有方法。
  • 结合自进化教师模型与课程学习,适合需要自主搜索的智能体应用。

检索增强型语言模型(RALMs)通过专用模块检索外部知识来增强生成能力。近期代理技术使大语言模型能自主调用工具进行检索、规划与推理。然而,现有基于训练的方法受限于特定任务数据的特性。为此,我们提出全新预训练框架MaskSearch。预训练阶段引入检索增强掩码预测(RAMP)任务,让模型在大量数据上学习利用搜索工具填补掩码片段,从而获得通用检索与推理能力。后续通过监督微调(SFT)和强化学习(RL)进一步优化:SFT采用多代理系统(规划器、重写器、观察者)与自进化教师模型生成数据;RL采用DAPO框架及答案奖励与格式奖励结合的混合奖励机制。此外,引入课程学习策略,按掩码数量从易到难逐步训练。在开放域多跳问答场景下评估表明,MaskSearch显著提升基于大模型的搜索代理在域内与域外任务上的表现。

原文摘要 · Abstract (English)

Retrieval-Augmented Language Models (RALMs) represent a classic paradigm where models enhance generative capabilities using external knowledge retrieved via a specialized module. Recent advancements in Agent techniques enable Large Language Models (LLMs) to autonomously utilize tools for retrieval, planning, and reasoning. While existing training-based methods show promise, their agentic abilities are limited by inherent characteristics of the task-specific data used during training. To further enhance the universal search capability of agents, we propose a novel pre-training framework, MaskSearch. In the pre-training stage, we introduce the Retrieval Augmented Mask Prediction (RAMP) task, where the model learns to leverage search tools to fill masked spans on a large number of pre-training data, thus acquiring universal retrieval and reasoning capabilities for LLMs. After that, the model is trained on downstream tasks to achieve further improvement. We apply both Supervised Fine-tuning (SFT) and Reinforcement Learning (RL) for training. For SFT, we combine agent-based and distillation-based methods to generate training data, starting with a multi-agent system consisting of a planner, rewriter, observer, and followed by a self-evolving teacher model. While for RL, we employ DAPO as the training framework and adopt a hybrid reward system consisting of answer rewards and format rewards. Additionally, we introduce a curriculum learning approach that allows the model to learn progressively from easier to more challenging instances based on the number of masked spans. We evaluate the effectiveness of our framework in the scenario of open-domain multi-hop question answering. Through extensive experiments, we demonstrate that MaskSearch significantly enhances the performance of LLM-based search agents on both in-domain and out-of-domain downstream tasks.

大模型智能体检索增强预训练

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