Instructions to use ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2 with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("merged_new_fixed/l2_alt") model = PeftModel.from_pretrained(base_model, "ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2") - Notebooks
- Google Colab
- Kaggle
Qwen3-4B-Instruct — L2 dialect (Structured shorthand)
A LoRA adapter that makes Qwen/Qwen3-4B-Instruct-2507 reason at compression level L2 — compressed prose / bulleted steps.
Results
| Accuracy | |
|---|---|
| After SFT | 89.8% |
| After GRPO (this adapter) | 90.0% |
| Difference | +0.2 pp |
GSM8K test (n=1317), greedy decoding, single-turn, no exemplars, no self-consistency.
Training data
GSM8K train, re-expressed at level L2 by a teacher model: 6950 examples, median chain length 140 characters inside <think>.
Across the family the median chain runs from 532 characters at L1 to 16 at L5 — a 33x span. An L2 chain looks like this:
- Sandoval: 12
- Hawkins: 12 / 2 = 6
- Sloan: 12 + 10 = 22
- Total: 12 + 6 + 22 = 40
Training setup
GRPO on top of the merged level-2 SFT model.
| Engine | trl.GRPOTrainer on stock transformers, attention sdpa |
| Reward | correctness, format |
| Loss type | dapo |
| Generations per prompt | 8 |
| Batch | 16 x 2 accum |
| Max completion | 256 tokens |
| Learning rate | 1e-05 |
| KL coefficient (beta) | 0.0 |
| Prompt set | gsm8k_grpo_balanced_1k.json |
| Trained on | merged_new_fixed/l2_alt |
| LoRA | r=16, alpha=32 |
| Hardware | 1x NVIDIA A100 80GB |
Reward components
correctness— +/- the gold solution's step count on an answer match, so harder problems are worth moreformat— the response must be one<think>...</think>block then#### <answer>
Engine note. Stock transformers with sdpa attention, not a fused-kernel wrapper. The fused path produced adapters whose lora_B matrices were all zero — mathematically inert despite loading without error. Every adapter in this collection was verified lora_B != 0 before publishing; 13 that failed that check were withheld.
Usage
Solve this using Level 2 (Concise).
Problem: {your problem}
Stacks on the SFT model, not the raw base. Trained against the merged SFT model, so loading it straight onto
Qwen/Qwen3-4B-Instruct-2507will not reproduce the number above.
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-4B-Instruct-2507", torch_dtype="bfloat16", device_map="auto")
model = PeftModel.from_pretrained(model, "ssurface/cot-dialect-qwen3-4b-instruct-sft-l2") # 1. SFT for this level
model = model.merge_and_unload()
model = PeftModel.from_pretrained(model, "ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2") # 2. this adapter
tok = AutoTokenizer.from_pretrained("Qwen/Qwen3-4B-Instruct-2507")
Limitations
- Trained and evaluated on math word problems only.
- Accuracy falls with problem difficulty, fastest at the compressed levels.
- Single seed unless the repo name says otherwise; differences of a couple of points are within noise (95% half-width ~2.7 pp at n=1317, ~4.4 pp at n=500).
Citation
@misc{cot-compression-dialects,
title = {Chain-of-Thought Compression Dialects},
author = {Frolov, Anatolii},
year = {2026}
}
- Downloads last month
- 17
Model tree for ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2
Base model
Qwen/Qwen3-4B-Instruct-2507Dataset used to train ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2
Collection including ssurface/cot-dialect-qwen3-4b-instruct-grpo-l2
Evaluation results
- Accuracy (exact match) on GSM8Ktest set self-reported90.000