AI agents自动优化蛋白质折叠模型代码,提升预测精度7.5%。
AgentFold: Closed-Loop Agentic Search for Protein Folding Model Design

- 构建多智能体系统,通过闭环搜索代码变体改进模型架构。
- 在2000+行代码上探索80个变体,计算耗时5000 GPU小时,提升lDDT 7.5%。
- 揭示稳定改进来自可学习先验和门控优化,非几何直接扰动。
科学大模型代理在文献推理、工具使用和实验规划中展现出潜力,但其能否通过可执行代码修改与高成本验证,自主优化大型紧密耦合的科学机器学习系统仍不明确。本文在蛋白质折叠领域研究此问题,该领域进展需协同架构调整、多目标评估与领域感知解读。我们提出AgentFold,一个将折叠模型开发建模为可执行代码变体闭环搜索的多智能体框架。从ESMFold出发,AgentFold提出假设,实现并调试代码级修改,评估模型变体,分析实验结果,并将成功与失败干预存入结构化记忆。采用类似蒙特卡洛树搜索的策略分配计算资源至高分搜索分支。在包含超过2000行代码的工程级折叠代码库上,AgentFold使用约5000 GPU小时与1.7亿个LLM标记,探索约80个模型变体。在相同计算预算下,相比独立的Codex提议,其最佳lDDT提升7.5%,优于随机搜索对照组。此外,干预轨迹揭示共现经验设计模式:稳定增益通常源于早期、软性、可学习先验与门控精炼;而直接几何扰动与几何条件反馈常导致训练不稳定。代码与实验资源已公开于https://github.com/lmqfly/AgentFold。
原文摘要 · Abstract (English)
Scientific LLM agents have shown promise in literature reasoning, tool use, and experiment planning, but it remains unclear whether they can autonomously improve large, tightly coupled scientific machine-learning systems through executable code changes and computationally expensive validation. We study this question in protein folding, where progress requires coordinated architectural modifications, multi-objective evaluation, and domain-aware interpretation. We present AgentFold, a multi-agent framework that formulates folding-model development as a closed-loop search over executable code variants. Starting from ESMFold, AgentFold proposes hypotheses, implements and debugs code-level modifications, evaluates model variants, analyzes experimental outcomes, and stores both successful and failed interventions in structured memory. An MCTS-style policy allocates computational resources across high-scoring search branches. On an engineering-scale protein-folding codebase comprising more than 2,000 lines of code, AgentFold explores approximately 80 model variants using approximately 5,000 GPU-hours and 170 million LLM tokens. Under a matched computational budget, AgentFold improves the best lDDT by 7.5% over independent Codex proposals and outperforms a random-search control. Beyond model improvement, the resulting intervention traces reveal recurring empirical design patterns: stable gains tend to arise from early, soft, learnable priors and gated refinement, whereas direct geometric perturbations and geometry-conditioned feedback often destabilize training. The code and experimental resources are publicly available at https://github.com/lmqfly/AgentFold.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。