论文

基于系统发生树的LLM驱动进化式代码优化

Large Language Model-Powered Evolutionary Code Optimization on a Phylogenetic Tree

上下文与知识记忆Agent记忆

摘要

为现代 GPU 优化科学计算算法是一个劳动密集的迭代过程,涉及在复杂软硬件栈上反复修改代码、基准测试与调参。近期工作探索了 LLM 辅助的进化式自动代码优化方法,但这些方法主要依赖基于结果的选择与随机变异,未充分利用迭代优化过程中产生的丰富轨迹信息。我们提出 PhyloEvolve,一个 LLM 智能体系统,把面向 GPU 的算法优化重构为上下文强化学习(ICRL)问题。这一形式化支持无需重训模型的轨迹条件化经验复用。PhyloEvolve 把算法蒸馏与基于提示的 Decision Transformers 融入迭代工作流,把算法修改序列与性能反馈作为一等学习信号。为组织优化历史,我们引入系统发生树表示,捕捉算法变种之间的继承、分化与重组,支持回溯、跨谱系迁移与可复现性。系统结合精英轨迹池、多岛并行探索与容器化执行,在异构硬件上平衡探索与利用。我们在包括 PDE 求解器、流形学习与谱图算法的科学计算负载上评估 PhyloEvolve,展示其相对基线与进化方法在运行时间、内存效率与正确性上的一致改进。代码发布于:https://github.com/annihi1ation/phylo_evolve

基于系统发生树的LLM驱动进化式代码优化
图3:PhyloEvolve 的简化流程。Orchestrator 协调多个 LLM 智能体生成精炼候选和更大规模的结构性重新设计。生成的变体被插入系统发生森林中,由 CodeExecutor 执行并进行基准测试,同时评估其性能。表现优异的变体被保存在 Elite Pool 中,Elite Pool 又通过轨迹条件化检索和跨谱系知识迁移来引导后续进化。