Spreadsheet-RL
收藏数据集概述
数据集名称:Spreadsheet-RL
语言:英语(en)
许可协议:Creative Commons Attribution-ShareAlike 4.0 International (CC-BY-SA-4.0)
任务类别:强化学习(reinforcement-learning)
标签:spreadsheet, excel, reinforcement-learning, tool-use, agents, verl, grpo
数据集简介:本数据集用于训练和评估 Spreadsheet-RL 模型,这是一个面向电子表格智能体的强化学习框架。该智能体通过工具编辑 Excel 工作簿,并根据工作簿重算结果与答案范围比较获得基于结果的奖励。
文件结构
数据集包含按解析器类型划分的 Parquet 文件和一个压缩的工作簿归档文件:
| 文件 | 行数 | 说明 |
|---|---|---|
train_hermes.parquet |
5,928 | 训练集,使用 Hermes 工具调用格式 |
train_qwen3_coder.parquet |
5,928 | 训练集,使用 Qwen3-Coder 工具调用格式 |
test_hermes.parquet |
2,726 | SpreadsheetBench 评估集,使用 Hermes 格式 |
test_qwen3_coder.parquet |
2,726 | SpreadsheetBench 评估集,使用 Qwen3-Coder 格式 |
test_domain_hermes.parquet |
1,662 | Domain-Spreadsheet 评估集,使用 Hermes 格式 |
test_domain_qwen3_coder.parquet |
1,662 | Domain-Spreadsheet 评估集,使用 Qwen3-Coder 格式 |
spreadsheets.zip |
10,316 个任务 | 工作簿文件与任务元数据 |
Parquet 文件命名规则:train_<parser_type>.parquet、test_<parser_type>.parquet、test_domain_<parser_type>.parquet。
数据表结构
每个 Parquet 文件包含以下列:
| 列名 | 说明 |
|---|---|
data_source |
数据来源标签,当前为 sheet_arena |
agent_name |
智能体循环名称,当前为 tool_agent |
prompt |
序列化的聊天提示,包含系统和用户任务消息 |
ability |
能力标签,当前为 spreadsheet |
reward_model |
奖励元数据,包括真实工作簿路径和奖励风格 |
extra_info |
任务元数据,如 id、split、answer_position 及主工作簿扩展名 |
工作簿归档内容
spreadsheets.zip 解压后包含三个目录:
excelforum/spreadsheetbench/domain/
每个任务目录包含:
instruction.jsoninput.xlsxoutput.xlsxtarget.xlsx
各组任务数量:
| 分组 | 任务数 |
|---|---|
| ExcelForum 训练任务 | 5,928 |
| SpreadsheetBench 评估任务 | 2,726 |
| Domain-Spreadsheet 评估任务 | 1,662 |
Domain-Spreadsheet 领域分布
| 领域 | 任务数 |
|---|---|
| finance_beginner | 597 |
| finance_intermediate | 388 |
| finance_advanced | 135 |
| hr | 185 |
| real_estate | 89 |
| sales | 87 |
| supply_chain | 181 |
使用方式
下载数据集: bash hf download Spreadsheet-RL/Spreadsheet-RL --repo-type dataset --local-dir data unzip data/spreadsheets.zip -d data
设置环境变量进行训练和评估(默认使用 Hermes 格式文件): bash export SPREADSHEET_RL_DATA_ROOT="$PWD/data" export TRAIN_FILE="$SPREADSHEET_RL_DATA_ROOT/train_hermes.parquet" export TEST_FILE="$SPREADSHEET_RL_DATA_ROOT/test_hermes.parquet"
若使用 Domain-Spreadsheet 验证集: bash export TEST_FILE="$SPREADSHEET_RL_DATA_ROOT/test_domain_hermes.parquet"
若使用 Qwen3-Coder 解析器数据,请使用 *_qwen3_coder.parquet 文件并相应配置 rollout 解析器。
引用
bibtex @misc{chi2026spreadsheetrl, title = {Spreadsheet-RL: Advancing Large Language Model Agents on Realistic Spreadsheet Tasks via Reinforcement Learning}, author = {Banghao Chi and Yining Xie and Mingyuan Wu and Jingcheng Yang and Jize Jiang and Zhaoheng Li and Shengyi Qian and Minjia Zhang and Klara Nahrstedt and Rui Hou and Xiangjun Fan and Hanchao Yu}, year = {2026}, eprint = {2605.22642}, archivePrefix = {arXiv}, primaryClass = {cs.AI}, doi = {10.48550/arXiv.2605.22642}, url = {https://arxiv.org/abs/2605.22642} }
相关资源
- 项目页面:https://spreadsheet-rl.github.io/
- 论文:https://arxiv.org/abs/2605.22642
- GitHub 代码库:https://github.com/Spreadsheet-RL/Spreadsheet-RL
- 模型检查点:https://huggingface.co/Spreadsheet-RL/Spreadsheet-RL-4B




