ARISE: 分层强化学习中具有内在技能演化的智能体推理 ARISE: Agent Reasoning with Intrinsic Skill Evolution in Hierarchical Reinforcement Learning
通过策略驱动的技能选择与分层奖励实现数学推理能力的持续进化
前置知识
强化学习 (RL)
一种机器学习范式,智能体通过与环境交互,获得奖励信号来优化策略,目标是最大化累积奖励。在数学推理中,智能体通过尝试解题,根据答案正确性获得二元奖励 (0 或 1)。
本文的整个框架建立在强化学习之上,理解奖励信号如何驱动策略优化是理解分层奖励机制 (r_0, r_1, r_2) 的基础。
GRPO (Group Relative Policy Optimization)
一种无评论家版本的 PPO 算法,通过在同组 G 个 rollout 中归一化奖励来估计优势,消除了对价值函数的依赖。计算方式为 A_hat_i = (r_i - mu_G)/(sigma_G + epsilon),其中 mu_G 和 sigma_G 是组内奖励的均值和标准差。
ARISE 以 GRPO 为基础训练框架,并扩展其奖励为分层结构 (R in {0, 1, 2}) 以引导技能使用。
策略梯度 (Policy Gradient)
一种直接优化策略参数的方法,通过计算期望奖励对参数的梯度来更新策略。公式为 gradient_theta J(theta) = E[gradient_theta log pi_theta(a|s) A_hat(s, a)],其中 A_hat 是优势估计。
本文的核心创新是让策略梯度直接塑造技能选择偏好,理解梯度如何通过共享策略传播是关键。
条件对数概率 (Conditional Log-Probability)
给定上下文后生成特定序列的概率的对数值,记为 log p(y|x) = sum_t log p(y_t|x, y_{<t})。本文用它来评分查询与技能的相关性 s_k = sum_l log pi_theta(m_k^{(l)}|q, m_k^{(<l)})。
这是 ARISE 实现策略驱动技能选择的核心机制,替代了传统的外部检索器。
研究动机
现有的强化学习方法在解决数学推理问题时存在系统性缺陷。以 GRPO 及其变体(Dr.GRPO、DAPO、GSPO)为代表的方法将每个问题实例视为孤立的训练样本,一旦 rollout 完成,解题过程中生成的成功推理策略就会被丢弃,而不是被保留和积累供未来使用。这种机制导致智能体在不同问题间无法传递经验,造成冗余探索,每次遇到类似问题都需要重新学习相同的推理模式。例如在 DeepScaleR 数据集上,训练约 40K 个问题后,GRPO 在 Omni-MATH 上的平均准确率仅达到 23.9%(Qwen3-4B),明显低于人类复用推理策略的能力。
本文的目标是本文的目标是构建一个能够持续积累和复用推理策略的强化学习框架,使智能体在训练过程中自动构建、维护和优化一个技能库。该框架需要解决三个核心挑战:(1) 如何让智能体自主生成高质量的技能文档,而不是依赖人工标注;(2) 如何确保技能选择与策略优化紧密耦合,避免外部检索器切断梯度传播;(3) 如何设计奖励机制以鼓励技能的合理使用,防止过度依赖或滥用。最终目标是让策略与技能库协同进化,提升在分布外任务上的泛化能力。
与已有工作不同的是,本文的独特切入点是将技能库从外部静态资源转化为智能体状态的内在组件。现有的记忆增强方法(如 EvolveR、SimpleMem、SkillRL)将技能检索委托给外部模型,技能库更新独立于 RL 目标,切断了策略改进与库丰富之间的反馈循环。ARISE 通过 Manager-Worker 分层架构和共享策略 pi_theta,让技能选择、生成和策略优化在统一的 RL 目标下协同进行,策略梯度能够端到端地塑造技能选择偏好。此外,本文引入分层奖励 R in {r_0, r_1, r_2},明确区分使用技能且正确 (r_2)、不使用技能但正确 (r_1) 和错误 (r_0) 三种情况,在组内相对优势机制下引导策略主动利用有用技能,这与仅使用二元奖励的现有方法有本质区别。
核心方法
ARISE 采用 Manager-Worker 分层架构,由单一共享策略 pi_theta 驱动所有组件。Manager 负责技能选择(通过 Download 通道检索相关技能)和技能生成(通过 Upload 通道将成功轨迹蒸馏为技能文档),Worker 基于选定技能生成解决方案。技能库采用两层 Cache-Reservoir 架构:Cache (M_c,容量 10)作为活跃池用于选择,Reservoir (M_r,容量 100)存档备用技能,通过五个管理操作(ADD、UPDATE、EVICT、LOAD、DELETE)维护质量。训练分两阶段:Phase I 用标准 GRPO 和二元奖励构建基础策略并静默填充技能库;Phase II 激活完整分层流程,Manager 开始选择技能,奖励切换为分层结构 R in {0, 1, 2},策略优化、技能选择和库丰富同步进行。这种设计使技能库成为智能体状态的内在组成部分,与策略协同进化。
核心创新是策略驱动的技能选择机制,使用策略自身的条件对数概率评分技能相关性,替代了传统的外部检索器。对于每个候选技能 m_k,计算 s_k = sum_{l=1}^{|m_k|} log pi_theta(m_k^{(l)}|q, m_k^{( 0} 生成结构化技能文档,确保技能来源于具体成功经验而非空泛归纳。
方法步骤详情
ARISE 的完整训练流程分为两个阶段。Phase I (前 500 步):(1) 初始化 Cache 包含 5 个种子技能(如 equation setup、case enumeration),Reservoir 为空。(2) 对每个查询 q ~ D,采样 G=8 条轨迹 tau_i ~ pi_theta^old(|q),计算二元奖励 r_task^i in {0, 1} 和组相对优势 A_hat_i = (r_i - mu_G)/(sigma_G + epsilon)。(3) 收集正向优势轨迹 tau^+,执行技能生成 rollout O_{G+1},生成新技能 m_new ~ pi_theta(|q, tau^+)。(4) 更新 M_c 和 M_r:插入新技能,执行 UPDATE (通过 EMA 更新选定技能的效用)、EVICT (将最低效用技能移至 Reservoir)、LOAD (高效用 Reservoir 技能与低效用 Cache 技能交换)、DELETE (移除无用 Reservoir 技能)。(5) 通过 L_GRPO(theta) 更新策略参数。注意此阶段技能选择未激活,rollouts 不使用技能。Phase II (501 步开始):(1) Manager 对每个查询,为所有 Cache 技能计算条件对数概率得分 s_k(通过 Eq. 2),转换为选择分布 p_k,结合 epsilon-greedy (epsilon=0.1) 采样技能 z_t。(2) 如果 max_k p_k >= delta (置信门阈值 0.35),将选定技能以 SKILL:{...skill JSON...} 前缀形式注入 prompt,否则不注入。(3) Worker 基于增强 prompt 生成 G 条轨迹 tau_i ~ pi_theta^old(|q, m_tilde_{z_t})。(4) 计算分层奖励 R_i = r_task^i + r_skill^i in {0, 1, 2},其中 r_skill^i = 1 仅在使用技能且正确时授予。(5) 执行 O_{G+1} 技能生成和库维护(同 Phase I)。(6) 通过 L_GRPO(theta) 更新策略,现在重要性采样比率为 rho_{i,l} = pi_theta(x_{i,l}|q, m_tilde_{z_t}, x_{i,<l}) / pi_theta^old(x_{i,l}|q, m_tilde_{z_t}, x_{i,<l})。
技术新颖性
ARISE 的技术新颖性体现在三个层面。首先,技能选择的内在化:通过策略自身的条件对数概率评分替代外部检索器,使优势信号能够端到端地塑造选择偏好。实验显示当同组包含 R=2 和 R=1 轨迹时,组相对优势会为 R=1 分配负值(当 mu_G > 1),actively down-weighting 正确但未使用技能的轨迹,这是一种强化的 skill adoption 信号。其次,两层 Cache-Reservoir 架构平衡了效率与持久性:Cache 限制在 10 个技能控制计算开销(每个 query 约 0.3 秒,占总时间 4.7%),Reservoir 保留 100 个技能防止过早丢弃。LOAD/EVICT 机制确保库饱和后改进主要来自更好的选择而非持续扩张,实验显示库大小在 1500 步后饱和但准确率持续上升。最后,技能生成的约束机制:220 字符硬限制确保 log-probability 评分反映语义相关性而非长度偏差,JSON schema 统一格式(skill name、problem type、key insight、method、check)保证技能结构一致,实验显示这种蒸馏方式成功率在 warm-up 后超过 85%。相比 EvolveR(7.9% token overhead,0.7-point Omni-MATH gain),ARISE 仅 3.0% overhead 但获得 2.9-point gain,在 accuracy-gain-versus-overhead 空间达到 Pareto 最优。
实验结果
在两个基模型和七个基准测试上,ARISE 一致优于 GRPO 家族和记忆增强方法。在 Qwen3-4B-Instruct-2507 上,ARISE 在分布内竞赛基准(AMC 23、AIME 24、AIME 25)上的平均提升为 2.2 个百分点,相对于最强 GRPO baseline (DAPO),在 Omni-MATH 的四个子领域(代数、数论、组合、几何)上平均提升 2.9 个百分点,代数领域达到 3.5 个百分点提升(从 33.5% 到 37.0%),这表明技能库积累了更多可复用的推理模式而非仅强化训练集中的模式。在较弱的 Phi-4-mini-instruct-3.8B 上,ARISE 依然获得一致提升,Omni-MATH 平均提升 2.0 个百分点(从 9.4% 到 11.4%),确认分层技能机制即使在正向训练信号稀疏时仍然有效。消融实验确认分层奖励是 skill adoption 的主要驱动力:使用二元奖励导致技能利用率从 73% 降至 31%,AIME 24 准确率从 56.4% 降至 54.7%。随机技能注入保持高利用率(82%)但准确率下降(AIME 24 从 56.4% 降至 55.1%),证明不匹配技能会引入误导性上下文。移除 O_{G+1} rollout 冻结库在 24 个种子技能,Omni-MATH 受影响最大(从 28.0% 降至 26.0%),因为静态启发式无法覆盖奥林匹克级多样性。移除置信门影响最小但利用率升至 91%,确认其作为边缘情况噪声过滤器的角色。训练动态显示库大小和准确率在 Phase II 协同上升,库大小在 1500 步后饱和(约 200-220 技能)但准确率持续改进,表明后期收益来自更好的选择而非库扩张。计算成本增加 16.6%(从 51.2s 到 59.7s/步),主要由技能生成 O_{G+1} (4.8s, 9.4%) 和技能评分 (2.4s, 4.7%) 贡献。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| AMC 2023 (分布内竞赛数学) | Pass@1 Accuracy (%) | 75.4 | 74.2 (DAPO) | +1.2 |
| AIME 2024 (分布内竞赛数学) | Pass@1 Accuracy (%) | 56.4 | 55.3 (DAPO) | +1.1 |
| AIME 2025 (分布内竞赛数学) | Pass@1 Accuracy (%) | 48.3 | 47.6 (DAPO) | +0.7 |
| Omni-MATH Algebra (分布外奥林匹克) | Pass@1 Accuracy (%) | 37.0 | 35.2 (DAPO) | +1.8 |
| Omni-MATH Number Theory (分布外奥林匹克) | Pass@1 Accuracy (%) | 27.2 | 25.7 (DAPO) | +1.5 |
| Omni-MATH Combinatorics (分布外奥林匹克) | Pass@1 Accuracy (%) | 17.6 | 16.3 (DAPO) | +1.3 |
| Omni-MATH Geometry (分布外奥林匹克) | Pass@1 Accuracy (%) | 25.5 | 23.9 (DAPO) | +1.6 |
| Omni-MATH Average (分布外平均) | Pass@1 Accuracy (%) | 26.8 | 23.9 (GRPO) | +2.9 |
| Phi-4-mini Omni-MATH Average (弱基模型) | Pass@1 Accuracy (%) | 11.4 | 10.5 (DAPO) | +0.9 |
局限与改进
作者承认的局限性包括:(1) 仅在数学推理领域评估,尚未扩展到多工具智能体任务或代码生成,其中可复用技能形式可能不同(可执行过程而非推理启发式);(2) 技能文档限制在 220 字符可能丢失某些细粒度推理细节;(3) 技能效用评估仅基于 EMA 奖励,未充分利用信息增益指标 IG(m_k) = E_{tau^+}[sum_t log pi_theta(tau_t^+|m_k, q, tau_{ 0 的轨迹,在训练早期正向稀少时可能生成质量不稳定的技能。
独立分析的弱点
独立分析的弱点包括:(1) 技能压缩损失:220 字符硬限制强制技能高度抽象,可能丢失领域特定细节。改进方向是可变长度技能或分层技能结构(宏技能调用微技能),但需平衡 log-probability 评分的公平性。(2) EMA 效用偏差:当前效用更新 u_{z_t} <- beta u_{z_t} + (1-beta) R_t 主要反映累积奖励而非因果贡献,高分技能可能因被选中解决简单问题而非实际提升性能。改进方向是结合信息增益指标 IG(m_k) 进行加权效用估计,或在 LOAD/EVICT 中增加因果约束。(3) 种子技能有限性:仅 5 个种子技能初始化可能限制早期探索,特别是在数学基础较弱的模型上。改进方向是领域自适应种子技能(针对不同数学领域初始化不同种子集)或课程式种子注入(随训练进展逐步引入更专业种子)。(4) 置信门静态阈值:delta=0.35 是全局常数,不适应问题难度分布。改进方向是动态阈值 delta(q) 基于查询复杂性(如 token 数量、问题类型)或历史成功率自适应调整,或学习门控网络 pi_gate(q)。(5) 技能生成依赖正向稀疏性:训练早期 A_hat_i > 0 轨迹稀少时,O_{G+1} 可能生成质量不稳定技能。改进方向是早期使用种子轨迹或更宽松的正向标准,或引入技能质量评估模块过滤低质技能。
未来方向
作者提出的未来工作方向包括扩展 ARISE 到多工具智能体任务和代码生成,其中可复用技能形式可能是可执行过程而非推理启发式。基于成果的可延伸方向:(1) 多技能注入:当前每次选择一个技能,可扩展为技能组合,使用注意力机制或软投票融合多个技能,类似 few-shot prompting 中的 example selection。(2) 跨任务技能迁移:在多个任务间共享技能库,通过元学习框架让技能成为可迁移的抽象,而非任务特定。(3) 技能层次化:构建技能 DAG(有向无环图),高阶技能调用低阶技能,形成推理技能的层次结构,类似于人类专家知识体系。(4) 技能质量预测:训练独立的技能质量评估器,基于技能文档的语义特征、历史效用和 IG 指标预测未来贡献,用于更智能的 LOAD/EVICT。(5) 自适应管理策略:用强化学习学习五个管理操作的策略而非固定顺序,使库管理成为可优化组件。(6) 推理时技能选择优化:当前推理时用 epsilon=0 (纯 exploitation),可引入推理时探索或上下文自适应选择,基于查询不确定度动态调整选择策略。(7) 与其他记忆机制结合:ARISE 侧重技能级抽象,可与轨迹级记忆(如 EvolveR)或参数级记忆(如 LoRA 适配器)互补,形成多粒度记忆架构。
复现评估
复现评估:(1) 数据集:DeepScaleR 公开可用,包含约 40K 问题-答案对来自 AMC、AIME、MATH 和 OlympiadBench。评估基准 AMC 2023、AIME 2024&2025 和 Omni-MATH (4428 问题) 均公开。(2) 基模型:Qwen3-4B-Instruct-2507 和 Phi-4-mini-instruct-3.8B 公开可访问,但具体 checkpoint 版本未明确说明。(3) 算力需求:8x A100-80GB GPU,训练 2000 步,每步约 59.7 秒,总时长约 33 小时。(4) 代码和配置:论文未明确声明代码开源,附录提供了详细超参数(表 3)和完整训练流程(附录 A.2),但未提供代码仓库链接。训练细节如 prompt 模板、技能验证 pipeline、库管理实现等均有文字描述,但缺乏实际代码。(5) 复现难度:中等偏高。虽然论文提供了详细实现细节和超参数,但两层 Cache-Reservoir 管理的五个操作顺序、技能生成验证流程、条件 log-probability 评分实现等存在较多工程细节,完全复现需要仔细实现。此外,评估报告 32 次独立运行的平均 Pass@1,复现需要大量计算资源来获得可比的统计显著性。(6) 额外考虑:论文未提供生成技能的示例种子集、验证 pipeline 的具体错误处理逻辑、以及动态采样过滤零方差组的精确实现,这些细节可能导致实现差异。
论文图表
这个算法描述了 ARISE 的完整训练流程:输入包括数据集 D、初始策略 pi_theta_0、组大小 G、预热步数 N_w、种子技能 M_seed。初始化 M_c = M_seed, M_r = empty。Phase I(前 1-8 行):构建基础策略并填充技能库。对于每个 q ~ D,采样 G 条轨迹 tau_i,计算 r_task^i 和 A_hat_i,收集 tau^+,生成新技能 m_new ~ pi_theta(|q, tau^+),更新 M_c, M_r(Eq. 4),通过 L_GRPO(theta) 更新 theta。Phase II(第 10-18 行):激活完整分层流水线。对于每个 q ~ D,通过 Eq. 2, 3 评分并选择技能 z_i,采样 G 条轨迹 tau_i ~ pi_theta^old(|q, m_tilde_{z_i}),计算 R_i in {0, 1, 2} 和 A_hat_i,更新 u_{z_i} <- beta u_{z_i} + (1-beta) R_i,收集 tau^+,生成新技能 m_new 并更新 M_c, M_r,通过 L_GRPO(theta) 更新 theta。
这个算法对理解论文的实现至关重要,它提供了 ARISE 两阶段训练的伪代码描述,明确区分了 Phase I 和 Phase II 的操作差异。
这个图展示了用于技能生成 rollout O_{G+1} 的 prompt 模板:You are a skill distiller for math reasoning. Given one question and trajectories from the same rollout group, summarize ONE reusable skill. Question: {question} Group trajectories: [SUCCESS #1] {clipped response 1 (<=400 chars)} [SUCCESS #2] {clipped response 2 (<=400 chars)} Output MUST be a valid JSON object and nothing else (no markdown/code fences). Schema: {"skill name", "problem type", "key insight", "method", "check"} Rules: - Be generic and transferable; do not copy specific numbers. - Keep the whole skill within 220 characters. - key insight is the most important field. - method must contain 2--3 concise steps. - Focus on improving correctness, not style.
这个图对理解论文的技能生成机制至关重要,它展示了 O_{G+1} rollout 的 prompt 模板,确认技能生成是基于具体成功轨迹的结构化总结。