← 返回 2026-01-13

可控记忆使用:在长期人机交互中平衡锚定与创新 Controllable Memory Usage: Balancing Anchoring and Innovation in Long-Term Human-Agent Interaction

Muzhao Tian, Zisu Huang, Xiaohua Wang, Jingwen Xu, Zhengkang Guo, Qi Qian, Yuanzhe Shen, Kaitao Song, Jiakang Yuan, Changze Lv, Xiaoqing Zheng 📅 2026-01-08 👍 24 2026-07-13 08:35
个性化Agent 偏好对齐 强化学习 记忆管理 长程对话

提出 SteeM 框架,让用户动态控制 AI 对历史记忆的依赖程度,解决记忆锚定问题

前置知识

Memory Anchoring(记忆锚定)

记忆锚定是指当 LLM 智能体拥有历史交互记忆时,无论用户明确要求什么依赖程度,模型都会默认产生高度依赖记忆的输出。就像一个被过去经历「锚住」的人,即使被要求「放空思维想新点子」,仍然会不自觉地沿用旧思路。论文发现,即使用自然语言明确指示模型「忽略历史」(LOW 模式),模型输出的记忆依赖分数仍然集中在 4-5 分(满分 5),与无指令时几乎没有区别。这一现象在 Qwen3-4B/8B、Gemini-2.5-Pro、GPT-5 等多个模型上普遍存在。

这是本文要解决的核心问题。理解记忆锚定才能理解为什么简单的提示工程(prompting)不够用,需要更深层的训练干预。

Memory-Dependence Score (MD-Score)

MD-Score 是论文提出的一个 1-5 分的行为度量指标,用于量化模型输出对历史记忆的依赖程度。1 分表示「外部化/通用重构」——回答完全基于通用知识,不需要具体历史;5 分表示「延续模式/深度嵌入」——回答是内部日志的直接延续,没有特定记忆就无法理解。该指标通过基于 rubric 的 LLM-as-a-judge 评估器实现,涵盖三个维度:内容依赖(事实是否来自内部项目材料)、模式依赖(推理结构是否沿用历史工作流)、风格依赖(是否使用内部术语和表达方式)。

没有这个量化指标,就无法系统性地研究和优化记忆依赖行为。它是整个评估和训练框架的基石。

GRPO (Group Relative Policy Optimization)

GRPO 是 DeepSeek-Math 提出的一种强化学习算法,是 PPO 的变体。它的核心思想是在一个 group 内采样多个候选输出,计算每个输出的相对优势(相对于 group 内其他输出的奖励均值),然后用 clipped 目标函数更新策略。相比 PPO 不需要单独训练 value model,计算开销更小。论文中用 GRPO 做第二阶段训练,奖励函数由三部分组成:对齐奖励(负对齐误差)、任务奖励(rubric 评分)、通用质量奖励(reward model 评分)。

SteeM 的 SFT+RL 两阶段训练中,GRPO 是 RL 阶段的核心算法,理解它才能理解奖励如何驱动模型学会可控记忆依赖。

Preference-Aligned Data Generation(偏好对齐数据生成)

由于记忆锚定的存在,直接采样难以获得低记忆依赖的训练数据。论文设计了一个巧妙的数据生成流水线:先用用户模拟器将原始查询改写为带偏好暗示的增强查询,从模型池采样多个候选回复,然后计算每个回复的实际 MD-Score,再用用户模拟器将原始查询改写为与实际 MD-Score 匹配的对齐查询。这样就得到了 (对齐查询, 记忆, 回复) 的训练三元组,其中回复的记忆依赖程度自然匹配查询中暗示的偏好。

这是解决「数据稀缺」问题的关键创新。没有这个流水线,就无法获得覆盖 1-5 全部依赖水平的高质量训练数据。

研究动机

在长期人机交互场景中,LLM 智能体通常配备记忆组件来存储用户画像、历史偏好和项目状态。然而,现有架构将记忆使用视为一个「全有或全无」的黑箱策略:一旦记忆被检索到,它对输出的影响力就由模型内部注意力机制隐式决定,用户无法实时调节。实验数据清楚地展示了这个问题的严重性:在 Qwen3-4B 上,无论用户使用 NONE(无指令)还是 LOW(明确要求低依赖)提示,输出的记忆依赖分数分布几乎没有差异(均集中在 4-5 分);在 Qwen3-8B 上,Rubric Instruct 提示仅将平均对齐误差从 1.57 降低到 1.40,改善微乎其微。即使是 GPT-5 和 Gemini-2.5-Pro 这样的顶级模型也表现出同样的记忆锚定行为。这意味着一个研究者想要「全新视角的审稿人」来审视项目时,AI 仍然会不由自主地沿用历史方案,只做微小的增量修改。

本文的目标是本文的具体目标是将记忆依赖程度建模为一个显式的、用户可控的行为维度,让用户能够根据当前任务需求实时调节 AI 对历史记忆的依赖程度。技术上,给定用户查询 $q$ 和其查询特定记忆 $M(q)$,优化模型参数 $\theta$ 使得生成的回复 $y \sim \pi_\theta(\cdot | q, M(q))$ 的记忆依赖分数 $D_R(y; q, M(q))$ 尽可能匹配用户指定的偏好 $p(q) \in \{1,2,3,4,5\}$,即最小化对齐误差 $\delta_{align}(q, M(q), y) = |D_R(y; q, M(q)) - p(q)|$。同时,这种对齐不能以牺牲回复质量为代价。

与已有工作不同的是,已有工作(如 RMM、LD-Agent、O-Mem)专注于提升记忆检索和组织的效率,但几乎没有提供用户调节记忆依赖程度的机制。已有的个性化评估工作(如 PrefEval、PersonaMem)主要关注事实性偏好满足,忽略了记忆依赖这一重要但未被探索的偏好维度。更重要的是,已有工作隐式假设每个查询的偏好与历史交互一致,但现实中的偏好是意图依赖的——一个通常严谨的用户可能偶尔需要天马行空的回复。本文的独特切入角度是:不改变记忆检索本身,而是在记忆已经可用的情况下,通过训练让模型学会根据用户意图灵活调节对记忆的依赖程度。这是一种「行为层面」的控制,而非「信息层面」的过滤。

核心方法

SteeM 的整体思路可以用一个类比来理解:想象一个经验丰富的助手,桌上摆满了过去的会议记录。当老板说「给我一个全新方案」时,助手会有意识地把旧记录推到一边,用新鲜眼光思考;当老板说「接着上次的继续」时,助手会仔细翻阅之前的记录,确保连贯性。SteeM 就是要训练 AI 学会这种「情境感知的记忆使用」能力。技术路线分为三步:首先构建一个模拟长期交互的合成数据集,用于量化研究记忆依赖行为;然后定义基于 rubric 的记忆依赖度量和偏好对齐目标;最后通过偏好对齐的 SFT 和 RL 两阶段训练,让模型学会根据查询中隐含的偏好信号调节记忆依赖程度。

SteeM 的核心创新在于两个方面。第一,它将记忆依赖识别为一个可以被显式控制的连续维度,而非传统的二元「用/不用」。第二,它通过自然语言偏好表达而非预定义标签来实现控制——用户不需要在界面上拖动滑块,只需在查询中自然表达意图(如「忽略历史框架,我需要新想法」),模型就能理解并调节记忆依赖。这与直接在查询中添加 LOW/HIGH 标签的方法形成对比:实验表明(Table 6),标签训练虽然在对齐误差上略优(1.12 vs 1.16),但在 AlpacaEval 通用质量上显著退化(8.45 vs 8.73)。SteeM 的自然语言方式在保持通用能力的同时实现了有效的偏好对齐。此外,SteeM 的数据生成流水线解决了「低依赖数据稀缺」的关键难题——由于记忆锚定,直接采样几乎无法获得低依赖回复,论文通过用户模拟器重写查询和对齐匹配巧妙地绕过了这一瓶颈。

方法步骤详情

SteeM 的完整流程包括四个阶段。第一阶段是合成数据集构建:为 Research 和 Tutoring 两个场景各构建 200 个主题,通过 Gemini-2.5-Pro 迭代生成-验证的方式合成项目时间线(7539 个事件、7745 个工件),然后在时间线上实例化四类任务(Plan & Design、Revise、Analyze & Critique、Concept Explanation),为每个查询构建由用户画像 $m_{prof}$、跨会话摘要 $m_{inter}(q)$ 和会话内摘要 $m_{intra}(q)$ 组成的查询特定记忆 $M(q)$,最终得到超过 10,000 个 $(q, M(q))$ 对。第二阶段是偏好对齐数据生成:用 Gemini-2.5-Pro 驱动的用户模拟器将原始查询改写为带偏好暗示的增强查询 $q_{aug}$,从 Qwen3-8B/14B 模型池采样 4 个候选回复,计算每个回复的 MD-Score,再用用户模拟器生成与实际依赖分数匹配的对齐查询 $q_{align}$,经过质量过滤后得到 7000 个 SFT 训练样本。第三阶段是 SFT 训练:用标准 token 级交叉熵目标在 Qwen3-4B 和 Qwen3-8B 上微调,学习率 $1 \times 10^{-5}$,3 个 epoch。第四阶段是 GRPO 强化学习:从 2000 个不重叠样本出发,奖励函数 $R = R_{align} + R_{task} + R_{general}$,其中 $R_{align} = -\delta_{align}$ 直接优化对齐误差,$R_{task}$ 基于 rubric 评估任务质量,$R_{general}$ 由 Skywork-Reward-V2 评分保证通用质量。

技术新颖性

SteeM 的技术新颖性体现在三个层面。首先,在问题定义上,它是首个将「记忆依赖偏好」形式化为可量化、可优化目标的工作,提出了 $p(q) \in \{1,2,3,4,5\}$ 的偏好建模和 $\delta_{align}$ 对齐误差度量。其次,在数据工程上,论文设计了一套精巧的偏好对齐数据生成流水线,通过「用户模拟器重写 → 多模型采样 → MD-Score 计算 → 对齐查询匹配」的四步流程,解决了低依赖训练数据稀缺的问题。这比简单的「采样-过滤」策略高效得多,因为记忆锚定使得低依赖回复在自然采样中极其罕见。第三,在训练策略上,SFT+GRPO 两阶段设计配合三组件奖励函数,既保证了偏好对齐又保全了任务质量。实验表明,Tag-cued 训练(用预定义标签替代自然语言偏好表达)虽然对齐误差略低,但在 AlpacaEval 上质量显著下降,说明 SteeM 的自然语言方式是一种更优雅的平衡方案。

SteeM 方法全景图:评估、数据生成、训练和效果
Figure 2: SteeM 方法全景图:评估、数据生成、训练和效果

实验结果

论文的核心实验结果可以从四个维度来分析。第一,在偏好对齐方面,SteeM 在所有场景和任务上一致性地大幅降低对齐误差 $\delta_{align}$。以 Qwen3-8B 为例,基线(None)的平均对齐误差为 1.57,Rubric Instruct 为 1.40,SteeM (SFT) 降至 1.15,SteeM (SFT+RL) 进一步降至 1.13,相比基线降低了 0.43-0.44。在 Qwen3-4B 上,SteeM (SFT+RL) 的平均对齐误差为 1.19,相比基线 1.59 降低了 0.39。值得注意的是,即使在表现最差的 Revise 任务上,SteeM 也实现了 0.31-0.32 的降低。第二,在实现的依赖分数分布上(Figure 4),基线模型的质量高度集中在 4-5 分(高依赖),而 SteeM 显著将分布推向对角线,即实现的依赖分数更好地匹配目标偏好。第三,在泛化能力上(Figure 5),在训练中未见过的 Medical 和 Humanities 主题上,SteeM 仍然保持良好的对齐表现,RL 增强版的泛化能力优于纯 SFT 版本。第四,在回复质量保全方面(Figure 6),SteeM 的 Skywork-Reward 评分与基线相当,甚至在多个设置下略高;在 AlpacaEval 上(Table 4),SteeM (SFT+RL) 在 Qwen3-4B 上得分 8.73(基线 8.85),在 Qwen3-8B 上得分 10.43(基线 10.49),退化幅度极小。此外,在与 Memory Masking 的对比中(Figure 7),SteeM 在所有任务上的 pairwise 胜率都超过 Masking,总体胜率达 51.7%-58.6%,说明行为层面的控制优于简单的信息过滤。

各场景和任务上的对齐误差 δalign 对比
Table 1: 各场景和任务上的对齐误差 δalign 对比
PROJECT_METHOD 精修的案例研究
Table 5: PROJECT_METHOD 精修的案例研究
人类-评分器一致性验证与各模型的记忆依赖分数分布
Figure 3: 人类-评分器一致性验证与各模型的记忆依赖分数分布
目标偏好与实现依赖分数的混淆矩阵
Figure 4: 目标偏好与实现依赖分数的混淆矩阵
未见主题(Medical 和 Humanities)上的对齐误差雷达图
Figure 5: 未见主题(Medical 和 Humanities)上的对齐误差雷达图
SteeM 与基线的回复质量对比
Figure 6: SteeM 与基线的回复质量对比
SteeM 与 Memory Masking 的 pairwise 胜率对比
Figure 7: SteeM 与 Memory Masking 的 pairwise 胜率对比
查看结构化数据
任务指标本文基线提升
Research 场景平均对齐误差 δalign(平均,越低越好) SteeM (SFT+RL): Qwen3-8B 1.19, Qwen3-4B 1.24 None: Qwen3-8B 1.53, Qwen3-4B 1.59 Qwen3-8B 降低 22%, Qwen3-4B 降低 22%
Tutoring 场景平均对齐误差 δalign(平均,越低越好) SteeM (SFT+RL): Qwen3-8B 1.20, Qwen3-4B 1.25 None: Qwen3-8B 1.57, Qwen3-4B 1.62 Qwen3-8B 降低 24%, Qwen3-4B 降低 23%
AlpacaEval 通用质量 Skywork-Reward 评分 SteeM (SFT+RL): Qwen3-8B 10.43, Qwen3-4B 8.73 None: Qwen3-8B 10.49, Qwen3-4B 8.85 质量退化极小: Qwen3-8B -0.6%, Qwen3-4B -1.4%
与 Memory Masking 对比 Pairwise Win Rate SteeM: 51.7%-58.6% Memory Masking: 41.4%-48.2% SteeM 在所有任务上胜率超过 Masking

局限与改进

论文在 Limitations 章节承认了三个主要局限。首先,合成数据集虽然尽力模拟真实长期交互,但与真实人类交互仍有差距——真实的项目时间线可能包含更多意外、回退和跨领域关联,而合成数据的事件遵循相对规范的生成-验证流程。其次,记忆依赖偏好被建模为 1-5 的离散等级,但真实用户的偏好可能是更细粒度的甚至连续的。第三,当前只覆盖 Research 和 Tutoring 两个场景,扩展到更多应用场景和更丰富的任务分布是必要的。从独立观察来看,还有几个值得关注的局限:(1) 用户模拟器本身也是 LLM,其生成的「偏好暗示查询」质量取决于模拟器能力,可能引入系统性偏差;(2) MD-Score 作为 LLM-as-a-judge 指标,虽然通过人类验证了 pairwise 一致性,但绝对分值的校准精度未经验证;(3) 实验只在 4B 和 8B 参数量级的开源模型上做了训练,更大模型是否同样存在记忆锚定、以及 SteeM 训练方法是否同样有效,尚不清楚。

独立分析的弱点

SteeM 有几个值得关注的弱点。第一,记忆依赖的控制粒度受限于 5 级 rubric,这在实际使用中可能不够灵活。例如,用户可能只想「稍微减少一点」对某个特定历史工件的依赖,而不是整体切换到一个粗粒度的等级。改进方向是探索连续控制机制,比如让用户指定 0-1 之间的连续值,或者支持对不同记忆组件(用户画像、跨会话、会话内)分别设置依赖权重。第二,当前方法需要为每个场景专门构建合成数据集和训练,这限制了通用性。如果要将 SteeM 应用到新的场景(如编程助手、客服机器人),需要重新执行整个数据合成和训练流程。一个可能的改进方向是研究跨场景的迁移学习,或者设计场景无关的记忆依赖控制机制。第三,论文的评估完全依赖 LLM-as-a-judge(MD-Score),缺乏大规模人类评估来验证对齐误差的绝对值是否可信。虽然 pairwise 一致性验证结果不错(尤其在分数差距大时),但当两个回复的依赖分数接近时(差距 ≤ 1),人类-自动化的不一致率明显上升,这意味着在「中等依赖」区域的控制精度可能被高估。

未来方向

论文提出和可延伸的未来方向包括以下几点。首先,将记忆依赖偏好从离散等级扩展到连续光谱,这需要新的评估方法(可能需要回归而非分类的 rubric)。其次,扩展到更多应用场景——当前只覆盖了 Research 和 Tutoring,但长期交互还包括编程开发、创意写作、客服支持等场景,每个场景的记忆依赖模式可能不同。第三,研究动态偏好跟踪——当前方法假设每次查询的偏好是独立的,但真实用户可能在一次会话中频繁切换依赖程度,模型需要学会识别和适应这种动态变化。第四,探索 SteeM 与现有记忆管理系统(如 RMM、LD-Agent)的集成——SteeM 控制的是「如何使用记忆」,而这些系统控制的是「检索哪些记忆」,两者的结合可能产生更好的效果。第五,在更大规模模型(如 70B+)上验证 SteeM 是否同样有效,因为大模型可能表现出不同的记忆行为模式。

复现评估

论文在复现方面有一定优势。数据合成使用的是 Gemini-2.5-Pro(闭源 API),这增加了复现成本,但论文详细描述了合成流水线的每个步骤(附录 A),包括事件类型定义(Table 3)、查询模板、记忆构建方法等,理论上可以用其他 LLM 替代。训练使用 Qwen3-4B/8B(开源)和 MS-SWIFT 框架(开源),SFT 超参完整公开(batch size 64, lr 1e-5, 3 epochs),GRPO 使用 EASYR1 框架(batch size 32, lr 5e-6, max length 6144, 8 rollouts)。SFT 数据 7000 条、RL 数据 2000 条,规模适中。算力需求方面,在 4B 和 8B 模型上训练应该可以在单机多卡环境下完成,估计需要几张 A100 运行数小时到一天。总体复现难度中等——主要瓶颈是数据合成需要 Gemini-2.5-Pro API 调用(成本和速率限制),以及 MD-Score rubric 的 LLM-as-a-judge 实现。论文目前未明确说明是否开源代码和数据。