其实,在科研方面,AI智能体也是一把能手。



Agent Laboratory由LLM驱动的多个专业智能体组成,自动处理编码、文档编写等重复耗时的任务。
在研究的每个阶段,用户都可以提供反馈与指导。Agent Laboratory旨在助力研究人员实现研究创意,加速科学发现,提高研究效率。

论文地址:https://arxiv.org/abs/2501.04227
研究发现:
由LLM驱动的专业智能体(如博士、博士后等)协同工作,承担文献综述、实验规划、数据准备和结果解释等工作。这些智能体还会集成arXiv、Hugging Face、Python和LaTeX等外部工具,来优化结果。
文献综述
在这个过程中,博士智能体借助arXiv API检索相关论文,并执行三个主要操作:摘要、全文和添加论文。
摘要:从与初始查询相关的前20篇论文中提取摘要
全文:提取特定论文的完整内容
该过程并非一次性完成,而是迭代进行。智能体多次执行查询,依据论文内容评估其相关性,筛选出合适的论文,构建全面的文献综述。
当通过「添加论文」命令达到指定数量(N=max)的相关文献后,文献综述才会完成。
实验环节
制定计划
在这个阶段,依据文献综述和研究目标,智能体需要制定一份详尽且可行的研究计划。
博士和博士后智能体通过对话协作,明确研究方法,比如要采用哪些机器学习模型、使用什么数据集,以及实验的主要步骤。
达成一致后,博士后智能体通过「计划」命令提交该计划,作为后续子任务的行动指南。
数据准备
在此阶段,ML工程师智能体负责执行Python命令来运行代码,为实验筹备可靠的数据。该智能体有权限访问 HuggingFace数据集。
代码完成后,ML工程师智能体通过「提交代码」命令提交。在正式提交前,代码会先经过Python编译器检查,确保不存在编译问题。若代码有错误,这个过程将反复进行,直至代码无误。
运行实验
在运行实验阶段,ML工程师智能体借助mle-solver模块来执行之前制定的实验计划。
mle-solver是一个专门的模块,主要功能是自主生成、测试以及优化机器学习代码,其工作流程如下:
A. 命令执行
在命令执行阶段,初始程序是从预先维护的高性能程序中选取的。
mle-solver通过「REPLACE」和「EDIT」这两个操作,对这个程序进行迭代优化。
「EDIT」操作会选定一系列行,用新生成的代码替换指定的内容。「REPLACE」操作会直接生成一个全新的Python文件。
B. 代码执行
执行代码命令后,编译器会检查新程序在运行时是否存在错误。
若程序成功编译,系统会给出一个得分。若该得分高于现有程序,顶级程序列表就会更新。
要是程序编译失败,智能体就会尝试修复代码,最多尝试3次。如果修复失败,就会返回错误提示,重新选择或生成代码。
C. 程序评分
通过基于LLM奖励模型对编译成功的代码打分,评估mle-solver生成的机器学习代码的有效性。
该奖励模型会依据研究计划、生成的代码以及观察到的输出,对程序进行评分,评分范围是0到1。得分越高,表明程序能够更有效地实现研究目标。
D. 自我反思
无论代码运行成功与否,mle-solver都会依据实验结果或者错误信号进行反思。智能体会思考每个步骤,力求优化最终结果。
如果程序编译失败,求解器就会琢磨下一次迭代时该怎么解决这个问题。要是代码成功编译且有了得分,求解器则会思考怎样提高这个分数。这些反思旨在帮助系统从错误中学习,并在后续迭代中提高代码质量和稳定性。
E. 性能稳定化
为避免性能出现波动,采用了两种机制:顶级程序采样和批量并行化。这两种策略在探索新解决方案和优化现有方案之间找到平衡,让代码修改过程更加稳定 。
研究者在MLE-bench的10个ML挑战中单独评估了mle-solver。mle-solver始终优于其他求解器,获得了更多奖牌,并在10个基准中的6个中达到了高于中位数的人类表现。
解释结果
在此阶段,博士和博士后智能体一同探讨对mle-solver得出的实验结果的理解,旨在从实验结果中提炼出有价值的见解。
当他们就某个有意义的解释达成共识,且认为该解释能为学术论文增添价值时,博士后智能体便会通过「解释」命令提交该解释,为后续的报告撰写提供支撑。
撰写研究报告
报告写作阶段,博士和教授智能体负责把研究成果整理成一份完整的学术报告。这一过程借助名为paper-solver的模块,来迭代生成并完善报告。
A. 初始报告框架
paper-solver的首要任务是生成研究论文的初始框架。该框架框架遵循学术规范,采用了LaTeX编译所需的格式,生成的论文能直接进入审阅和修改环节。
B. ArXiv研究
paper-solver可按文献综述接口访问arXiv,探索与当前撰写主题相关的文献,还可以查找可引用的论文。
C. 报告编辑
使用「EDIT」命令,对LaTeX代码进行迭代和修改,确保论文与研究计划相符、论点清晰且满足格式要求。
D. 论文审阅
这个系统借助基于LLM的代理,模拟科学论文的审阅过程,遵循NeurIPS会议的审稿指南对论文进行评估。
E. 论文完善
在论文修改阶段,根据三个评审代理给出的反馈意见,博士智能体负责决定论文是需要修订。这一过程能够持续优化研究报告,直至达到较高标准。

辅助驾驶模式
人工审阅者有两个选择:一是让系统继续推进到下一个子任务;二是要求代理重复当前子任务,并给出改进建议,助力代理在后续尝试中表现更佳。
o1-preview总分最高
内容中包含的图片若涉及版权问题,请及时与我们联系删除