arXiv:2508.15811cs.CLcs.AI2025-08KDD被引 10

用多阶段框架让对话系统更懂用户偏好,点击率提升34%

From Clicks to Preference: A Multi-stage Alignment Framework for Generative Query Suggestion in Conversational System

  • 分四步逐步对齐生成结果与用户意图:提示工程、监督微调、高斯奖励建模、强化学习
  • 引入概率化偏好模型GaRM,更好捕捉用户决策的不确定性,提升推荐准确性
  • 适合做对话系统个性化推荐、LLM生成优化的研究者和工程师

基于大语言模型的生成式查询建议能有效提升对话系统性能,但如何对齐生成结果与用户细微偏好仍是关键挑战。为此,我们提出一种多阶段对齐框架,实现生成策略与用户意图的渐进式对齐。流程始于提示工程作为冷启动策略,随后是监督微调阶段,通过点击日志蒸馏构建稳健的基础模型。为更精准建模用户偏好并捕捉其内在不确定性,我们设计了高斯奖励模型(GaRM),将用户偏好表示为概率分布而非点估计。最后,利用强化学习在复合奖励函数引导下对齐生成策略,该函数融合GaRM与辅助启发式规则以缓解奖励黑客问题。为保障训练稳定性,引入新颖的分布外正则化方法和两阶段奖励融合技术。大量实验表明,该框架在自动与人工评估中显著优于基线,在真实A/B测试中点击率相对提升34%。

原文摘要 · Abstract (English)

Generative query suggestion using large language models offers a powerful way to enhance conversational systems, but aligning outputs with nuanced user preferences remains a critical challenge. To address this, we introduce a multi-stage framework designed for progressive alignment between the generation policy and user intent. Our pipeline begins with prompt engineering as a cold-start strategy, followed by the Supervised Fine-Tuning stage, in which we introduce a distillation method on click logs to create a robust foundational model. To better model user preferences while capturing their inherent uncertainty, we develop a Gaussian Reward Model (GaRM) that represents user preferences as probability distributions rather than point estimates. Finally, we employ reinforcement learning to align the generation policy with these preferences, guided by a composite reward function that integrates GaRM with auxiliary heuristics to mitigate reward hacking. To maintain training stability, this process is enhanced by a novel out-of-distribution regularization method and a two-stage reward fusion technique. Extensive experiments demonstrate that our framework significantly outperforms baselines on both automatic and human evaluations and yields a 34\% relative increase in user engagement as measured by click-through rate in live A/B tests.

对话系统生成推荐偏好建模强化学习

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