词与权重:通过协同适配简化多轮交互 Words & Weights: Streamlining Multi-Turn Interactions via Co-Adaptation
ROSA2 提出词与权重的联合优化框架,将多轮测试时策略适配的准确率提升30%并减少40%对话轮次。
前置知识
测试时策略适配 (T2PAM)
Test-Time Policy Adaptation for Multi-Turn Interactions,指在推理阶段不重新训练而通过实时调整模型行为来对齐用户偏好的范式。它通过在多轮交互中持续优化策略,避免了传统微调的高昂成本,是连接静态训练与动态部署的关键桥梁。
本文核心框架 ROSA2 正是在 T2PAM 范式下展开,理解该范式才能明白为什么需要一种既能调提示又能调参数的方法。
低秩适配 (LoRA)
Low-Rank Adaptation,一种参数高效微调方法,通过向原始权重矩阵注入低秩分解的可训练矩阵 A 和 B 来近似权重更新 $\Delta W = BA$,其中 $B \in \mathbb{R}^{d \times r}, A \in \mathbb{R}^{r \times k}$,秩 $r$ 远小于原始维度 $d, k$。推理时只需在前向计算中加上低秩项,原始权重保持冻结。
ROSA2 的参数优化模块基于 ROSA,而 ROSA 正是使用 LoRA 作为参数更新载体,因此理解 LoRA 有助于把握本文参数适配的具体实现机制。
KL 散度与奖励加权回归
KL 散度 $D_{KL}(P\|Q) = \sum_y P(y) \log\frac{P(y)}{Q(y)}$ 衡量两个分布的差异;奖励加权回归 (RWR) 通过将目标策略表示为参考策略的指数加权形式 $\pi^*(y) \propto \pi_{ref}(y) \exp(r(y)/\beta)$,将策略优化转化为加权最大似然问题。
本文的理论推导(定理 4.1 和 4.2)建立在最小化 Forward KL 散度以及 RWR 构造目标策略 $\pi^*_t$ 的基础上,是理解收敛性证明的核心数学工具。
Textual Gradient (TextGrad)
一种用自然语言文本而非数值梯度来优化提示的方法。它使用一个大语言模型作为"优化器",根据当前输出的失败原因以自然语言形式生成改进指令,从而迭代地精炼用户查询或系统提示。
ROSA2 的语义优化模块直接调用 TextGrad 来计算"文本梯度" $\nabla_x \mathcal{L}$ 并精炼查询 $x_{t+1} \to x^*_{t+1}$,理解 TextGrad 才能明白 Words 维度优化的具体操作。
前向-反向传播与全梯度近似
在联合优化框架中,全梯度 $\nabla_{joint} \mathcal{L}$ 包含对所有优化变量的偏导。本文将多轮交互损失 $\mathcal{L}$ 在联合状态 $\phi_t = \{x_{t+1}, \theta_t\}$ 上展开,得到 $d\mathcal{L} \propto -\mathbb{E}\left[\exp(r(y)/\beta) (\nabla_x \log\pi \cdot dx + \nabla_\theta \log\pi \cdot d\theta)\right]$,从而同时驱动语义与参数更新。
该公式是 ROSA2 区别于单轴方法的理论核心,必须理解如何把语义优化与参数优化在数学上统一为单一梯度下降过程。
研究动机
现有大语言模型在多轮交互中表现出的"适应性塌陷"问题:当用户在多轮对话中逐步提供反馈或纠正时,模型的响应准确率常常在前几轮小幅提升后便陷入停滞,既不能持续纠正错误,也无法在复杂推理任务(如 MATH 基准)上突破固有上限。具体而言,作者在 Qwen3-8B 上模拟 10 轮 MATH 交互时观察到,基线方法几乎是一条水平线,自纠错能力接近为零。这一现象源于现有方法将测试时适配视为单轴问题:纯提示工程(如 TextGrad)受限于模型冻结的内在能力,遇到能力天花板;纯测试时训练(如 ROSA、TTRL)在面对含噪、模糊的用户反馈时容易过拟合到噪声历史,产生"虚假关联"。两类方法分别落入作者所称的"Deficit Trap"(能力不足陷阱)和"Overfitting Trap"(过拟合陷阱),导致下游任务失败、轮次冗余,用户接受度显著下降。
本文的目标是本文旨在提出一种统一的多轮测试时适配框架 ROSA2,使得 LLM 能够在实时对话中同时精炼用户意图(Words)与增强自身能力(Weights),从而绕过单轴优化的局部最小值并逼近全梯度下降。文章的具体量化目标包括:在 MATH 等推理基准上比单轴基线提升约 30% 准确率(如 Qwen3-8B 上从 50.0% 提升到 80.8%),同时将平均交互轮次削减 40%(如 MATH 上从 7.2 降至 4.4 轮),在不显著增加显存(峰值仅多 3.1 GB)的前提下大幅降低单题延迟(如 AIME25 上从 557.4 s 降至 467.2 s)。此外,框架需要在稀疏奖励的真实代理任务(如 OSWorld、AndroidWorld)上验证有效性,并给出严格的收敛性证明。
与已有工作不同的是,本文的独特切入角度在于将交互失败明确归因为"语义模糊"和"能力缺口"两类耦合因素,并把 T2PAM 重新表述为关于异构变量 $\{x, \theta\}$ 的联合优化问题。区别于仅在 Words 或仅在 Weights 上做条件优化的现有工作,作者提出语义清晰度是参数更新的"预条件器"——只有先把上下文对齐到模型现有知识边界附近,参数梯度才会朝向真正的任务意图而非累积噪声。这一"共适配"(Co-Adaptation)视角既为现象(停滞、恢复)提供了优化景观的物理解释,也为算法提供了全梯度近似的理论基础,是论文最具新意的认知贡献。
核心方法
ROSA2 的整体思路是把多轮对话建模为在异构空间 $\phi_t = \{x_{t+1}, \theta_t\}$ 上的联合优化任务:Words 维度通过 TextGrad 计算文本梯度以精炼用户查询,Weights 维度通过 ROSA 的闭式解更新 LoRA 适配器权重,两者交替执行以逼近完整的交互损失梯度。直觉上,可把它想成一位教师与学生协作:教师先把学生含糊的问题改写得更清晰(语义优化),学生再针对改写后的问题更新自己的解题策略(参数优化);下一轮时教师再根据新的回答继续精炼,如此循环直到 $r_t = +1$ 或达到最大轮次 $T_{max}$。技术上,每一轮包含"生成-评估"和"联合优化"两阶段:先用组合参数 $\theta = \theta_{base} + \theta_t$ 生成响应 $\hat{y}_t$,接收奖励 $r_t$ 与原始反馈 $x_{t+1}$,然后并行执行语义流($x^*_{t+1} \leftarrow x_{t+1} - \nabla_{text}\mathcal{L}(\hat{y}_t)$)和参数流(构造目标 $\pi^*$ 并计算 $\Delta\theta_t$),最后将更新后的状态推入历史 $H_t$ 准备下一轮。
ROSA2 的核心创新是把 T2PAM 重新定义为带 reward-weighted 似然的全梯度近似问题,并证明语义精炼可作为参数更新的预条件器严格缩减所需参数位移。具体而言,作者将 Forward KL 散度 $D_{KL}(\pi^*\|\pi)$ 沿联合状态做全微分,得到 $d\mathcal{L} \propto -\mathbb{E}_{y \sim \pi_t}[\exp(r(y)/\beta) \nabla_\phi \log\pi(y|\phi) \cdot d\phi]$,从而理论上"迫使"系统同时优化两条路径。这与 ROSA(仅做权重)和 TextGrad(仅做提示)的本质区别在于:ROSA2 不是把语义和参数当作两个独立模块顺序执行,而是把它们视作同一目标函数的两条偏导数分支,让语义清晰度的提升直接降低参数侧的残差范数 $\|\Delta\theta\|_2$。这一"协同"性质在定理 4.1 中被严格证明为 $\|\Delta\theta_t(x^*)\|^2 < \|\Delta\theta_t(x)\|^2$,在定理 4.2 中则进一步给出收敛上界。
方法步骤详情
ROSA2 的完整算法流程(Algorithm 1)按轮迭代,每轮包含两阶段共 6 个具体步骤。第一阶段(生成与评估):步骤 1 将当前 LoRA 适配器参数 $\theta_t$ 加到冻结的基座权重 $\theta_{base}$ 上形成有效参数 $\theta$;步骤 2 用当前策略 $\pi_\theta$ 在历史 $H_{t-1}$ 条件下采样响应 $\hat{y}_t \sim \pi(\cdot | H_{t-1}, \theta)$;步骤 3 从环境获取二元奖励 $r_t \in \{+1, -1\}$ 以及原始用户反馈 $x_{t+1}$;步骤 4 若 $r_t = +1$ 或达到 $T_{max}$ 则直接返回 $\hat{y}_t$,否则进入阶段二。第二阶段(联合优化):步骤 5A(语义流)调用 TextGrad 计算 $\hat{y}_t$ 的文本梯度,得到精炼查询 $x^*_{t+1} \leftarrow x_{t+1} - \nabla_{text}\mathcal{L}(\hat{y}_t)$,即使没有显式反馈也会基于失败信号自动合成纠错指令;步骤 5B(参数流)利用 $r_t$ 与当前策略 $\pi_\theta$ 通过奖励加权回归构造目标分布 $\pi^*_t(y|H_{t-1}) \propto \pi_{t-1}(y|H_{t-1}) \exp(r(y)/\beta)$,然后用 ROSA 的闭式最小二乘解 $(J^TJ)\Delta\theta = J^T R(x)$ 算出 $\Delta\theta_t$,更新 $\theta_{t+1} = \theta_t - \nabla_\theta \mathcal{L}(\theta|r_t, \pi^*, \pi_\theta)$。步骤 6 将 $\{\hat{y}_t, x^*_{t+1}\}$ 追加到历史 $H_t$,轮次加一并回到步骤 1。所有步骤的输入均为历史 $H_{t-1}$ 与当前适配器 $\theta_t$,输出均为更新后的状态 $(\theta_{t+1}, x^*_{t+1}, H_t)$,总计算开销仅为一轮前向 + 两次反向(语义和参数)以及一次最小二乘求解。
技术新颖性
技术新颖性体现在三个层面。第一,**问题形式化的统一性**:现有 T2PAM 文献要么只看 Words(TextGrad、Feedback Descent),要么只看 Weights(ROSA、TTRL),本文首次把 T2PAM 写成一个关于异构变量的联合优化目标,并通过 Forward KL + RWR 把两类变量纳入同一目标函数,从而导出统一的全微分方程 (Eq. 8)。第二,**理论深度**:作者给出两个非平凡定理——定理 4.1 证明语义精炼后的参数更新范数严格小于精炼前(参数位移收缩),定理 4.2 把 T 轮总 KL 散度差分解为"初始误差 - 累积改进 + 累积近似误差",并证明在 $L$-Lipschitz 平滑假设下近似误差项被 $\sum_t \frac{L}{2}(\|\Delta x_t\|^2 + \|\Delta\theta_t\|^2)$ 严格界定,这是对 ROSA 原定理 4 的实质性扩展。第三,**错误归因机制**:即便没有用户反馈,语义流也能基于失败响应自动合成纠错查询,这相当于在传统 RL 稀疏奖励场景下提供了"自反馈"通道,使框架在 OSWorld、AndroidWorld 等稀疏奖励环境中仍能稳定收敛。
实验结果
实验覆盖 5 个模型(0.5B–8B 跨越小/大/推理三类)和 7 个推理基准,结果一致显示 ROSA2 在所有任务上达到最优。**主结果表 (Table 1)**:在 Qwen3-8B 上,ROSA2 在 MATH 取得 80.8% 准确率(+30.8% 相对基线 50.0%),MATH-500 上 80.6%(+37.8%),MMLU-Redux 上 84.4%(+27.4%),SuperGPQA 上 52.4%(+28.2%),MT-AIME24 上 44.4%(+15.0%),MT-MATH100 上 93.6%(+18.4%),HumanEval 上 88.4%(+10.4%),每一项均显著超越 TextGrad(最高 +19.6%)和 ROSA(最高 +18.8%)。在小模型 Qwen2.5-0.5B-Instruct 上,ROSA2 仍把 MATH 从 23.0% 拉到 40.8%(+17.8%),证明方法对算力受限场景同样有效。在推理专精模型 DeepSeek-R1-Distill-Llama-8B 上,MATH 从 27.6% 跃升至 54.2%(+26.6%),表明该方法可叠加在已具备强推理能力的模型之上继续提升。**交互动态表 (Table 2)** 显示 ROSA2 在 MATH 上的 Correction Uplift 达 81.4%(基线 70.0%、TextGrad 75.1%、ROSA 77.3%),且平均轮次从 7.2 降至 4.4(-2.8 轮,相对 -40%);MMLU 上从 6.6 降至 4.1 轮(-37.9%)。**稀疏奖励 UI Agent 表 (Table 3)**:在 OSWorld 上 UI-TARS-7B-SFT 配合 ROSA2 取得 23.6%(+10.4%,远超 ROSA 的 +4.6%),AndroidWorld 上 35.3%(+7.7%),证明语义预条件对低层动作精度的必要性。**计算成本表 (Table 4)**:ROSA2 在 MATH 上单题平均时间从 334.5 s 降至 297.6 s(-36.9 s),AIME25 从 557.4 s 降至 467.2 s(-16.2%),峰值显存最多增加 3.1 GB(MATH),说明高准确率不以算力为代价。**Figure 4** 进一步可视化在 AIME25、GPQA-Diamond、M-IMO、BigCodeBench-Hard 四个高难度基准上 ROSA2 曲线(红)单调上升直至第 8–10 轮仍无饱和迹象,而 TextGrad 和 ROSA 在第 4 轮后即停滞。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| MATH 数学推理 (Qwen3-8B) | Accuracy (%) | 80.8 | 50.0 (Baseline) | +30.8 绝对百分点 (相对 +61.6%) |
| MATH-500 数学推理 (Qwen3-8B) | Accuracy (%) | 80.6 | 42.8 (Baseline) | +37.8 绝对百分点 (相对 +88.3%) |
| MMLU-Redux 通用推理 (Qwen3-8B) | Accuracy (%) | 84.4 | 57.0 (Baseline) | +27.4 绝对百分点 (相对 +48.1%) |
| SuperGPQA 研究生级 QA (Qwen3-8B) | Accuracy (%) | 52.4 | 24.2 (Baseline) | +28.2 绝对百分点 (相对 +116.5%) |
| MT-AIME24 多语数学 (Qwen3-8B) | Accuracy (%) | 44.4 | 29.4 (Baseline) | +15.0 绝对百分点 (相对 +51.0%) |
| HumanEval 代码生成 (Qwen3-8B) | Accuracy (%) | 88.4 | 78.0 (Baseline) | +10.4 绝对百分点 (相对 +13.3%) |
| OSWorld UI 代理 (UI-TARS-7B-SFT) | Success Rate (%) | 23.6 | 13.2 (Baseline) | +10.4 绝对百分点 (相对 +78.8%) |
| AndroidWorld UI 代理 (UI-TARS-7B-DPO) | Success Rate (%) | 36.6 | 28.9 (Baseline) | +7.7 绝对百分点 (相对 +26.6%) |
| MATH 平均交互轮次 (Qwen3-8B) | Avg Turn (↓) | 4.4 | 7.2 (Baseline) | -2.8 轮 (-38.9%) |
| AIME25 单题延迟 (Qwen3-8B) | Avg Time (s) (↓) | 467.2 | 557.4 (Baseline) | -90.2 秒 (-16.2%) |
局限与改进
作者在正文中未单独列出"Limitations"小节,但从算法与实验设置可观察到以下几方面局限。其一,**对二元奖励的依赖**:ROSA2 的参数流依赖 $r_t \in \{+1, -1\}$ 的稀疏奖励信号,对于无法定义可验证 ground truth 的开放式写作、对话质量评估等场景,需要额外的奖励模型来提供 $r_t$,这会引入训练开销与偏差。其二,**TextGrad 的隐式成本**:语义流调用 TextGrad 本质上是让一个 LLM(如 GPT-4o 或本地大模型)作为"优化器"反复生成改进指令,作者虽然在 Table 4 中报告了端到端延迟下降,但未单独拆解 TextGrad 调用的 token 成本与延迟;在严格成本敏感场景中这可能成为瓶颈。其三,**LoRA 秩选择敏感**:参数流继承 ROSA 的低秩适配策略,但对不同任务所需的秩 $r$ 缺乏自适应机制,作者承认该超参数需要预先设定。其四,**长轮次的收敛保证**:定理 4.2 给出的收敛上界依赖 $L$-Lipschitz 平滑假设,但实际 LLM 的策略函数在长上下文下并不严格满足该假设,因此极深对话(>10 轮)时收敛界可能偏松。其五,**评估偏置**:实验集中在 Qwen 系列与 DeepSeek 蒸馏版,未涵盖闭源商业模型(如 GPT-5、Claude、Gemini),且主要在数学与代码这类可自动验证任务上测试,对创意、咨询类任务的适用性仍需进一步验证。
独立分析的弱点
独立审视 ROSA2 可以识别若干尚未充分解决的弱点。**第一,错误归因仍然依赖二元奖励**:在 $r_t = -1$ 时,ROSAA2 假定"回答错就启动联合优化",但现实对话中失败原因可能是多模态的(事实性错误、风格不合规、语气生硬等),统一使用同一套文本梯度+参数更新可能无法精准定位问题类型;改进方向是引入多维奖励分解(事实性、相关性、风格)并分别构造对应的语义梯度。**第二,Textual Stream 的可控性不足**:TextGrad 通过自然语言生成精炼查询,但其过程是黑箱的,难以保证精炼后查询的语义保真度,可能出现"过度改写"反而引入新歧义;改进方向是引入约束解码或基于嵌入相似度的语义保真度正则项 $\|\phi(x^*) - \phi(x)\| < \epsilon$。**第三,参数流只更新 LoRA 适配器**:对需要更深层能力跃迁的任务(如从未见过的推理模式),低秩更新的表达能力有限;改进方向是引入基于重要性分数的动态秩选择或稀疏差分更新。**第四,理论分析依赖近似假设**:定理 4.1 假设线性化参数更新 $(J^TJ)\Delta\theta = J^T R(x)$,但真实 LLM 的策略梯度高度非线性,闭式解与真实更新之间存在差距;改进方向是引入二阶近似的信赖域方法。**第五,未考虑多模态扩展**:当前仅在文本交互下定义 T2PAM,未涵盖图像、音频等多模态反馈,限制了其在 GUI Agent 场景中的扩展性;改进方向是把 Words 推广到多模态 token 序列并相应调整 KL 散度的支撑集。
未来方向
作者在论文末段并未明确列出未来工作,但结合结果可自然延伸出多个方向。**第一,跨任务迁移**:当前每轮 LoRA 从零初始化 $\theta_1 \leftarrow 0$,能否在相似任务间复用已学到的适配器以加速冷启动?可探索基于任务嵌入的元学习初始化。**第二,奖励模型协同进化**:当前 $r_t$ 由规则或固定 RM 提供,未来可让语义流本身充当元 RM,即利用 LLM 的内在评判能力减少对外部 RM 的依赖。**第三,可信度与可解释性**:Textual Stream 生成的纠错指令可作为可解释性窗口,未来可结合归因分析工具揭示失败根因,进一步用于模型审计。**第四,更复杂的多模态 Agent**:将 ROSA2 推广到 OSWorld 之外的多步 GUI 任务、长程网页交互等场景,测试其在大规模真实环境中的稳定性。**第五,安全与对齐**:测试时持续适配可能放大用户偏好注入风险,未来可探索如何在联合优化框架中加入安全性约束(如宪法 AI 风格的红队提示)以防止适配过程中的价值偏离。
复现评估
**开源情况**:论文未在正文中给出官方代码仓库链接,但作者团队(Wei、Shu 等)在 ROSA 原论文 (Wei et al., 2025b) 中已开源相关实现,ROSA2 在很大程度上是 ROSA + TextGrad 的组合,理论上可基于 ROSA 仓库扩展。**数据**:所有使用的 7 个基准(MATH 7500/5000、AIME25 30 题、MATH-500 500 题、GPQA-diamond 198 题、MMLU-Redux 3000 题、SuperGPQA 26500 题、HumanEval 164 题、MT-AIME24、MT-MATH100、M-IMO、BigCodeBench-Hard、OSWorld、AndroidWorld)均为公开学术基准,可从原始仓库直接获取。**算力**:Table 4 显示峰值显存约 95 GB(推断使用单卡 H100 或 A100 80GB 配置),单卡 GPU 即可复现 Qwen3-8B 与 Qwen2.5-7B 实验;DeepSeek-R1-Distill-Llama-8B 同等规模。**复现难度**:中等偏高。算法本身清晰(Algorithm 1 仅 20 行),但需要同时接入 TextGrad 的 LLM-as-optimizer 接口和 ROSA 的 LoRA 微调流水线,且需配置 7 个基准各自的 ground-truth 抽取规则(数学 $\boxed{}$ 解析、HumanEval 的 unit test 执行、UI Agent 的状态匹配);建议复现者按 Table 1 中模型从 Qwen2.5-0.5B 起步逐级验证,遵循控制变量先跑 Baseline 再加 TextGrad/ROSA 最后跑 ROSA2 的顺序。总体而言,由于所有组件均为公开工具且超参数(学习率、最大轮次 $T_{max}$、LoRA 秩)在附录中有据可查,本工作的可复现性评估为"较好"。
论文图表
左图 (a) 在 MATH(Qwen3-8B)10 轮交互中对比四种方法:Baseline 几乎水平,Prompt-Only 与 Param-Only 在第 3-4 轮后停滞,但 Switch Method(虚线,在第 5 轮切换优化轴)触发明显回升;ROSA2(红色实线)则从一开始就快速爬升并在第 7 轮接近 80% 准确率。右图 (b) 把这些现象映射到优化景观:Prompt-Only 落入 Deficit Trap(绿色),Param-Only 落入 Overfitting Trap(蓝色),Switch 沿虚线箭头逃离局部最小,ROSA2 沿红色轨迹直接进入 Success Zone。
这张图同时承担"现象展示"与"理论解释"双重职能,是论文论证"为什么需要联合优化"的关键证据,左侧支撑右侧景观理论的实证基础。