Companion dataset for the paper "Deep Research as Rubric for Reinforcement Learning".
Paper repository: https://github.com/miiikiik/dr-as-rubric
This dataset contains two types of data produced by the DR-as-Rubric pipeline:
sft_data/)| File | Format | Samples | Description |
|---|---|---|---|
trajectory_sharegpt.json | ShareGPT JSON | 991 | Multi-turn trajectories for supervised fine-tuning of an 8B model. Each sample contains a system prompt, query, and multi-step research reasoning trajectory. |
rubrics/)Evaluation rubrics generated by different models and bootstrap steps. Each CSV has columns query and criteria.
| File | Rows | Source |
|---|---|---|
real-rubric-bootstrap-step1.csv | 999 | 8B model, bootstrap step 1 |
real-rubric-bootstrap-step2.csv | 999 | 8B model, bootstrap step 2 |
real-rubric-bootstrap-step3.csv | 999 | 8B model, bootstrap step 3 |
real-rubric-bootstrap-step4.csv | 999 | 8B model, bootstrap step 4 |
gpt5-rubric-1k.csv | 999 | GPT-5 generated rubrics |
gemini-3.1-pro-rubric.csv | 999 | Gemini 3.1 Pro generated rubrics |
import json
import csv
# Load SFT data
with open("sft_data/trajectory_sharegpt.json") as f:
sft_data = json.load(f)
# Load rubrics
with open("rubrics/real-rubric-bootstrap-step1.csv", encoding="utf-8-sig") as f:
reader = csv.DictReader(f)
rubrics = list(reader)
@article{mei2026dr,
title={Deep Research as Rubric for Reinforcement Learning},
author={Mei, Wangyi and Gu, Zhouhong and Bai, Zhenhan and Cai, Yin and Zhang, Lefan and Ding, Zhenxin and Chen, Bo and Gao, Yan and Wu, Yi and Hu, Yao and Liang, Jiaqing and Yang, Deqing},
year={2026}
}
Apache License 2.0
Companion dataset for the paper "Deep Research as Rubric for Reinforcement Learning".
Paper repository: https://github.com/miiikiik/dr-as-rubric
This dataset contains two types of data produced by the DR-as-Rubric pipeline:
sft_data/)| File | Format | Samples | Description |
|---|---|---|---|
trajectory_sharegpt.json | ShareGPT JSON | 991 | Multi-turn trajectories for supervised fine-tuning of an 8B model. Each sample contains a system prompt, query, and multi-step research reasoning trajectory. |
rubrics/)Evaluation rubrics generated by different models and bootstrap steps. Each CSV has columns query and criteria.
| File | Rows | Source |
|---|---|---|
real-rubric-bootstrap-step1.csv | 999 | 8B model, bootstrap step 1 |
real-rubric-bootstrap-step2.csv | 999 | 8B model, bootstrap step 2 |
real-rubric-bootstrap-step3.csv | 999 | 8B model, bootstrap step 3 |
real-rubric-bootstrap-step4.csv | 999 | 8B model, bootstrap step 4 |
gpt5-rubric-1k.csv | 999 | GPT-5 generated rubrics |
gemini-3.1-pro-rubric.csv | 999 | Gemini 3.1 Pro generated rubrics |
import json
import csv
# Load SFT data
with open("sft_data/trajectory_sharegpt.json") as f:
sft_data = json.load(f)
# Load rubrics
with open("rubrics/real-rubric-bootstrap-step1.csv", encoding="utf-8-sig") as f:
reader = csv.DictReader(f)
rubrics = list(reader)
@article{mei2026dr,
title={Deep Research as Rubric for Reinforcement Learning},
author={Mei, Wangyi and Gu, Zhouhong and Bai, Zhenhan and Cai, Yin and Zhang, Lefan and Ding, Zhenxin and Chen, Bo and Gao, Yan and Wu, Yi and Hu, Yao and Liang, Jiaqing and Yang, Deqing},
year={2026}
}
Apache License 2.0