CoDA-Bench
收藏数据集概述
CoDA-Bench(Code and Data-intensive Benchmark) 是一个用于评估 AI 智能体在数据密集型分析任务中表现的基准测试,相关论文已被 ICML 2026 接收。
核心目标
- 测试 AI 智能体在给定自然语言问题和包含数百个数据文件的 Linux 沙箱环境后,能否自主发现相关数据、编写代码并得出正确答案。
- 区别于直接提供标准数据集的现有基准,CoDA-Bench 要求智能体具备以下能力:
- 数据发现:在数百个语义相似的文件中定位相关数据。
- 文件层级导航:在复杂的 Linux 沙箱文件结构中导航。
- 多源数据整合:整合来自多个异构数据源的信息。
- 代码生成:为数据分析任务生成正确的代码。
数据集统计
| 指标 | 数值 |
|---|---|
| 总任务数 | 1,009(完整) / 119(困难子集) |
| 社区数量 | 31(完整) / 15(困难子集) |
| 源数据集 | 199 个 Kaggle 数据集 |
| 每个任务平均文件数 | ~980 个文件 |
| 总大小 | ~43 GB(压缩后) |
数据来源与内容
- 数据来源于 Kaggle 社区,包含 199 个 Kaggle 数据集,被组织为 31 个社区(community)。
- 每个任务对应一个社区,智能体需要访问该社区的完整数据目录(
datasets/communities/{release_community}/full_community)。 - 任务格式为 JSON,包含:
instance_id:唯一标识符question:自然语言问题answer:预期答案release_community:对应的社区数据目录名称
可用性与访问
- 主页:
https://coda-bench.github.io/ - 数据集下载:HuggingFace 地址
https://huggingface.co/datasets/RUC-DataLab/CoDA-Bench - 代码与文档:GitHub 仓库
https://github.com/ruc-datalab/CoDA-Bench - 许可证:MIT License(注:各个 Kaggle 数据集可能有自己的许可证)
评估方式
- Docker 模式(推荐):提供安全的隔离环境,限制智能体访问网络(仅允许 LLM API),并强制执行资源限制(内存、CPU、超时)。
- 手动评估模式:允许用户直接加载数据集,运行自定义智能体,并生成预测结果(JSONL 格式)。
- 评估指标:
- 执行准确率 (EA):答案是否正确。
- 发现准确率 (DA):是否发现了正确的相关数据。
任务示例
json { "instance_id": 0, "question": "What is the percentage of missing values in the RBC feature?", "answer": "38%", "release_community": "community_26" }
智能体需导航至 datasets/communities/community_26/full_community/,在约 980 个文件中找到 ckdisease/source/kidney_disease.csv,加载数据并计算 RBC 特征缺失值的百分比。
当前性能
截至论文发表时,Best 系统为 Mini-SWE-Agent(基于 GPT-5.5),在完整数据集的执行准确率为 61.1%,困难子集为 49.6%,发现准确率为 52.3%。
引用信息
bibtex @inproceedings{zhang2026codabench, title={CoDA-BENCH: Can Code Agents Handle Data-Intensive Tasks?}, author={Zhang, Yuxin and Fan, Ju and Fan, Meihao and Zhang, Shaolei and Du, Xiaoyong}, booktitle={Proceedings of the 43rd International Conference on Machine Learning}, year={2026}, organization={PMLR} }




