R3A: Reinforced Reasoning for Relevance Assessment for RAG in User-Generated Content Platforms
针对 UGC 平台 RAG 系��中相关性评估的两大挑战——用户意图模糊和非对称相关性(相关性由局部答案片段驱动而非全局相似度),提出 R3A 框架。方法将相关性评估分解为意图推理和证据锚定两个阶段:利用高点击文档作为辅助信号推断潜在查询意图,再提取原文证据片段锚定相关性判断,降低噪声敏感性。蒸馏后的 R3A-1.5B 模型在大规模线上 A/B 测试中取得显著提升,兼顾了效果和部署效率。
INSIGHTR3A 的核心创新在于将相关性评估从「端到端打分」重构为「推理链式分解」:先推断用户真实意图,再基于证据片段做判断。这种设计对 UGC 短文本搜索极具实践意义——短文本中用户 query 天然模糊,而答案往往隐藏在帖子的某个段落中。通过强化学习优化推理过程(而非直接优化评分),模型能学到更鲁棒的相关性判断策略。1.5B 蒸馏版本在线上验证了可部署性,是大模型能力落地到搜索场景的良好范例。
1. 问题与动机
在 UGC 平台(如小红书)的 RAG 系统中,相关性评估面临两个传统检索方法难以解决的挑战。第一,用户意图模糊:UGC 平台缺少文档级点击日志来消歧用户意图,用户查询往往简短且含义多变(例如「减肥方法」可能指饮食控制、运动方案或药物推荐)。第二,非对称相关性:UGC 内容噪声大、格式不规范,文档的相关性并非由全局相似度决定,而是取决于其中是否存在局部的「答案承载片段」。传统基于全局语义匹配的方法(如 bi-encoder)在面对 UGC 噪声时性能严重退化。
现有的 LLM 相关性评估方法(如 UMbrela 提示工程方案)虽然在通用搜索上表现良好,但直接应用于 UGC 场景时效果不佳:QwQ-32B、DeepSeek-R1 等模型在 UGC 数据上的 F1 分数远低于专门训练的小模型。这表明 UGC 平台需要专门针对其数据特性设计的相关性评估框架。
2. 方法详解
R3A 将相关性评估分解为两个阶段的推理链:
阶段一:意图推理 (<intent>)。模型接收查询和最多 4 篇辅助文档(同一查询下的高点击文档)作为输入。辅助文档提供了用户潜在意图的上下文信号——如果用户搜索「感冒怎么办」后高频点击了中医偏方类帖子,模型可以推断用户更关注中医疗法而非西药推荐。模型输出对查询意图的自然语言分析。
阶段二:证据提取与评分 (<extract> + <score>)。基于第一阶段推断的意图,模型审视候选文档并必须提取与查询相关的原文证据片段(verbatim fragments),或返回None表示无相关内容。这种设计强制模型将评分锚定在具体的文本证据上,而非依赖模糊的语义印象。最终输出三级相关性评分:0-无关、1-部分相关、2-高度相关。
强化学习训练采用 GRPO(Group Relative Policy Optimization)而非 PPO,在分组 rollout 内计算相对优势,避免了 PPO 中 value network 的额外开销。奖励函数设计为:R = I_format × r_score,其中 I_format 确保输出格式正确(包含 <intent>、<extract>、<score> 标签),r_score 对精确匹配给 1.0 分、差 1 级给 λ=0.3、差 2 级给 0 分。
冷启动初始化:在 RL 训练前,使用 DeepSeek-R1 对 50k 无标注样本生成结构化 chain-of-thought 推理,作为 SFT 冷启动数据。这一步骤为模型建立正确的输出格式和基础推理模式,使后续 RL 训练更加稳定。
知识蒸馏:将 R3A-7B 教师模型蒸馏为 R3A-1.5B 学生模型,使用交叉熵损失拟合标签分布。学生模型仅输出相关性分数,不生成推理链,将推理延迟从 ~493ms 降至 ~11ms(每万条文档)。
3. 核心技术贡献
贡献一:非对称相关性的证据锚定机制。通过强制模型提取原文证据片段,R3A 将「是否相关」转化为「能否找到具体证据」的问题。消融实验表明,移除 <extract> 组件导致 F1 从 77.1 下降到 74.9,是所有组件中影响最大的。证据片段与标注相关性之间呈强正相关(相似度 0.8-1.0 时准确率接近最优),验证了证据提取是方法生效的核心机制。
贡献二:辅助文档驱动的意图消歧。利用同查询高点击文档作为隐式用户意图信号,无需显式的点击日志或用户画像。移除辅助文档使 F1 下降至 75.7,表明该信号提供了有效的意图上下文。
贡献三:高效的「推理→蒸馏」部署范式。通过 RL 训练大模型获得推理能力,再蒸馏到小模型用于部署。R3A-Distill-1.5B 不仅超越了同规模的 SFT 基线,甚至接近 7B 教师模型的性能,证明推理能力可以通过蒸馏有效传递。
贡献四:两阶段分离设计。将意图推理与文档评估分离,防止过长的上下文(辅助文档 + 候选文档)淹没模型的注意力。消融显示单阶段合并方案(combined intent+extraction)F1 仅为 73.8,比两阶段方案低 3.3 个百分点。
4. 实验结果分析
在小红书内部 NoteRel 数据集(7k 人工标注样本)上的核心对比:
| 模型 | Class 0 F1 | Class 1 F1 | Class 2 F1 | AUC |
|---|---|---|---|---|
| R1 (Qwen2.5-7B) | 74.4 | 49.6 | 63.7 | 81.2 |
| SFT (Qwen2.5-7B) | 76.0 | 52.3 | 61.8 | 82.0 |
| R3A-7B | 77.1 | 56.0 | 64.2 | 83.1 |
| R3A-Distill-1.5B | 78.3 | 55.9 | 60.3 | - |
关键发现:(1) R3A-7B 全面超越 R1 和 SFT 基线,尤其在最难判断的 Class 1(部分相关)上提升 +6.4 F1;(2) 蒸馏后的 1.5B 模型在 Class 0 和 Class 1 上甚至超过了 7B SFT 基线;(3) 在通用 T²Ranking 基准上,R3A-7B 达到 GPT-4o 水平,验证了跨域泛化能力。
线上部署结果(小红书,10% 流量,一周):人工评估 100 个样本中 23% 改善(Good:Same:Bad = 23:71:6),二次搜索率下降 1.03%。推理延迟 ~150ms,吞吐量 200 QPS/GPU,部署在 1000 GPU 集群上。
消融实验揭示的优先级:<extract>(证据提取)> 辅助文档 > <intent>(意图推理)> λ 参数。这表明在 UGC 场景中,「找到具体证据」比「推断用户意图」更为关键。
5. 局限与未来方向
作者坦承 R3A 在通用搜索基准(T²Ranking)上的优势小于在 UGC 数据上的表现。原因在于「UGC 平台和通用搜索引擎具有不同的上下文特征」,模型针对 UGC 噪声优化的设计(如证据提取、辅助文档)在规范化文本上的边际收益递减。
一个明显但作者未提及的局限是:辅助文档的质量依赖于初始检索系统的能力。如果第一阶段检索质量差,提供的辅助文档可能误导意图推断。此外,三级相关性标注可能过于粗粒度,无法捕捉 UGC 内容中更细微的相关性差异。
6. 工业实践意义
R3A 对搜索推荐从业者有三方面直接启示:
第一,「推理→蒸馏」是 LLM 能力落地的实用范式。 先用 7B 模型通过 RL 学习推理能力,再蒸馏到 1.5B 用于线上服务。推理延迟从 493ms 降至 11ms(每万条),达到生产可用水平。这种范式适用于任何需要 LLM 推理能力但受延迟约束的搜索场景。
第二,证据锚定比全局匹配更适合 UGC 短文本。 UGC 内容噪声大、答案隐藏在局部片段中,传统全局语义匹配容易被噪声干扰。R3A 的证据提取机制迫使模型聚焦于答案承载片段,这对社区搜索(帖子、问答、评论)尤其适用。
第三,高点击文档作为免费的意图信号。 利用已有的点击行为数据作为辅助上下文,无需额外标注成本即可提升意图理解。对于有一定流量基础的 UGC 平台,这是一个低成本高回报的策略。
部署考量:200 QPS/GPU 的吞吐表明 1.5B 模型的服务成本可控。但 1000 GPU 集群的规模提示该方案更适合头部平台。中小型平台可考虑进一步蒸馏到更小的模型(如 0.5B)或使用量化技术降低部署成本。