CoolFace
Modelpublic

budget-internalization-iclr2027/qwen3.5-4b-8k-drgrpo-equalminnow-s300

sourceHugging Faceapache-2.0updated 4d agoView on Hugging Face
0likes29downloads
Model Card

Qwen3.5-4B · 8k-token budget · drgrpo · equalminnow

RL-finetuned Qwen/Qwen3.5-4B trained for math reasoning under a 8,192-token generation budget. Released as part of an anonymous ICLR 2027 submission.

Run ID (petname): `equalminnow` · checkpoint step 300

Training

Base modelQwen/Qwen3.5-4B
AlgorithmDr. GRPO: GRPO without reward std-normalization and with a constant length normalizer.
Generation budget (max_new_tokens)8,192
DataDeepScaleR (math), 3 epochs max
Batch32 prompts × 8 rollouts per step
OptimizerAdam, cosine LR schedule, peak LR 5e-7, 10 warmup steps
Steps300
Rewardbinary answer correctness (\boxed{} extraction)
Weights dtypeBF16

Training prompt (user turn, rendered with the base model's chat template):

Think step-by-step to solve the following problem. Output your answer inside of \\boxed{} tags.:
{problem}

Let's think step-by-step

Usage

python
from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "budget-internalization-iclr2027/qwen3.5-4b-8k-drgrpo-equalminnow-s300"
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="auto", device_map="auto")

With vLLM: vllm serve budget-internalization-iclr2027/qwen3.5-4b-8k-drgrpo-equalminnow-s300

License

Inherits the license of the base model (Qwen/Qwen3.5-4B).