Views
No views yet
experiment_name=grpo_omr_4node_full_v1_8b_base_ppexplore_v1, global_step_250 — the best 8B OMR checkpoint of the whole controlled-exploration study.0.714 @ step 250, +4.6 pt over the cold-start GRPO baseline's peak of 0.668, and — critically — it stabilizes training: the baseline collapsed to 0.499 by step 125, while this run never dropped below 0.659 over 225 steps. Exploration was warm-started from the baseline's global_step_50.overall-6 = unweighted mean of the 6 benches. Metric = accuracy (weight-independent).global_step_250:| metric | overall-6 | mmmu | mathvista | mathverse | wemath | charxiv | dynamath |
|---|---|---|---|---|---|---|---|
| ppexplore τ0.95 @250 | 0.7138 | 0.6789 | 0.8352 | 0.8967 | 0.8132 | 0.4150 | 0.6436 |
| baseline peak @25 | 0.6681 | 0.6311 | 0.8019 | 0.8404 | 0.7437 | 0.3930 | 0.5986 |
| stock Qwen3-VL-8B ckpt-0 | 0.659 | 0.629 | 0.811 | 0.824 | 0.723 | 0.396 | 0.570 |
| Δ (explore − baseline peak) | +4.6 | +4.8 | +3.3 | +5.7 | +6.9 | +2.2 | +4.5 |
Qwen/Qwen3-VL-8B-Instruct.ngquangtrung57/verl@videorl-mods. Fully-async GRPO: FSDP2 trainer + vLLM rollouter, partial rollout, staleness-bounded off-policy.grpo_omr_4node_full_v1_8b_base_perf) global_step_50 (resume_mode=resume_path).score = 0.8·accuracy + 0.2·format (FORMAT_WEIGHT=0.2, FORMAT_MIN_THINK_CHARS=100). No KL penalty (use_kl_in_reward=false, use_kl_loss=false).ppo_mini_batch_size=16 × require_batches=4 × rollout.n=8 = 512 trajectories/step.1e-6, warmup 25 steps; total_epochs=2; clip_ratio 0.2 / 0.3 (clip_c=10.0); max_prompt_length=2048, max_response_length=16384; enforce_eager=true; gpu_memory_utilization=0.75; staleness 0.5.| key | value |
|---|---|
enable | true |
trigger_mode | high |
top_prob_threshold (τ) | 0.95 |
k_explore | 4 (of n=8 rollouts explore; 4 stay clean) |
prompt_exploration_prob | 0.5 |
deterministic | true |
perturb_prob | 1.0 |
mask_from_loss | true |
drop_top_k | 1 |
restrict_to_think_region | true |
selection_seed | 42 |
resume_path hang (4× confirmed) — keeper step_250 is well before that.verl_fully_async (entity quangtrung5705-nanyang-technological-university-singapore). The logical run spans two crash-resume segments:<think>…</think> then-answer format. Not a general-purpose chat model; not tuned for video (see the video-8b-grpo-* siblings, which are a documented dead-heat at ~0.485). No safety/RLHF alignment beyond the base model.1from transformers import AutoModelForImageTextToText, AutoProcessor
2from PIL import Image
3
4model_id = "ngqtrung/omr-8b-grpo-ppexplore"
5model = AutoModelForImageTextToText.from_pretrained(model_id, dtype="auto", device_map="auto")
6processor = AutoProcessor.from_pretrained(model_id)
7
8messages = [{
9 "role": "user",
10 "content": [
11 {"type": "image", "image": Image.open("problem.png")},
12 {"type": "text", "text": "Solve the problem. Think step by step inside <think>...</think>, then give the final answer."},
13 ],
14}]
15inputs = processor.apply_chat_template(
16 messages, add_generation_prompt=True, tokenize=True,
17 return_dict=True, return_tensors="pt"
18).to(model.device)
19out = model.generate(**inputs, max_new_tokens=2048)
20print(processor.batch_decode(out[:, inputs["input_ids"].shape[1]:], skip_special_tokens=True)[0])ngquangtrung57/verl@videorl-mods; fully-async GRPO (FSDP2 + vLLM).docs/experiments_summary_8b.md).