用梯度优化替代树搜索,让大模型更高效完成机器学习工程任务。
Reasoning as Gradient: Scaling MLE Agents Beyond Tree Search
- 将诊断推理映射为梯度计算,实现方向性优化。
- 在单张V100上12小时达成35.1%任意奖牌率,刷新纪录。
- 适合追求高效推理的模型工程师和系统开发者。
基于大模型的机器学习工程(MLE)代理主要依赖树搜索,这是一种无梯度优化方法,通过标量验证分数对候选方案进行排序。随着大模型推理能力提升,穷举搜索相比有方向性的更新愈发低效,如同精确梯度使优化远胜于随机搜索。我们提出Gome,一个实现梯度优化的MLE代理:将结构化诊断推理映射为梯度计算,成功记忆作为动量,多轨迹执行对应分布式优化。在隔离外部知识影响的封闭世界协议下,Gome在MLE-Bench上以单张V100 GPU、12小时限制达成35.1%的任意奖牌率,达到当前最优。跨10个模型的扩展实验显示关键转折点:对于弱模型,树搜索仍因通过穷举弥补推理不可靠而占优;当推理能力增强,梯度优化逐步超越,且在前沿模型上差距持续扩大。鉴于推理型大模型快速进步,该研究预示梯度优化正成为更优范式。代码与GPT-5轨迹已开源:https://github.com/microsoft/RD-Agent。
原文摘要 · Abstract (English)
LLM-based agents for machine learning engineering (MLE) predominantly rely on tree search, a form of gradient-free optimization that uses scalar validation scores to rank candidates. As LLM reasoning capabilities improve, exhaustive enumeration becomes increasingly inefficient compared to directed updates, analogous to how accurate gradients enable efficient descent over random search. We introduce Gome, an MLE agent that operationalizes gradient-based optimization. Gome maps structured diagnostic reasoning to gradient computation, success memory to momentum, and multi-trace execution to distributed optimization. Under a closed-world protocol that isolates architectural effects from external knowledge, Gome achieves a state-of-the-art 35.1\% any-medal rate on MLE-Bench with a restricted 12-hour budget on a single V100 GPU. Scaling experiments across 10 models reveal a critical crossover: with weaker models, tree search retains advantages by compensating for unreliable reasoning through exhaustive exploration; as reasoning capability strengthens, gradient-based optimization progressively outperforms, with the gap widening at frontier-tier models. Given the rapid advancement of reasoning-oriented LLMs, this positions gradient-based optimization as an increasingly favorable paradigm. We release our codebase and GPT-5 traces at https://github.com/microsoft/RD-Agent.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。