用强化学习让大模型学会基于用户历史个性化推理。
Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning
- 通过打标签的方式构建可解释的推理链数据集
- 多阶段强化学习使模型提升32.65%个性化表现
- 适合需要精准用户适配的智能客服场景
当前大语言模型虽具备强大通用推理能力,但在个性化推理(即分析用户历史、推断独特偏好并生成定制回应)方面仍表现不足。为此,本文提出TagPR框架,通过‘打标签思考’方法显著增强模型内在个性化推理能力。首先构建数据驱动的自动推理链生成与语义标注流水线,形成结构化可解释数据集;随后采用分阶段强化学习策略:先在标注数据上进行监督微调建立基础推理模式,再通过融合标签约束与基于用户嵌入的个性化奖励模型(PRMU)的复合奖励信号进行优化,实现对用户特定逻辑的精细对齐。在LaMP公开基准和自建数据集上的实验表明,该方法在所有任务上相较基线模型平均提升32.65%,验证了结构化可解释推理是解锁大模型真实个性化能力的有效路径。
原文摘要 · Abstract (English)
Recent advancements have endowed Large Language Models (LLMs) with impressive general reasoning capabilities, yet they often struggle with personalization reasoning - the crucial ability to analyze user history, infer unique preferences, and generate tailored responses. To address this limitation, we introduce TagPR, a novel training framework that significantly enhances an LLM's intrinsic capacity for personalization reasoning through a tagging the thought approach. Our method first develops a data-driven pipeline to automatically generate and semantically label reasoning chains, creating a structured dataset that fosters interpretable reasoning. We then propose a synergistic training strategy that begins with Supervised Fine-Tuning (SFT) on this tagged data to establish foundational reasoning patterns, followed by a multi-stage reinforcement learning (RL) process. This RL phase is guided by a unique composite reward signal, which integrates tag-based constraints and a novel Personalization Reward Model with User Embeddings (PRMU) to achieve fine-grained alignment with user-specific logic. Extensive experiments on the public LaMP benchmark and a self-constructed dataset demonstrate that our approach achieves state-of-the-art results, delivering an average improvement of 32.65% over the base model across all tasks. Our work validates that structured, interpretable reasoning is a highly effective pathway to unlocking genuine personalization capabilities in LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。