miiikiik/dr-as-rubric-data

Dataset

DR as Rubric Data

0

9 commits

1 linked in READMEs

updated Jun 1, 2026

See the code

README

DR as Rubric Data

Companion dataset for the paper "Deep Research as Rubric for Reinforcement Learning".

Paper repository: https://github.com/miiikiik/dr-as-rubric

Dataset Description

This dataset contains two types of data produced by the DR-as-Rubric pipeline:

1. SFT Training Data (sft_data/)

FileFormatSamplesDescription
trajectory_sharegpt.jsonShareGPT JSON991Multi-turn trajectories for supervised fine-tuning of an 8B model. Each sample contains a system prompt, query, and multi-step research reasoning trajectory.

2. Rubric Data (rubrics/)

Evaluation rubrics generated by different models and bootstrap steps. Each CSV has columns query and criteria.

FileRowsSource
real-rubric-bootstrap-step1.csv9998B model, bootstrap step 1
real-rubric-bootstrap-step2.csv9998B model, bootstrap step 2
real-rubric-bootstrap-step3.csv9998B model, bootstrap step 3
real-rubric-bootstrap-step4.csv9998B model, bootstrap step 4
gpt5-rubric-1k.csv999GPT-5 generated rubrics
gemini-3.1-pro-rubric.csv999Gemini 3.1 Pro generated rubrics

Usage

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)

Citation

@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}
}

License

Apache License 2.0

deep-research
reinforcement-learning
rubric

miiikiik/dr-as-rubric-data

Dataset

DR as Rubric Data

0

9 commits

1 linked in READMEs

updated Jun 1, 2026

See the code

README

DR as Rubric Data

Companion dataset for the paper "Deep Research as Rubric for Reinforcement Learning".

Paper repository: https://github.com/miiikiik/dr-as-rubric

Dataset Description

This dataset contains two types of data produced by the DR-as-Rubric pipeline:

1. SFT Training Data (sft_data/)

FileFormatSamplesDescription
trajectory_sharegpt.jsonShareGPT JSON991Multi-turn trajectories for supervised fine-tuning of an 8B model. Each sample contains a system prompt, query, and multi-step research reasoning trajectory.

2. Rubric Data (rubrics/)

Evaluation rubrics generated by different models and bootstrap steps. Each CSV has columns query and criteria.

FileRowsSource
real-rubric-bootstrap-step1.csv9998B model, bootstrap step 1
real-rubric-bootstrap-step2.csv9998B model, bootstrap step 2
real-rubric-bootstrap-step3.csv9998B model, bootstrap step 3
real-rubric-bootstrap-step4.csv9998B model, bootstrap step 4
gpt5-rubric-1k.csv999GPT-5 generated rubrics
gemini-3.1-pro-rubric.csv999Gemini 3.1 Pro generated rubrics

Usage

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)

Citation

@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}
}

License

Apache License 2.0

deep-research
reinforcement-learning
rubric