用强化学习让对话搜索自动生成更准的查询,无需人工标注。
ConvSearch-R1: Enhancing Query Reformulation for Conversational Search with Reasoning via Reinforcement Learning
- 通过强化学习直接优化重写,利用检索结果反馈代替人工标注。
- 在TopiOCQA上提升超10%,用30亿参数模型就达顶尖水平。
- 适合做对话搜索、低资源场景下的智能检索系统研发者。
对话搜索系统需处理依赖上下文、模糊、省略和指代等问题的查询。对话查询重写(CQR)通过将这些查询转化为可独立检索的形式来应对挑战。然而现有方法存在两大缺陷:严重依赖昂贵的人工标注或大语言模型监督,且重写模型与下游检索器对齐不足。我们提出ConvSearch-R1,首个完全摆脱外部重写监督的自驱动框架,通过强化学习直接以检索信号优化重写过程。其创新的两阶段方法包括:自驱动策略预热,通过检索引导的自蒸馏解决冷启动问题;以及检索引导的强化学习,采用专门设计的排名激励奖励机制缓解传统检索指标稀疏性问题。在TopiOCQA和QReCC数据集上的大量实验表明,ConvSearch-R1显著优于先前最先进方法,在挑战性较强的TopiOCQA数据集上提升超过10%,且仅使用30亿参数模型,无需任何外部监督。
原文摘要 · Abstract (English)
Conversational search systems require effective handling of context-dependent queries that often contain ambiguity, omission, and coreference. Conversational Query Reformulation (CQR) addresses this challenge by transforming these queries into self-contained forms suitable for off-the-shelf retrievers. However, existing CQR approaches suffer from two critical constraints: high dependency on costly external supervision from human annotations or large language models, and insufficient alignment between the rewriting model and downstream retrievers. We present ConvSearch-R1, the first self-driven framework that completely eliminates dependency on external rewrite supervision by leveraging reinforcement learning to optimize reformulation directly through retrieval signals. Our novel two-stage approach combines Self-Driven Policy Warm-Up to address the cold-start problem through retrieval-guided self-distillation, followed by Retrieval-Guided Reinforcement Learning with a specially designed rank-incentive reward shaping mechanism that addresses the sparsity issue in conventional retrieval metrics. Extensive experiments on TopiOCQA and QReCC datasets demonstrate that ConvSearch-R1 significantly outperforms previous state-of-the-art methods, achieving over 10% improvement on the challenging TopiOCQA dataset while using smaller 3B parameter models without any external supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。