75 个 Kaggle 竞赛
从 Kaggle 精选 75 个与机器学习工程相关的竞赛,构成多样且有难度的任务集合,用来测试训练模型、准备数据集、运行实验等真实工程技能。
拖动滑块完成拼图,通过后发送验证码
独立第三方 AI 评测平台 · 只用实测数据
用 75 个 Kaggle 竞赛衡量 AI 代理的机器学习工程能力
MLE-bench 由 OpenAI 在 2024 年 10 月 10 日发布,是一个评估 AI 代理机器学习工程能力的基准。它从 Kaggle 精选 75 个机器学习工程相关竞赛,覆盖训练模型、准备数据集、运行实验等真实技能,并借助 Kaggle 公开排行榜为每个竞赛建立人类基线。测试中表现最好的配置是 OpenAI 的 o1-preview 配合 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。基准代码已开源,方便后续研究 AI 代理的机器学习工程能力。
MLE-bench 是 OpenAI 推出的一个基准,用来衡量 AI 代理在机器学习工程任务上的表现。它从 Kaggle 挑选了 75 个与机器学习工程相关的竞赛,组成一组多样且有难度的任务,考察训练模型、准备数据集、运行实验等真实工程技能。
每个竞赛的人类基线来自 Kaggle 公开排行榜。OpenAI 用开源代理脚手架在基准上评测了多个前沿语言模型,其中表现最好的是 o1-preview 搭配 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。研究还考察了面向 AI 代理的多种资源扩展方式,以及预训练数据污染带来的影响。
基准代码已开源,供后续研究使用。研究人员可以用它复现评测、比较不同代理脚手架与语言模型的组合;关注机器学习工程自动化的开发者,可以据此了解当前代理在真实竞赛任务上的能力边界。
从 Kaggle 精选 75 个与机器学习工程相关的竞赛,构成多样且有难度的任务集合,用来测试训练模型、准备数据集、运行实验等真实工程技能。
为每个竞赛建立人类基线,基线取自 Kaggle 公开排行榜,使代理成绩可以直接与人类参赛者的奖牌水平对照。
使用开源代理脚手架评测多个前沿语言模型,其中最好配置为 o1-preview 搭配 AIDE 脚手架,在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。
除主结果外,还研究面向 AI 代理的多种资源扩展形式,观察增加资源投入对代理在机器学习工程任务上表现的影响。
考察预训练数据污染对评测结果的影响,帮助判断代理成绩在多大程度上来自真实工程能力而非数据泄漏。
OpenAI 将基准代码开源,便于后续研究复现评测、比较不同代理脚手架与模型组合,继续理解 AI 代理的机器学习工程能力。
官方说明 ↗当你要评估 AI 代理能否胜任真实机器学习工程任务时,MLE-bench 值得一看:它用 75 个 Kaggle 竞赛和人类基线给出可对照的分数,最好配置在 16.9% 的竞赛中达到至少 Kaggle 铜牌水平。如果你是研究者或团队要选代理脚手架与模型,它能提供统一的比较口径;但若你需要的是开箱即用的建模工具,它本身不代跑竞赛,也不提供产品化界面。
需要可复现、有人类基线对照的评测来比较不同代理脚手架和语言模型组合,MLE-bench 的 75 个 Kaggle 竞赛正适合这类研究。
想了解当前 AI 代理在训练模型、准备数据集、运行实验等真实工程任务上能达到什么水平,可用铜牌率作为参考。
需要判断代理成绩是否受预训练数据污染影响时,可以直接参考基准中关于污染影响的分析。
MLE-bench 是评测基准而非产品,不提供托管建模界面,期待开箱即用工具的用户会觉得不匹配。
它是 OpenAI 发布的一个基准,用来衡量 AI 代理在机器学习工程上的表现。
除主结果外,研究还考察面向 AI 代理的多种资源扩展形式,以及预训练数据污染对评测结果的影响,帮助判断代理成绩在多大程度上来自真实工程能力。
可以,OpenAI 将该基准代码开源,供后续研究复现评测、比较不同代理脚手架与模型组合,进一步理解 AI 代理的机器学习工程能力。