PACEvolve++:改进进化搜索代理的测试时学习 PACEvolve++: Improving Test-time Learning for Evolutionary Search Agents
通过advisor模型和相位自适应RL,改进进化搜索代理在昂贵评估场景下的策略适应能力
前置知识
进化搜索代理
一种通过重复的提议-评估-选择循环来改进程序的系统。给定初始程序$p_0$、评估器$E: P \rightarrow R$和策略$\pi_\theta$,代理生成候选修改,评估它们,并在发现更高评分的后代时更新当前解决方案。这类系统包括FunSearch、AlphaEvolve等,在算法发现、符号回归等任务中表现出色。
本文的核心框架基于进化搜索代理,理解其基本工作原理(提议、评估、选择循环)和挑战(固定策略、缺乏适应)是理解本文改进的基础。
测试时训练
在推理过程中动态修改模型参数以提升性能的方法。与传统训练后固定的模型不同,测试时训练允许模型根据特定任务或数据分布的反馈进行调整。在进化搜索场景中,模型可以在搜索过程中从生成的候选和评估结果中学习,不断优化其决策策略。
PACEvolve++的核心创新就是在进化搜索过程中进行测试时策略适应,理解测试时训练的概念有助于理解本文为何要实时调整advisor模型。
Advisor模型范式
将模型的推理能力和实现能力分离的架构设计。小型可训练advisor模型负责战略决策(如想法生成、新颖性评估、假设选择),而更强的frontier模型负责将选中的假设转化为可执行代码。这种分离使得搜索特定信号主要集中在决策层面,而非实现层面。
这是本文的关键架构创新,理解advisor模型与实现模型的分工有助于理解为什么这种方式能稳定训练并提升搜索效率。
GRPO和PKPO
GRPO (Group Relative Policy Optimization) 使用分组基线替代价值函数,给定奖励$\{R_1, ..., R_n\}$,样本$i$的advantage为$\hat{A}_{GRPO}^i = \frac{R_i - \bar{R}}{\sigma_R + \epsilon_{num}}$。PKPO (Pass@k Policy Optimization)针对pass@k目标,通过组合权重构造无偏梯度估计器,使用SLOO$k-1$估计器将贡献信号转为边际贡献。
本文的相位自适应方法结合了GRPO和PKPO的优点,理解这两种算法的优缺点(GRPO在小批量、低方差下的不稳定性,PKPO过早收敛的问题)是理解本文改进的基础。
研究动机
现有进化搜索系统主要依赖固定的、提示触发的策略来采样下一个候选。这种限制在实际工程和研究任务中尤为严重,其中评估成本高昂,进步取决于学习任务特定的搜索动态。例如,在推荐系统设计、MoE负载平衡和蛋白质适应性外推等任务中,评估每个候选可能需要GPU训练、大规模模拟或系统基准测试,评估预算以分钟和小时为单位。在这种昂贵评估场景下,rollout组规模必然较小,而现有的RL方法(如ThetaEvolve和TTT-Discover)仍然依赖较大的rollout批次,通常每个训练步生成512个候选,使得一个强化学习步骤的成本可能超过整个样本高效的进化运行。此外,固定策略虽然可以通过上下文条件化历史,但它不会将由此产生的搜索反馈内化为稳定的决策偏好。
本文的目标是本文的目标是设计一个advisor模型强化学习框架,用于进化搜索代理的测试时策略适应。具体而言,要解决如何在候选评估成本高昂、搜索前沿非平稳的情况下,使RL目标保持稳定;如何根据搜索阶段调整信用分配的几何结构,在早期探索时鼓励广泛搜索方向,在后期精炼时关注frontier改进;如何将战略搜索决策与实现解耦,使强化学习专注于决定下一个评估什么,同时利用frontier编码模型进行实现。
与已有工作不同的是,本文的独特切入角度在于认识到进化搜索的不同阶段需要不同的训练信号。早期在探索阶段,候选在机制和质量上差异显著,中心化分数差异提供有用的密集反馈;后期在精炼阶段,候选通常是已经强有力父代的近邻变体,关键事件是响应是否改变best-of-k前沿。现有方法(GRPO、熵目标、PKPO)都使用单一固定的信用分配方式,无法适应这种动态变化。GRPO在后期奖励压缩时会放大小的数值差异为过大的梯度幅度,熵目标会过度集中在奖励异常值上,PKPO会过早收敛导致失去多样性。本文提出的相位自适应方法动态调整信用分配来源,与进化搜索动力学的对数递减奖励结构保持一致。
核心方法
PACEvolve++的整体思路是在进化搜索过程中训练一个advisor模型来学习任务特定的搜索先验。框架包含两个核心组件:advisor模型训练和搜索动力学感知策略优化。Advisor模型负责进化搜索中的战略决策部分,包括想法生成、新颖性分类和假设选择,而更强的frontier实现模型将选中的假设转化为具体的代码修改。这种分离将尝试什么与如何实现解耦,使强化学习专注于战略推理,同时保持强大的编码能力。搜索动力学感知策略优化则根据搜索阶段动态调整训练信号,早期使用分组相对反馈鼓励广泛探索,后期使用frontier贡献信号支持稳定精炼,整个过程通过相位自适应的advantage计算实现。
核心创新点是双重的架构和方法创新。架构上,采用advisor模型范式将战略搜索决策与代码实现解耦,这与标准的进化编码框架形成鲜明对比,后者通常使用同一个模型既决定尝试什么又实现结果。这种解耦在研究和工程任务中尤为重要,因为实现失败可能源于复杂代码库、集成细节和系统约束,而搜索特定信号主要在于判断哪个假设是新颖、可行且可能改进进化前沿的,与模型的一般编码能力分离。方法上,设计相位自适应强化学习算法,根据搜索动力学动态调整信用分配:在探索阶段使用原始分组相对基线$\hat{A}_{G_i} = R_i - \bar{R}$保持密集的组内信用分配;在精炼阶段使用pass@k边际贡献信号$\hat{A}_{top-k_i}^{SLOO}$,该信号不简单模仿最高评分rollout,而是根据候选对前沿改进的贡献分配信用。
方法步骤详情
完整的工作流程如下:首先,在每个迭代中,advisor模型条件化于父程序和搜索历史,生成和选择假设。其次,frontier实现模型将这个假设转化为具体的代码编辑。然后,任务特定的评分器评估结果,将结果整合到进化种群中。接着,对该rollout批次进行优化,更新advisor参数,并将更新后的advisor参数同步到rollout工作器,用于下一次迭代。在训练过程中,采用相位自适应advantage计算:给定rollout迭代$t$中的rollout组$G_t$,首先在当前rollout组内标准化每个标量估计器:$\tilde{A}_{(\cdot)_i} = \frac{\hat{A}_{(\cdot)_i} - \mu_{G_t}(\hat{A}_{(\cdot)})}{\sigma_{G_t}(\hat{A}_{(\cdot)}) + \epsilon_{num}}$。然后形成混合标量分数$A_{mix_i}(t) = (1 - \alpha_t)\tilde{A}_{G_i} + \alpha_t\tilde{A}_{top-k_i}$,其中相位调度将$\alpha_t$从0线性缩放到1贯穿整个训练过程。最后,使用掩码裁剪代理目标优化:$\mathcal{L}(\theta) = -\mathbb{E}_{(i,\tau) \sim T_t} [\min(r_{i,\tau}(\theta) A_{tok_i,\tau}, \text{clip}(r_{i,\tau}(\theta), 1-\epsilon_{lo}, 1+\epsilon_{hi}) A_{tok_i,\tau})]$,其中$T_t$表示rollout组$G_t$中的有效响应token。
技术新颖性
技术新颖性主要体现在三个方面。首先,advisor模型范式将战略搜索决策与实现解耦,这是首次在进化搜索代理中明确分离决定尝试什么和如何实现的架构设计。这种分离使强化学习专注于战略推理层面的学习,而不会受到实现正确性的噪声信号干扰。其次,相位自适应强化学习算法根据搜索动力学动态调整信用分配来源,这是对GRPO、熵目标和PKPO等单一固定信号方法的重要改进。算法在早期探索时维持高熵以维持探索(不像Max@k过早收敛),后期使用frontier贡献分支改进精炼而不继承GRPO的梯度尖峰。第三,理论上的尺度条件化信用分配(定理1)形式化了奖励压缩下的条件化观点:当对应的分支标准差主导$\epsilon_{num}$时,标准化移除了全局仿射奖励尺度并保留了分支特定的信用排序,这为算法的稳定性提供了理论保证。
实验结果
在三个真实世界的机器学习相关工程和研究任务中,PACEvolve++一致性地提供最强的最终奖励,同时优化平滑且收敛最快。在Expert-parallelism Load Balancing (EPLB)任务上,PACEvolve++和非RL的PACEvolve基线都达到饱和的近优解,但PACEvolve++仅使用了一半的进化预算。在Sequential Recommendation (KuaiRec)任务上,本文方法收敛到比基线更好的解决方案。在Protein Fitness Extrapolation (Multi-Evolve)任务上,PACEvolve++同样表现出更优的最终性能和更快的收敛速度。辅助指标揭示了基线停滞的原因:ThetaEvolve的GRPO风格目标在EPLB的原始奖励上保持竞争力,但在后期出现重复的梯度范数尖峰,包括几次超过2的偏移和超过4的峰值,这与分组奖励压缩时的方差爆炸一致;PKPO表现出相反的病理:其熵几乎单调从约1.0下降到0.4以下,表明过早承诺开发并在搜索前沿饱和之前失去多样性;熵目标由于集中的奖励分布是最不稳定的。相比之下,PACEvolve++的搜索动力学感知目标通过将训练信号与搜索阶段匹配避免了这些病理,保持梯度范数在1左右的相对窄带内,同时维持比Max@k明显更高的熵,这些更平滑的动态转化为最强的最终搜索性能。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Expert-parallelism Load Balancing | balancedness与speed的算术平均值 | 达到饱和近优解,仅需一半进化预算 | ThetaEvolve/TTT-Discover/Max@k | 收敛速度提升约50% |
| Sequential Recommendation (KuaiRec) | NDCG@10、Hit Rate@10和MRR的算术平均值 | 收敛到比基线更好的解决方案 | ThetaEvolve/TTT-Discover/Max@k | 最终性能和收敛速度均优于所有基线 |
| Protein Fitness Extrapolation (Multi-Evolve) | 0.7×r + 0.3×P@5,跨数据集平均 | 更强的最终搜索性能和更快的收敛 | ThetaEvolve/TTT-Discover/Max@k | 最终性能和收敛速度均优于所有基线 |
局限与改进
作者在论文中承认了一些局限性。首先,框架假设进化搜索代理暴露当前父程序、最近的搜索历史、评估器评分以及rollout边界的同步点,这可能不适用于所有进化搜索系统。其次,虽然advisor模型范式在MLE任务中表现出色,因为高层搜索推理和低层代码实现有不同的容量需求,但在其他类型任务中的适用性可能需要进一步验证。第三,相位调度将$\alpha_t$从0线性缩放到1贯穿整个训练过程,这种线性调度可能不是所有任务的最优选择,不同任务可能需要不同的相位转换曲线。除了作者承认的局限性外,我还观察到一些潜在问题:方法依赖于昂贵的frontier实现模型(如Gemini-3.1-pro-preview)进行候选实现,这可能在资源受限场景中成为瓶颈;框架当前主要针对算法设计和模型架构优化任务,在其他类型进化搜索任务(如数学定理证明、科学发现)中的泛化能力有待验证;相位自适应方法需要对搜索阶段有先验知识,在实际应用中可能难以准确估计当前所处的搜索阶段。
独立分析的弱点
独立分析的第一个弱点是advisor模型训练需要大量的rollout数据,在评估成本极其昂贵的场景中可能无法收集足够的数据进行有效训练。改进方向是研究更数据高效的advisor训练方法,如离线强化学习或元学习,从少量rollout中快速适应新任务。第二个弱点是相位调度是线性的,没有根据实际搜索进度自适应调整。改进方向是开发自适应相位检测机制,通过监控奖励方差、熵、梯度范数等指标动态调整$\alpha_t$,而不是依赖固定的线性调度。第三个弱点是当前实现需要强大的frontier模型(如Gemini-3.1-pro-preview),在资源受限场景中可能不可用。改进方向是研究如何使用更小或本地部署的模型作为frontier实现,或者探索advisor模型自身也能够完成实现任务的能力提升。第四个弱点是方法的复杂性较高,需要协调多个组件(advisor、frontier、相位调度、标准化等),增加了工程实现的难度。改进方向是简化框架设计,提供更易于实现的API和配置选项,降低使用门槛。
未来方向
作者提出了一些未来研究方向,包括将advisor模型范式扩展到其他类型的进化搜索任务,如数学定理证明、科学发现和自动化系统设计。基于本文成果可以延伸的研究方向包括:探索更复杂的相位调度策略,如基于强化学习的自适应相位选择,或多阶段的相位转换而非简单的线性过渡;研究advisor模型的迁移学习和元学习能力,使得在源任务上训练的advisor可以快速适应新任务,减少在新任务上的训练成本;将相位自适应方法与其他测试时训练技术结合,如上下文学习、提示优化和模型编辑,以进一步提升搜索代理的性能;探索advisor模型在多目标进化搜索中的应用,处理多个可能冲突的优化目标;研究如何将框架扩展到分布式进化搜索场景,多个advisor模型协作或竞争,进一步提升搜索效率和多样性;开发更全面的评估基准,涵盖更多真实世界的工程和研究任务,以验证方法的泛化能力。
复现评估
论文没有明确提及代码开源情况,但提供了详细的实验设置和评估协议,包括任务特定评估器超时、进化表面定义、评估指标等。算力需求方面,实验使用了Qwen3.5-4B和DeepSeek-R1-0528-Qwen3-8B两个模型尺寸,并使用Gemini-3.1-pro-preview进行候选实现,这表明需要相当规模的GPU资源。评估过程包括训练16个epoch(Sequential Recommendation)或更长时间的计算,每个候选有1200秒的wall-clock预算约束,这意味着完整复现可能需要数天甚至数周的计算时间。复现难度评估为中等到困难,主要挑战在于:需要访问强大的frontier模型(如Gemini-3.1-pro-preview);评估成本高昂,可能需要优化评估管道以提高效率;框架涉及多个组件的协调,需要仔细实现以确保正确性。为提高可复现性,建议作者开源代码、提供预训练的advisor模型、公开评估数据集或提供详细的数据生成协议、提供Docker容器或环境配置文件以简化设置过程。
论文图表