MLE-bench

MLE-bench

AI TOOL PROFILE工具资料

用 75 个 Kaggle 竞赛衡量 AI 代理的机器学习工程能力

官网地址 https://openai.com/index/mle-bench/

访问官网
价格官网未标价
平台Web
归类大模型与开发平台
收录2024-10-10
复核2024-10-10
编辑评分 4.4 / 5

MLE-bench编辑总结

EDITOR REVIEW编辑评测

MLE-bench 由 OpenAI 在 2024 年 10 月 10 日发布,是一个评估 AI 代理机器学习工程能力的基准。它从 Kaggle 精选 75 个机器学习工程相关竞赛,覆盖训练模型、准备数据集、运行实验等真实技能,并借助 Kaggle 公开排行榜为每个竞赛建立人类基线。测试中表现最好的配置是 OpenAI 的 o1-preview 配合 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。基准代码已开源,方便后续研究 AI 代理的机器学习工程能力。

MLE-bench是什么?

MLE-bench 是 OpenAI 推出的一个基准,用来衡量 AI 代理在机器学习工程任务上的表现。它从 Kaggle 挑选了 75 个与机器学习工程相关的竞赛,组成一组多样且有难度的任务,考察训练模型、准备数据集、运行实验等真实工程技能。

每个竞赛的人类基线来自 Kaggle 公开排行榜。OpenAI 用开源代理脚手架在基准上评测了多个前沿语言模型,其中表现最好的是 o1-preview 搭配 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。研究还考察了面向 AI 代理的多种资源扩展方式,以及预训练数据污染带来的影响。

基准代码已开源,供后续研究使用。研究人员可以用它复现评测、比较不同代理脚手架与语言模型的组合;关注机器学习工程自动化的开发者,可以据此了解当前代理在真实竞赛任务上的能力边界。

MLE-bench核心功能

CAPABILITIES工具标签

75 个 Kaggle 竞赛

从 Kaggle 精选 75 个与机器学习工程相关的竞赛,构成多样且有难度的任务集合,用来测试训练模型、准备数据集、运行实验等真实工程技能。

人类基线对比

为每个竞赛建立人类基线,基线取自 Kaggle 公开排行榜,使代理成绩可以直接与人类参赛者的奖牌水平对照。

前沿模型评测

使用开源代理脚手架评测多个前沿语言模型,其中最好配置为 o1-preview 搭配 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。

资源扩展研究

除主结果外,还研究面向 AI 代理的多种资源扩展形式,观察增加资源投入对代理在机器学习工程任务上表现的影响。

污染影响分析

考察预训练数据污染对评测结果的影响,帮助判断代理成绩在多大程度上来自真实工程能力而非数据泄漏。

开源基准代码

OpenAI 将基准代码开源,便于后续研究复现评测、比较不同代理脚手架与模型组合,继续理解 AI 代理的机器学习工程能力。

官方说明 ↗

MLE-bench值不值得用?

VERDICT工具标签

当你要评估 AI 代理能否胜任真实机器学习工程任务时,MLE-bench 值得一看:它用 75 个 Kaggle 竞赛和人类基线给出可对照的分数,最好配置在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。如果你是研究者或团队要选代理脚手架与模型,它能提供统一的比较口径;但若你需要的是开箱即用的建模工具,它本身不代跑竞赛,也不提供产品化界面。

建议选它的情况

  • 研究 AI 代理机器学习工程能力的团队
  • 需要统一基准比较不同代理脚手架与模型组合的人
  • 关注预训练污染对评测影响的研究者

不建议选它的情况

  • 想要开箱即用建模工具而非评测基准的用户
  • 需要产品化界面与托管服务的团队

MLE-bench使用建议

HOW TO USE工具标签
入门先把路子走对
  1. 先读论文从 MLE-bench 页面进入论文,先看 75 个竞赛的选取口径与人类基线定义,再决定是否在自己的代理上复现。
  2. 对齐脚手架配置复现时先确认代理脚手架与模型组合,页面上主结果使用的是 o1-preview 搭配 AIDE 脚手架,便于与 16.9% 的铜牌率对照。
  3. 查看开源代码从页面上的开源链接获取基准代码,按其说明准备竞赛数据与运行环境,再开始自己的评测。
进阶省时间也省钱
  1. 控制污染影响做结果解读时把预训练污染分析一并纳入,避免把数据泄漏带来的分数误判为工程能力。
  2. 按竞赛分组看结果不要只看总铜牌率,按 75 个竞赛逐个看代理成绩与人类基线的差距,能更清楚能力边界在哪。

MLE-bench适合谁用?

WHO IS IT FOR工具标签

代理与基准研究者

推荐

需要可复现、有人类基线对照的评测来比较不同代理脚手架和语言模型组合,MLE-bench 的 75 个 Kaggle 竞赛正适合这类研究。

机器学习工程团队

推荐

想了解当前 AI 代理在训练模型、准备数据集、运行实验等真实工程任务上能达到什么水平,可用铜牌率作为参考。

关注污染问题的评估者

推荐

需要判断代理成绩是否受预训练数据污染影响时,可以直接参考基准中关于污染影响的分析。

只想用现成建模工具的人

不推荐

MLE-bench 是评测基准而非产品,不提供托管建模界面,期待开箱即用工具的用户会觉得不匹配。

MLE-bench优点和缺点

PROS & CONS工具标签

优点

  • 覆盖 75 个 Kaggle 竞赛,任务类型多样
  • 每个竞赛都有人类基线,成绩可对照
  • 最好配置在 16.9% 竞赛中达到至少铜牌水平
  • 同时分析资源扩展与预训练污染影响
  • 基准代码开源,便于复现与后续研究

缺点

  • 官网未公开定价与商业支持方式
  • 自身不是托管建模产品,需自行准备运行环境
  • 主结果仅覆盖部分前沿模型与代理脚手架组合

MLE-bench常见问题

FAQ工具标签
MLE-bench 是什么?

它是 OpenAI 发布的一个基准,用来衡量 AI 代理在机器学习工程上的表现。

评测还会考察哪些方面?

除主结果外,研究还考察面向 AI 代理的多种资源扩展形式,以及预训练数据污染对评测结果的影响,帮助判断代理成绩在多大程度上来自真实工程能力。

基准代码可以获取吗?

可以,OpenAI 将该基准代码开源,供后续研究复现评测、比较不同代理脚手架与模型组合,进一步理解 AI 代理的机器学习工程能力。