← 返回 2026-04-09

面向强化学习的提示学习:一种迁移感知的提示器-推理器协同训练框架 Learning to Hint for Reinforcement Learning

Yu Xia, Canwen Xu, Zhewei Yao, Julian McAuley, Yuxiong He 📅 2026-04-01 👍 9 2026-07-13 08:36
GRPO 协同训练 大语言模型推理 强化学习 提示学习

HiLL:联合训练提示器与推理器,让 GRPO 在难题上找回学习信号。

前置知识

GRPO(Group Relative Policy Optimization)

GRPO 是一种用于 RLVR 的策略优化方法。它对同一问题采样 $G$ 条轨迹,按组内奖励均值与标准差归一化得到优势 $A_i=\frac{r_i-\bar{r}}{\mathrm{std}(r_{1:G})+\epsilon}$,去掉价值网络,直接用组内相对优劣估计优势。

HiLL 直接针对 GRPO 的缺陷设计,因此必须理解 GRPO 如何用组内归一化构造优势,才能理解什么叫"组内奖励坍缩"以及为什么需要外部干预。

优势坍缩(Advantage Collapse)

当一个组内所有 $G$ 条轨迹都拿到相同奖励(要么全对要么全错),组内方差为零,于是每条轨迹的优势都被归一化为 0,整个样本对策略梯度没有贡献,浪费了推理算力也学不到东西。论文给出非退化概率 $s(p;G) = 1 - p^G - (1-p)^G$ 量化这一现象。

这是 HiLL 要解决的核心痛点。理解该现象后才能理解为什么要在难题($p \approx 0$)上下额外功夫,以及非退化概率 $s$ 如何进入提示器的奖励设计。

RLVR(Reinforcement Learning with Verifiable Rewards)

一种用可自动验证正确性的奖励(如数学题最终答案是否匹配)来训练 LLM 推理能力的方法范式,奖励一般是 0/1 二值。这套范式摆脱了对人类偏好数据的依赖,使得大规模 RL 训练成为可能。

HiLL 的全部实验与论证都建立在 RLVR 设定下,二值奖励是触发优势坍缩的直接原因,理解 RLVR 才能理解为什么提示器要专门解决"全错组"。

在线策略(On-Policy)与特权上下文(Privileged Context)

在线策略指 rollout 采样与策略梯度损失必须基于同一上下文;特权上下文指训练时输入额外信息(如参考解、提示、解答前缀),测试时移除。SAGE、HiLL 等方法都属于这种范式。

HiLL 强调训练中需要"采样与对数概率都用加提示输入",并通过理论保证这种特权上下文下的学习能迁移到无提示推理场景,这是它区别于普通 prompt engineering 的关键。

研究动机

GRPO 在 RLVR 训练中存在系统性的"优势坍缩"问题:对每道题采样 $G=8$ 条轨迹,当题过难导致推理器全部答错、或者题过易导致推理器全部答对时,组内奖励全为 0 或全为 1,归一化优势全部归零,整批数据里这些样本完全贡献不了梯度,浪费了大量算力。论文指出难题上尤为严重——非退化概率 $s(p;G) \approx Gp$ 在 $p \approx 0$ 时几乎为 0。现有思路分三类:自适应采样(多分配 rollout 给难题)、过滤/剪裁退化组、给输入加"提示/脚手架"让模型拿到混合结果。第三条线最直接,但它有两大未解决的问题:其一,现有提示通常由外部教师或离线规则生成,是固定的,不会随推理器自身能力漂移而调整;其二,即便提示让混合组出现,也不意味着学到的策略能在测试时(无提示)起作用——提示可能把难题变成"太简单"的题,模型只是学会了照抄提示,并未真正学到可迁移的推理能力。

本文的目标是论文提出 HiLL(Hint Learning for Reinforcement Learning),目标是把"提示生成"本身也作为一个可学习的策略,由 hint 后置策略 $H_\phi$ 与推理器策略 $\pi_\theta$ 在 RL 中协同训练。具体目标有三:第一,让提示器实时观察推理器当前的错误样本("失败条件化"),使提示随推理器能力演进而自适应;第二,引入"提示依赖度"(hint reliance)$\rho(\tau;q,h)$ 来量化"去掉提示后正确轨迹是否还可能",并推导出无提示成功概率的下界 $p \ge p_h \cdot \exp(-\rho_c(q,h))$;第三,用迁移加权奖励 $R(q,h) = s(\hat{p}_h;G) \cdot \exp(-\max(\hat{\rho}_c,0)/T)$ 同时鼓励"产生混合组"与"提示依赖低",让提示器学会给出"把题变容易但模型原本就能解出"的提示。

与已有工作不同的是,与已有提示方法相比,HiLL 的切入角度独特:(1) 相对于 SAGE(自生成提示)只在训练初期用一次且离线,HiLL 在每个退化组都现生成并随推理器更新持续调整;(2) 相对于 Scaf-GRPO(外部教师提示)和 LUFFY(注入 DeepSeek-R1 的离线轨迹),HiLL 不依赖任何外部大模型,所有提示由一个 4B 的 Qwen3 提示器策略实时生成并用 GRPO 自己训练出来;(3) 相对于所有已有方法只看"提示后组是否变成混合",HiLL 第一次形式化地定义并优化了"提示依赖度"这一迁移性度量,从理论上证明并实验上验证"低依赖度 $\Rightarrow$ 强迁移 $\Rightarrow$ 无提示准确率提升"这一链条。这是首个把 hint generation 作为在线 co-training 问题、同时用信息论工具显式刻画可迁移性的工作。

核心方法

HiLL 把"给难题加提示"这件事从一个预处理步骤改造成一个与推理器同步训练的在线策略。其整体思路是:训练批次 $B$ 先让推理器 $\pi_\theta$ 正常 rollout $G=8$ 条,识别出全错题集 $\mathcal{I}$;对 $\mathcal{I}$ 里每道题,提示器 $H_\phi$(Qwen3-4B)观察(题 $q$、当前推理器的一条错误 rollout $\tau_k$、参考解 $z^*$),现场生成 $M=4$ 个候选提示 ${h_j}_{j=1}^M$;推理器再对每个候选提示 $q+h_j$ 重采样 $G$ 条轨迹,并按式 (7) 计算每条提示的"信号创建 $\times$ 信号迁移"复合奖励 $R(q,h_j)$;用最高奖励的提示对应的 rollout 替换原退化组,去更新推理器;同时用 $M$ 个候选提示作为提示器的 GRPO 组来更新提示器 $H_\phi$。直觉上,这相当于让一个 4B 小模型担任"家教",家教要看学生哪道题错了、错在哪、参考答案是什么,再给学生一个恰到好处的提示(不是直接给答案,而是给方法),学生按提示重做后再训练;家教自己的"教学水平"也通过 GRPO 持续提升。

核心创新有两点,本质上都和"可迁移性"相关。第一,**失败条件化提示生成**:提示器输入里强制包含一条推理器的错误 rollout $\tau_k$ 和参考解 $z^*$,使提示能针对性纠正当前模型的真实失误,而不是套用通用模板;这是相对于 SAGE(只用参考解)和 Scaf-GRPO(只让外部教师自由发挥)最本质的区别。第二,**提示依赖度作为可迁移代理指标**:定义 $\rho(\tau;q,h) = \log \pi_\theta(\tau|q+h) - \log \pi_\theta(\tau|q)$,刻画"去掉提示后这条正确轨迹还剩多少可能性";证明 Proposition 1:$p \ge p_h \cdot \exp(-\rho_c(q,h))$,即低依赖度意味着无提示场景也能拿到至少 $\exp(-\rho_c)$ 比例的成功率下界;据此构造迁移加权奖励 $R(q,h) = s(\hat{p}_h;G) \cdot \exp(-\max(\hat{\rho}_c,0)/T)$,温度 $T=0.3$ 控制"创建信号"与"信号迁移"的权衡。整篇文章的"killer feature"是把训练开销显式换成了 on-policy rollouts 的语义质量,让每个梯度更新都对应"将来真能用上"的推理能力,而不是只在提示辅助下临时跑通。

方法步骤详情

HiLL 的训练流程严格按 Algorithm 1 执行,每一步可拆成 5 个动作。**步骤 1(识别退化组)**:对批次 $B$ 的每道题用 $\pi_\theta$ 采样 $G$ 条轨迹并算奖励,收集所有 $G$ 条全 0 奖励的题 $\mathcal{I}={q\in B: \sum_{i=1}^G r_i(q)=0}$。**步骤 2(生成候选提示)**:对每个 $q\in\mathcal{I}$,构造提示器输入 $c=(q,\tau_k,z^*)$($\tau_k$ 从全错组随机挑一条,$z^*$ 仅训练可见),让 $H_\phi$ 采样 $M=4$ 条候选提示 ${h_j}_{j=1}^M$,每条最多 1024 token。**步骤 3(评估每条提示)**:先做合法性检查(不能泄漏最终答案、不能超出 10240 token 上限、格式要规范),不合法则 $R(q,h_j)=R_{\text{fail}}=-0.2$;否则推理器在 $q+h_j$ 上重采样 $G$ 条,估 $\hat{p}_h = \frac{1}{G}\sum_i r_i^{(h_j)}$;当 $0<\hat{p}_h<1$ 时再估 $\hat{\rho}_c(q,h_j)=\frac{1}{|C|}\sum_{\tau\in C}\frac{\rho(\tau;q,h)}{|\tau|}$(需要两次 teacher-forced 前向:一次在 $q+h$ 下,一次在 $q$ 下),然后按式 (7) 算 $R(q,h_j)$。**步骤 4(替换与推理器更新)**:选 $h^\star = \arg\max_j R(q,h_j)$,若 $R(q,h^\star)>0$ 则把 $(q,{\tau_i})\leftarrow(q+h^\star,{\tau_i^{h^\star}})$ 作为新输入更新推理器,否则保留原样;推理器用 DAPO 风格 GRPO($\epsilon_{\text{low}}=0.2,\epsilon_{\text{high}}=0.28$,禁用 KL)按式 (8) 训练 500 步、batch 128、lr $10^{-6}$。**步骤 5(提示器更新)**:对每道 $q\in\mathcal{I}$,$M$ 个候选提示作为提示器 GRPO 组,按式 (9) 用组内归一化优势更新 $H_\phi$,让提示器学会避开无效提示、倾向于低依赖度且能制造混合组的提示。两步更新在同一 GPU 节点上串行执行,Ray 协调调度避免显存冲突。

技术新颖性

技术新颖性体现在三个层面。**理论层面**:Proposition 1 给出"无提示成功率 $\ge$ 提示后成功率 $\times \exp(-\text{提示依赖度})$",第一次把提示设计的优化目标与无提示泛化性能用严格的 KL 散度关系联系起来,$p$ 与 $p_h$ 的对数比加上正确轨迹分布的 KL 就等于平均依赖度,因此降低依赖度本质上等于缩小两个条件分布之间的差距——这是把 prompt engineering 放进信息论框架的一次尝试。**算法层面**:把"hint generation"从离线启发式(partial solution、手写 scaffold、DeepSeek-R1 蒸馏)改造为可微可优化的策略 $H_\phi$,并通过共享 FSDP 分片与 Ray 同节点调度让 hinter/reasoner 共用显存,避免了通常 RL pipeline 中两倍模型的开销。**奖励设计层面**:双因子乘积奖励 $s(\hat{p}_h;G)\cdot\exp(-\max(\hat{\rho}_c,0)/T)$ 把"信号创建"和"信号迁移"做成可独立调温的模块,$T$ 越小越保守(依赖度惩罚越重),$T$ 越大越激进(更看重产生混合组),Figure 3 显示 $T\in[0.2,0.4]$ 内都能超过无迁移加权版本。这种"模块化奖励 + 温度调节"的设计也提示后续工作可以在此基础上叠加其他目标(如多样性、推理步数控制)。

Overview of our HiLL framework
Figure 1: Overview of our HiLL framework

实验结果

实验在 Llama-3.2-3B-Instruct 与 Qwen2.5-7B-Instruct 两个推理器上、$8\times B200$ GPU 上跑 500 步训练,核心结论有四点。**第一,HiLL 在 8 个基准上一致超过 GRPO 与所有提示类基线**(Table 1)。在 Llama-3.2-3B 上,HiLL 域内 6 题平均从 GRPO 的 21.9 提升到 24.6(+2.7),域外 GPQA/MMLU-Pro 平均从 33.1 提升到 35.3(+2.2);在更强的 Qwen2.5-7B 上,HiLL 域内平均 44.2(GRPO 41.1,SAGE 42.3,HiLL w/o TW 42.7),域外平均 51.0(GRPO 47.4,SAGE 48.6),绝对提升仍有 2-3 个百分点,证明"适应性 + 迁移感知"提示对强弱模型都有效。**第二,迁移加权是关键**。HiLL w/o TW(去掉 $\exp(-\hat{\rho}_c/T)$)相比 HiLL 在 Llama 上掉 0.9(24.6→23.7)、在 Qwen 上掉 1.5(44.2→42.7),说明仅靠"产生混合组"还不够,必须显式压低提示依赖。**第三,依赖度确实被压低且不损失信号创建**(Figure 2)。训练全程 HiLL 的 hint reliance 都显著低于 HiLL w/o TW,同时 all-incorrect ratio 仍维持在 GRPO 的低水平(Llama 上从 ~0.7 降到 ~0.3),证明两者可以兼得。**第四,温度 $T$ 控制 trade-off**(Figure 3)。$T=0.3$ 是平衡点;$T=0.2$ 偏保守(transfer 高、creation 低),$T=0.4$ 偏激进,三档都优于无迁移加权。**第五,提示风格确实变简洁**(Figure 4 + Table 2)。HiLL 在 Llama/Qwen 上平均提示长度从 HiLL w/o TW 的 79/75 词降到 41/29 词,每条提示含数学表达式从 5.6/2.7 降到 2.1/1.6——提示器自动学会了"给方法不给计算"的教学风格,例如对同一道参数化消参题 HiLL 只说"express in terms of parameter $a$ then eliminate"(18 词),HiLL w/o TW 则直接列出坐标、方程组(108 词)。代价是训练耗时:Llama 上 wall-clock 是 GRPO 的 ~3.8x、Qwen 上 ~2.6x(SAGE 同条件 2.3x),作者认为这是"把算力花在原本零信号的题上"的可接受交换。

Main results on in-distribution and out-of-distribution benchmarks
Table 1: Main results on in-distribution and out-of-distribution benchmarks
Example hints generated by HiLL and HiLL w/o TW on the same question
Table 2: Example hints generated by HiLL and HiLL w/o TW on the same question
All-incorrect ratio and hint reliance over training steps
Figure 2: All-incorrect ratio and hint reliance over training steps
Effect of transfer temperature $T$ on signal creation, signal transfer, and in-distribution accuracy
Figure 3: Effect of transfer temperature $T$ on signal creation, signal transfer, and in-distribution accuracy
Hint length and math expressions per hint
Figure 4: Hint length and math expressions per hint
查看结构化数据
任务指标本文基线提升
Llama-3.2-3B 域内平均(AIME24/25 + AMC23 + MATH-500 + Minerva + Olympiad) Average@16 准确率 24.6 GRPO 21.9 / SAGE 23.9 / HiLL w/o TW 23.7 +2.7 vs GRPO,+0.7 vs SAGE,+0.9 vs HiLL w/o TW
Llama-3.2-3B 域外平均(GPQA-Diamond + MMLU-Pro) Average@16 准确率 35.3 GRPO 33.1 / SAGE 34.0 / HiLL w/o TW 34.5 +2.2 vs GRPO,+1.3 vs SAGE,+0.8 vs HiLL w/o TW
Qwen2.5-7B 域内平均(AIME24/25 + AMC23 + MATH-500 + Minerva + Olympiad) Average@16 准确率 44.2 GRPO 41.1 / SAGE 42.3 / HiLL w/o TW 42.7 +3.1 vs GRPO,+1.9 vs SAGE,+1.5 vs HiLL w/o TW
Qwen2.5-7B 域外平均(GPQA-Diamond + MMLU-Pro) Average@16 准确率 51.0 GRPO 47.4 / SAGE 48.6 / HiLL w/o TW 49.2 +3.6 vs GRPO,+2.4 vs SAGE,+1.8 vs HiLL w/o TW
AIME24(Llama-3.2-3B) Average@16 准确率 8.5 GRPO 6.7 / SAGE 9.2 +1.8 vs GRPO,SAGE 在 AIME24 上略高 0.7
AIME25(Qwen2.5-7B) Average@16 准确率 15.3 GRPO 13.5 / SAGE 12.5 / HiLL w/o TW 14.4 +1.8 vs GRPO,+2.8 vs SAGE,+0.9 vs HiLL w/o TW
All-Incorrect 比例(Llama-3.2-3B 训练 500 步均值) 退化组占批次比例 约 0.3 GRPO 约 0.65 退化组大幅减少,GRPO 重新学到信号的题数显著增加
训练墙钟时间倍率 相对 GRPO 的每步耗时倍数 Llama 3.8x / Qwen 2.6x SAGE Qwen 2.3x 比 SAGE 略高,但换来 1.9-2.4 个百分点的域内准确率提升

局限与改进

作者在正文中坦率承认了三点局限。第一,**计算开销显著**:在难题多的 Llama-3.2-3B 上每步耗时是 GRPO 的 3.8x,主要来自 $M=4$ 个候选提示 $\times$ $G=8$ 条重 rollout + 两次 teacher-forced 前向估计依赖度;这套 pipeline 严重依赖 Ray 同节点共置两个 FSDP 分片模型,一旦 hinter/reasoner 必须放在不同节点就会爆炸式增长。第二,**提示器参数量与训练稳定性**:用 Qwen3-4B 当 hinter 需要额外显存与训练开销,且 hinter 也用 GRPO 训练,本身也存在优势坍缩问题(候选提示要么全好要么全差),论文没有给出 hinter 训练的退化曲线。第三,**依赖度估计的方差**:$\hat{\rho}_c$ 只在正确轨迹上取平均,组大小 $G=8$ 下若正确轨迹数很少(典型 1-4 条),估计值噪声很大,可能让 hinter 奖励信号抖动,论文没有给出置信区间或多次估计平均。我们自己观察到三点额外的隐患:(a) Proposition 1 的下界虽然数学上成立,但实际 $p$ 是否真的能逼近 $p_h \exp(-\rho_c)$ 取决于 KL 项是否小,论文未在实验中直接验证这个 KL 散度;(b) hinter 永远见不到 $\mathcal{I}$ 之外的题(也就是"简单题"),泛化到测试分布时可能给出过于针对难题的提示模式;(c) 域外(GPQA、MMLU-Pro)提升的机制不够清晰,论文没有拆解"是推理能力真正涨了还是题型匹配度涨了"。

独立分析的弱点

独立审视后我们认为有三个具体弱点值得改进。**弱点 1:候选提示评估的"全或无"瓶颈**——目前对每条候选提示要么成功要么记 $R_{\text{fail}}=-0.2$,这意味着当 hinter 整体探索不充分时,奖励方差会爆炸,hinter 难以学到稳定策略;改进方向是用 hindsight 经验回放或 hindsight relabeling:把"已经找到好提示的题"的失败候选也赋予部分奖励,反向训练 hinter 知道"哪种提示不该发"。**弱点 2:依赖度只在正确轨迹上估计**——这其实忽略了"错误轨迹的依赖度"也可能影响训练,比如某条提示让推理器在 $q+h$ 下产生高 reward 但完全靠提示走捷径的错误轨迹,这种轨迹进入 GRPO 组时仍会拉低其他正确轨迹的优势;改进方向是同时约束 $\rho(\tau;q,h)$ 在 $\tau\in C\cup C'$(正确+错误)上的某种均值或最大值,把"对提示依赖的整体性"也纳入奖励。**弱点 3:hinter 自身没有退化保护**——当某批次所有候选提示都是无效(泄漏答案、超长),hinter GRPO 组优势归零,hinter 完全学不到东西;改进方向是给 hinter 一个轻量"参考提示语料"做 offline warm-start,或者在 hinter 训练里加 KL 锚定项让它不至于漂移到极端输出格式。

未来方向

作者没有单独列 future work,但基于成果可以延伸出几条值得探索的方向。第一,**层级化提示**——目前提示只能改单条候选,可以扩展为 hint tree:先给出方法层提示,若推理器仍全错,再给出更细的步骤提示,形成 curriculum-like 的多轮 hinter。第二,**多模态 hinter**——把 hint 从纯文本扩展到图示、公式块、甚至是中间工具调用(calculator/SymPy),让提示空间更丰富,$\rho(\tau;q,h)$ 自然延伸到"多模态依赖度"。第三,**领域迁移与泛化**——把 HiLL 框架从数学拓展到代码、多步推理、Agent 任务,关键是设计各领域下"依赖度"的具体形式,例如代码任务可以定义为 $\rho(\tau;q,h)=\log\pi(\tau|q\oplus h) - \log\pi(\tau|q)$ 但用 pass@k 或单元测试通过率作可验证奖励。第四,**hinter 蒸馏**——4B Qwen3 作 hinter 是临时方案,未来可以用一个 70B 强模型当 teacher hinter,给小模型 hinter 做 SFT/distillation,让整套 pipeline 跑得更便宜。第五,**理论扩展**——Proposition 1 只给了下界,未来可以推出"提示数量 $M$、batch 大小与最终无提示性能"的有限样本收敛率,把 HiLL 从经验方法升级为有理论保证的算法。

复现评估

复现友好度整体不错,但需要相当算力。**代码**:作者明确在摘要里给出 GitHub 链接 https://github.com/Andree-9/HiLL,且声明实现基于 Liao et al. (SAGE) 的官方 codebase(verl [25] 框架 + vLLM [7] rollout)。**数据**:使用与 SAGE 一致的 15k OpenR1-Math-220k 子集(来自 NuminaMath 1.5 + ground-truth 答案与参考解),无需私有数据。**训练细节齐全**:500 步、batch 128、$G=8$、$M=4$、lr $10^{-6}$、max prompt 2048、max response 8192、hinter max prompt 10240 / response 1024、$T=0.3$、$R_{\text{fail}}=-0.2$、$\epsilon_{\text{low}}=0.2,\epsilon_{\text{high}}=0.28$、无 KL、Clip-Higher(DAPO 风格)、checkpoint 每 50 步评估一次取最优,所有超参在论文第 5 节末尾都给齐。**模型**:reasoner 用 Llama-3.2-3B-Instruct 与 Qwen2.5-7B-Instruct,hinter 用 Qwen3-4B-Instruct,均为公开权重。**评估**:温度 0.6、top-p 0.95、max response 8192、Average@16、8 个基准全部开源。**算力门槛**:8 块 B200 GPU 单次完整训练,在 Llama 上 wall-clock 大约是 GRPO 的 3.8 倍——若想完整复现所有表格(7 个基线 $\times$ 2 个模型 + 3 个温度消融 + 训练曲线),保守估计需要 200-400 GPU $\times$ hours。**评估难度**:依赖度估计的两次 teacher-forced 前向会让单步实现容易写错,建议严格按 Algorithm 1 一步步对齐;hinter 的 prompt template 在 Appendix B,是复现成败的关键。