BiboyQG commited on
Commit
13badf9
·
verified ·
1 Parent(s): 0903eb8

Add model card

Browse files
Files changed (1) hide show
  1. README.md +113 -0
README.md ADDED
@@ -0,0 +1,113 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen3-4B-Thinking-2507
4
+ library_name: transformers
5
+ pipeline_tag: text-generation
6
+ datasets:
7
+ - Spreadsheet-RL/Spreadsheet-RL
8
+ language:
9
+ - en
10
+ tags:
11
+ - spreadsheet
12
+ - excel
13
+ - reinforcement-learning
14
+ - grpo
15
+ - agents
16
+ - tool-use
17
+ - verl
18
+ - qwen3
19
+ ---
20
+
21
+ # Spreadsheet-RL-4B
22
+
23
+ [**Project Page**](https://spreadsheet-rl.github.io/) | [**Paper**](https://arxiv.org/abs/2605.22642) | [**Dataset**](https://huggingface.co/datasets/Spreadsheet-RL/Spreadsheet-RL) | [**Code**](https://github.com/Spreadsheet-RL/Spreadsheet-RL)
24
+
25
+ Spreadsheet-RL-4B is the RL-trained 4B spreadsheet agent checkpoint from **Spreadsheet-RL: Advancing Large Language Model Agents on Realistic Spreadsheet Tasks via Reinforcement Learning**. It starts from [`Qwen/Qwen3-4B-Thinking-2507`](https://huggingface.co/Qwen/Qwen3-4B-Thinking-2507) and is post-trained with outcome-based reinforcement learning in Spreadsheet Gym, a multi-turn Microsoft Excel environment with spreadsheet-native tools, sandboxed code execution, and Excel-based recalculation rewards.
26
+
27
+ This checkpoint is intended to be used with the Spreadsheet-RL agent harness and tool environment. Loading it as a plain chat model can be useful for inspection, but it will not reproduce the paper results without Spreadsheet Gym, the tool set, and the reward/evaluation pipeline.
28
+
29
+ ## News
30
+
31
+ - 2026-05-23: Released the Spreadsheet-RL-4B model checkpoint on Hugging Face at [`Spreadsheet-RL/Spreadsheet-RL-4B`](https://huggingface.co/Spreadsheet-RL/Spreadsheet-RL-4B).
32
+
33
+ ## Model Details
34
+
35
+ | Field | Value |
36
+ | --- | --- |
37
+ | Base model | [`Qwen/Qwen3-4B-Thinking-2507`](https://huggingface.co/Qwen/Qwen3-4B-Thinking-2507) |
38
+ | Training method | GRPO with outcome-based rewards |
39
+ | Environment | Spreadsheet Gym with Microsoft Excel 365, spreadsheet-native tools, SandboxFusion code execution, and async Excel recalculation/reward service |
40
+ | Training data | Spreadsheet-RL training split: 5,928 filtered ExcelForum tasks |
41
+ | Evaluation | SpreadsheetBench and Domain-Spreadsheet |
42
+ | License | Apache-2.0, following the base model license |
43
+
44
+ ## Training Configuration
45
+
46
+ For full details, please see the paper. The released 4B run uses:
47
+
48
+ | Hyperparameter | Value |
49
+ | --- | --- |
50
+ | Algorithm | GRPO; KL-regularized against a frozen reference model |
51
+ | Training steps | 60 |
52
+ | Prompt/response limits | 4,096 / 27,648 tokens |
53
+ | Rollout sampling | temperature 0.6; top-p 0.95; top-k 20 |
54
+ | Batching | 64 prompts/step; 16 rollouts/prompt; 1,024 rollouts/step |
55
+ | Multi-turn caps | max assistant turns 20; max user turns 20; max tool-response length 8,192 |
56
+ | Optimizer | AdamW; learning rate 1e-6; weight decay 0.01; betas (0.9, 0.999); grad clip 1.0 |
57
+ | KL loss | low-var KL; coefficient 0.001 |
58
+ | Actor update batching | mini-batch 32; dynamic batch sizing enabled |
59
+ | Hardware | 1 node x 4 NVIDIA H100 GPUs |
60
+ | Training time | about 40 hours wall-clock for the 4B run |
61
+
62
+ ## Results
63
+
64
+ Spreadsheet-RL improves the same 4B base model through spreadsheet-native interaction design, comprehensive tool access, and RL post-training.
65
+
66
+ | Benchmark | Base | + Native Harness | + Full Tools | Spreadsheet-RL-4B |
67
+ | --- | ---: | ---: | ---: | ---: |
68
+ | SpreadsheetBench Pass@1 | 12.0 | 15.6 | 19.3 | 23.4 |
69
+
70
+ On Domain-Spreadsheet, Spreadsheet-RL improves overall Pass@1 from 8.4 to 17.2 over 1,660 evaluation rollouts.
71
+
72
+ ## Usage
73
+
74
+ Install the standard Transformers stack and load the checkpoint:
75
+
76
+ ```python
77
+ from transformers import AutoModelForCausalLM, AutoTokenizer
78
+
79
+ model_id = "Spreadsheet-RL/Spreadsheet-RL-4B"
80
+
81
+ tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
82
+ model = AutoModelForCausalLM.from_pretrained(
83
+ model_id,
84
+ torch_dtype="auto",
85
+ device_map="auto",
86
+ trust_remote_code=True,
87
+ )
88
+ ```
89
+
90
+ For task evaluation and agent rollouts, use the full Spreadsheet-RL codebase with the released dataset and Spreadsheet Gym:
91
+
92
+ ```bash
93
+ hf download Spreadsheet-RL/Spreadsheet-RL --repo-type dataset --local-dir data
94
+ git clone https://github.com/Spreadsheet-RL/Spreadsheet-RL.git
95
+ ```
96
+
97
+ The default training/evaluation harness is maintained in the code repository under `configs/`, `scripts/`, `reward/`, and `verl/`.
98
+
99
+ ## Citation
100
+
101
+ ```bibtex
102
+ @misc{chi2026spreadsheetrl,
103
+ title = {Spreadsheet-RL: Advancing Large Language Model Agents on Realistic Spreadsheet Tasks via Reinforcement Learning},
104
+ 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},
105
+ year = {2026},
106
+ eprint = {2605.22642},
107
+ archivePrefix = {arXiv},
108
+ primaryClass = {cs.AI},
109
+ doi = {10.48550/arXiv.2605.22642},
110
+ url = {https://arxiv.org/abs/2605.22642}
111
+ }
112
+ ```
113
+