← 返回 2026-02-12

QP-OneModel:面向小红书搜索的统一生成式多任务查询理解大模型 QP-OneModel: A Unified Generative LLM for Multi-Task Query Understanding in Xiaohongshu Search

Jianzhao Huang, Xiaorui Huang, Fei Zhao, Yunpeng Liu, Hui Zhang, Fangcheng Shi, Congfeng Li, Zechen Sun, Yi Wu, Yao Hu, Yunhan Bai, Shaosheng Cao 📅 2026-02-10 👍 6 2026-07-13 08:35
多任务学习 大语言模型 工业落地 强化学习 搜索系统 查询理解

将搜索QP子任务统一为生成范式,通过渐进式对齐策略实现工业级部署

前置知识

Query Processing (QP)

查询处理是搜索引擎的核心上游模块,负责将用户原始查询转化为结构化的语义表示,以便下游检索和排序系统使用。典型的QP子任务包括:命名实体识别(NER)用于提取查询中的品牌、产品等实体;分词(Word Segmentation)将连续文本切分为语义单元;词权重(Term Weighting)量化每个词对用户意图的重要性;查询分类(Query Taxonomy)将查询归类到预定义的类别体系。在小红书这样的社交内容平台,QP还面临社交媒体特有的俚语、缩写和新兴表达的挑战。

这是本文要解决的核心问题领域,理解QP的各个子任务及其相互依赖关系是理解本文方法的基础

序列到序列生成范式 (Seq2Seq Generation)

将传统上通过多个独立判别模型(如BERT序列标注器、分类器)分别处理的任务,统一转化为自回归生成问题。模型接收查询和配置信息作为输入,直接生成包含所有子任务结果的结构化JSON输出。这种方法打破了传统级联管线的依赖关系,允许全局优化任务间的内在相关性。关键优势在于:单一模型实例即可处理所有任务,无需维护多个专用模型;端到端优化自然捕获任务间依赖。

这是本文的核心技术范式转变,理解这种统一生成方式的优势和挑战是理解方法论的关键

强化学习与GRPO

Group Relative Policy Optimization (GRPO) 是一种策略优化算法,是Proximal Policy Optimization (PPO)的变体。在多任务设置中,GRPO通过组内归一化来优化策略梯度,特别适用于可验证奖励的任务。本文设计了复合奖励函数,平衡所有子任务的性能,直接使用评估指标作为奖励信号,确保优化目标与最终性能评估严格一致。这种设计避免了传统SFT容易导致的表面模式死记硬背问题,使模型真正内化业务逻辑。

第三阶段对齐策略的核心技术,理解GRPO如何在多任务设置中实现精准奖励归因是理解方法创新的关键

领域自适应预训练

RedOne是针对社交网络服务(SNS)领域的大语言模型,通过在大规模SNS数据上进行持续预训练,内化了社交媒体特有的语言模式、俚语表达和新兴词汇。相比通用基础模型(如Qwen),RedOne能够更好地理解小红书平台用户使用的非正式语言、缩写(如'1c1'代表雅诗兰黛粉底液色号)和平台特定术语。这种领域适应为下游QP任务提供了更好的初始化基础。

实验表明RedOne骨干相比通用Qwen骨干在多个指标上表现更优,是性能提升的重要来源之一

研究动机

传统QP系统面临两大根本性限制。首先,判别式模型(如BERT)的语义理解能力有限,经常难以捕捉复杂或细微的用户意图,尤其是在面对长尾查询或数据分布偏移时。例如,社交媒体上的新兴俚语、缩写和快速演变的词汇模式(如'1c1'代表雅诗兰黛粉底液色号)对传统模型构成巨大挑战。其次,独立训练的多模型管线维护成本高昂,任何微小的分类法或输入模式更新都需要重新训练整个模型,严重阻碍了快节奏工业环境中的快速迭代。现有LLM解决方案也存在问题:子任务孤立优化,忽视了任务间的内在语义协同;通用LLM难以适应SNS数据的分布偏移;复杂业务规则和数据稀缺使得标准微调不充分。

本文的目标是本文的具体目标是:(1) 将异构的QP子任务统一为单一序列生成范式,消除级联管线的错误传播问题;(2) 设计领域自适应的骨干模型和渐进式对齐策略,系统性地从广泛知识注入到精确业务逻辑内化;(3) 引入生成式意图描述作为新的高保真语义信号,增强下游应用;(4) 在小红书工业级场景中进行全面验证,包括离线评估和大规模在线A/B测试。

与已有工作不同的是,本文的独特切入角度体现在三个层面。第一,范式创新:不是简单地用LLM替换BERT,而是将整个QP工作流重新定义为统一的序列生成问题,利用全局查询上下文解决局部歧义。第二,训练策略创新:采用渐进式三阶段对齐(知识注入→目标分布对齐→多奖励RL),而非直接在有限数据上进行标准SFT。第三,输出创新:生成Intent Descriptions,这是一种自然语言叙述,阐述用户的底层搜索目标,作为判别模型无法提供的新型语义信号,有效增强查询改写和排序等下游任务。

核心方法

QP-OneModel的方法论框架遵循从直觉到技术路线的递进逻辑。直觉上,传统QP管线的错误传播问题源于独立模型无法共享上下文信息,而LLM的强大生成能力可以自然地将所有任务整合到单一输出中。技术路线分为三个层次:首先,将异构QP任务统一为序列到序列生成,模型接收查询、业务规则和上下文信息,输出包含所有子任务结果的JSON对象;其次,利用RedOne领域自适应骨干桥接通用语料与SNS数据的分布鸿沟;最后,通过渐进式三阶段对齐策略,从广泛知识注入逐步过渡到精确业务逻辑内化。这种设计使模型能够利用任务间的语义协同,例如准确的实体识别直接优化分词,两者共同优化词权重。

核心创新点在于统一生成范式与渐进式对齐策略的结合。与已有方法的本质区别体现在:(1) 任务统一方式——传统方法使用独立判别头处理各任务,本文将所有任务转化为单一序列生成,输出结构化JSON包含entities、segments、weights、category、intent_desc五个字段;(2) 对齐策略——标准SFT容易导致表面模式的死记硬背,本文引入多奖励RL阶段,通过优化与评估指标直接一致的复合奖励函数,驱动模型内化底层推理逻辑;(3) 任务执行顺序——精心设计的NER→分词→词权重→分类→意图描述顺序,使下游子任务能够利用上游结果,最大化任务协同。

方法步骤详情

方法分为四个主要步骤。第一步是统一序列生成重构:给定输入查询q、静态任务指令I、可配置业务规则R和动态上下文信息C,模型自回归生成结构化输出y = {entities, segments, weights, category, intent_desc}。第二步是业务感知提示设计:提示P = I ⊕ R ⊕ q ⊕ C整合可配置业务规则(实体定义、分词规则、词权重逻辑、分类法、意图分析方法)、用户改写历史C_hist和候选笔记C_note。第三步是渐进式三阶段训练:Stage 1使用混合训练策略,将大规模辅助数据集(历史日志伪标签,约10^7样本)与小规模人工标注统一数据(约10^5样本)结合;Stage 2仅使用实时人工标注数据进行目标分布对齐;Stage 3使用GRPO进行多奖励强化学习。第四步是数据采样优化:针对多任务设置,保留目标子任务奖励差异而其他子任务奖励一致的样本,实现精准奖励归因。

技术新颖性

技术新颖性体现在多个维度。首先,将传统上需要多个独立模型的QP管线统一为单一生成模型,这在工业搜索领域具有开创性。其次,渐进式三阶段对齐策略系统性地解决了数据稀缺和业务复杂性的双重挑战:第一阶段通过任务分解和混合训练从大规模历史日志中吸收广泛知识;第二阶段通过实时标注数据对齐目标分布;第三阶段通过多奖励RL防止死记硬背。第三,复合奖励函数设计R(ŷ, y) = Σ w_k · r_k(ŷ_k, y_k) 直接使用评估指标作为奖励信号,确保优化目标与最终评估严格一致。第四,数据采样策略针对多任务RL设置,通过筛选特定子任务奖励差异的样本,解决组内归一化掩盖子任务性能差异的问题。

QP-OneModel整体框架图
Figure 1: QP-OneModel整体框架图
业务感知提示设计示意图
Figure 2: 业务感知提示设计示意图

实验结果

离线评估结果表明QP-OneModel-8B相比BERT基线在整体得分上提升7.35%,在需要深度语义理解的困难任务上表现尤为突出:NER F1提升9.01%(从74.85%到83.86%),词权重F1提升9.31%(从56.86%到66.17%)。值得注意的是,QP-OneModel-0.6B也大幅超越BERT基线,整体得分78.81%比基线高5.83%,词权重F1提升8.59%,这表明性能提升源于方法论的有效性而非单纯的参数规模扩大。统一建模与任务隔离训练的对比实验显示,统一模型在高度协同的任务上表现出显著优势,整体得分79.36% vs 78.11%,验证了统一范式促进语义协同和正向知识迁移的假设。骨干模型对比表明RedOne2.0相比通用Qwen3在所有指标上都具有一致优势,分词提升0.25%,分类法提升0.69%。在未见任务泛化能力评估中,QP-OneModel-8B在Document Intent任务上达到82.40%准确率,显著超越Qwen3-32B(74.80%),证明了多任务多阶段对齐策略赋予模型稳健的抽象语义理解能力。在线A/B测试中,替换基础信号后DCG 0/1改善0.21%(超过0.15%的显著性阈值),零/少结果率降低0.4631%;使用意图描述进行查询改写使NECTR提升0.17%,用户留存率提升0.044%。

离线Golden Test Set整体性能对比
Table 1: 离线Golden Test Set整体性能对比
统一建模与任务隔离训练对比(8B模型)
Table 2: 统一建模与任务隔离训练对比(8B模型)
骨干模型对比(8B模型)
Table 3: 骨干模型对比(8B模型)
训练阶段消融研究
Table 4: 训练阶段消融研究
Document Intent和Authority Intent任务泛化性能对比
Table 5: Document Intent和Authority Intent任务泛化性能对比
在线A/B测试结果
Table 6: 在线A/B测试结果
挑战性案例定性分析
Table 7: 挑战性案例定性分析
部署架构概览
Figure 3: 部署架构概览
查看结构化数据
任务指标本文基线提升
整体QP性能(综合得分) Overall Score 79.89% (8B) / 78.81% (0.6B) 72.54% (BERT-Pipeline) +7.35% (8B) / +6.27% (0.6B)
命名实体识别 F1 Score 83.86% (8B) / 81.64% (0.6B) 74.85% +9.01% (8B) / +6.79% (0.6B)
词权重 Joint F1 66.17% (8B) / 65.45% (0.6B) 56.86% +9.31% (8B) / +8.59% (0.6B)
Document Intent (ICL) Accuracy 82.40% (QP-OneModel-8B) 74.80% (Qwen3-32B) +7.60%
在线搜索相关性 DCG 0/1 QP-OneModel信号 BERT信号 -0.21% (改善)

局限与改进

论文存在多个局限性值得探讨。首先,虽然统一建模带来了性能提升,但模型仍需依赖大规模历史日志的伪标签进行知识注入,这些伪标签存在噪声和不完整性(每个样本仅覆盖单一子任务),可能引入语义漂移。其次,渐进式三阶段训练策略增加了训练复杂度和时间成本,Stage 3的GRPO需要精心设计数据采样策略来实现精准奖励归因,这在其他多任务场景中的通用性有待验证。第三,评估主要集中在小红书平台特有的查询模式上,对于其他搜索场景(如电商、通用搜索)的迁移能力未被充分探索。第四,在线A/B测试的绝对改善幅度较小(DCG 0.21%,留存0.044%),虽然统计显著,但实际业务影响需要结合平台规模评估。第五,论文未详细讨论模型推理延迟和计算成本,8B参数模型在高并发搜索场景中的部署挑战可能被低估。最后,Intent Descriptions的质量评估主要依赖下游任务的间接验证,缺乏直接的生成质量评估指标。

独立分析的弱点

尽管QP-OneModel取得了显著成果,仍存在几个可改进的弱点。首先,任务执行顺序虽然精心设计,但固定顺序可能不适用于所有查询类型,可以探索动态顺序或并行生成策略。其次,复合奖励函数中的权重w_k需要人工设定,可以引入自适应权重调整机制,根据训练过程中各子任务的收敛情况动态调整。第三,当前的Intent Descriptions生成缺乏显式的质量控制机制,容易产生幻觉或与结构化输出不一致的内容,可以引入一致性约束或后验证模块。第四,对于极端长尾查询(如完全无语义的纯数字代码),模型仍依赖内部知识进行推断,准确性可能不足,可以探索结合知识图谱或外部检索增强的方法。第五,多奖励RL的数据采样策略虽然解决了精准归因问题,但可能损失大量训练样本,可以探索更高效的采样或课程学习策略。

未来方向

基于本文成果可延伸多个研究方向。首先,统一生成范式可以扩展到更多搜索相关任务,如查询扩展、相关性判断、甚至端到端的检索生成,探索更广泛的搜索任务统一。其次,Intent Descriptions作为新的语义信号,其在个性化推荐、内容理解、广告定向等下游应用中的潜力有待挖掘。第三,渐进式对齐策略可以推广到其他需要内化复杂业务规则的工业场景,如电商搜索、金融风控等。第四,可以探索更高效的多任务RL方法,如基于课程学习的渐进式难度调整,或利用任务间的层次结构设计分层奖励。第五,当前模型仅支持中文SNS场景,可以探索多语言或多平台的迁移能力,构建通用的社交搜索QP模型。作者提出的方向包括将QP-OneModel的统一生成能力应用于搜索结果页的更多流量场景。

复现评估

论文的复现评估需要考虑多个因素。在开源情况方面,论文未明确提及代码或模型的开源计划,这限制了学术社区的直接复现。数据方面,训练数据依赖小红书的历史日志和人工标注数据,这些数据具有高度的平台特异性和商业敏感性,外部研究者难以获取完全相同的数据集。算力需求方面,8B参数模型的三阶段训练(特别是Stage 3的GRPO)需要显著的GPU资源,具体训练成本未在论文中详细说明。复现难度评估:(1) 核心方法论(统一生成范式、三阶段对齐)相对清晰,可以在其他平台数据上尝试复现;(2) 业务感知提示设计和数据采样策略需要根据具体场景调整;(3) 骨干模型RedOne是小红书自研的领域适配模型,外部复现需要寻找替代方案(如使用通用LLM+领域继续预训练)。建议的复现路径:使用开源LLM(如Qwen)作为基础,在公开的搜索查询数据集上验证统一生成范式和渐进式对齐策略的有效性。