Downloads · 30 days
9
15% of all-time downloads
ngqtrung/video-8b-grpo-ppexplore-n16k8
video-8b-grpo-ppexplore-n16k8 is a image-text-to-text model from ngqtrung. Use it for the image-text-to-text task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
RLVR post-training of Qwen/Qwen3-VL-8B-Instruct on multiple-choice video QA with fully-async GRPO and entropy-aware token-dropout exploration ("ppexplore", τ=0.8) at the rollout stage, using bigger GRPO groups (rollou…
Downloads · 30 days
9
15% of all-time downloads
All-time downloads
61
Public
Parameters
8.8B
17.5 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors17.5 GB · 100%
From the Hugging Face model README
RLVR post-training of Qwen/Qwen3-VL-8B-Instruct on multiple-choice video QA with fully-async GRPO and entropy-aware token-dropout exploration ("ppexplore", τ=0.8) at the rollout stage, using bigger GRPO groups (rollout.n=16, k_explore=8). Warm-started from the SFT-770-RL global_step_60 checkpoint. This is experiment_name=grpo_video_4node_full_v3_24f100k_8b_sft770_rl60_ppexplore_tau08_n16k8_v1, global_step_80 (the keeper).
Group-size ablation — also a WASH: bigger GRPO groups (n16/k8) peak at offline full-set val mean_accuracy 0.4913 @ step 80, then clean-fade. This ties the best-prior keeper (cold-base 0.4918) but does not break the ~0.485 video dead-heat. GRPO group size is not the missing lever — this run closes the 8B video group-size question (7-way dead-heat overall).
Offline full-set eval (VideoMME-v1 2700 + PerceptionComp 1108 + Video-Holmes 1837 = 5645 rows), scored with the repo's vero compute_score. mean = macro-mean of the 3 bench accuracies.
Keeper = global_step_80 (peak):
| metric | mean | videomme | holmes | perceptioncomp |
|---|---|---|---|---|
| τ0.8 n16/k8 @80 (keeper) | 0.4913 | 0.653 | 0.471 | 0.349 |
| cold-base RL keeper @80 | 0.4918 | 0.658 | 0.474 | 0.343 |
| τ0.95 sibling peak (n8/k4) | 0.4881 | — | — | — |
| τ0.8 sibling peak (n8/k4) | 0.4864 | — | — | — |
| stock Qwen3-VL-8B ckpt-0 (zero-shot) | 0.4444 | 0.6426 | 0.4143 | 0.2762 |
Trajectory (mean): 0.488 @20 → 0.490 @40 → 0.487 @60 → 0.491 @80 → clean monotone fade (0.481 @120, 0.473 @140), no collapse. (Step 100 eval skipped — reproducibly hung on one deterministic shard, step-100-specific not systemic.) Peak edges the n8 explore siblings by ~0.3 pt = within noise.
Qwen/Qwen3-VL-8B-Instruct, warm-started from the SFT-770-RL run's global_step_60 (merged HF _merged/sft770_perf_gs60_hf, itself val 0.4814). Lineage: stock 8B → OMR-SFT-770 → video-RL gs60 → this run.ngquangtrung57/verl@videorl-mods. Fully-async GRPO: FSDP2 trainer + vLLM rollouter.score = 0.8·accuracy + 0.2·format (FORMAT_WEIGHT=0.2, FORMAT_MIN_THINK_CHARS=100). No KL penalty.ppo_mini_batch_size=16 × require_batches=4 × rollout.n=16 = 1024 rollouts/fit-step (vs 512 at n=8; bigger GRPO groups = 8 explore + 8 anchor per selected prompt).GROUP_VIDEO_TRAIN_MC_24F100K (5 video-MC parquets, 24 frames / 100k pixels).1e-6, warmup 25 steps; total_epochs=2; clip_ratio 0.2 / 0.3 (clip_c=10.0); max_prompt_length=17408, max_response_length=16384; enforce_eager=true; gpu_memory_utilization=0.75; staleness 0.4. No OOM at the 256-traj actor update (token-bounded micro-batch held).| key | value |
|---|---|
enable | true |
trigger_mode | high |
top_prob_threshold (τ) | 0.8 |
k_explore | 8 (of n=16; 8 explore + 8 anchor) |
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 |
test_freq=10000); video val is offline full-set eval on a dedicated H100 node (vLLM TP1), every 20 fit-steps.Project verl_fully_async (entity quangtrung5705-nanyang-technological-university-singapore). The logical run has three crash-resume segments (vtlkaq8u, iwwcf7do, hb4vi5h3); the last/most-complete (hb4vi5h3) carries the curated label. Train metrics only — video val is offline, not on W&B:
https://wandb.ai/quangtrung5705-nanyang-technological-university-singapore/verl_fully_async/runs/hb4vi5h3
Research checkpoint — the group-size ablation arm of an 8B video study. Headline finding: bigger GRPO groups (n16/k8) do not break the ~0.485 video dead-heat (peak 0.4913 ties the best keeper then fades). With this run, seven levers (SFT warmstart, cold-base RL, cold exploration, base-warm explore, SFT-warm explore τ0.95, SFT-warm explore τ0.8, and this τ0.8 n16/k8) all peak ~0.485-0.49 — the video ceiling is bound by something outside the exploration / warm-start / RL-duration / τ / group-size space. Multiple-choice video QA, <think>…</think> then-answer format. No safety/RLHF alignment beyond the base.
from transformers import AutoModelForImageTextToText, AutoProcessor
model_id = "ngqtrung/video-8b-grpo-ppexplore-n16k8"
model = AutoModelForImageTextToText.from_pretrained(model_id, dtype="auto", device_map="auto")
processor = AutoProcessor.from_pretrained(model_id)
messages = [{
"role": "user",
"content": [
{"type": "video", "video": "clip.mp4"},
{"type": "text", "text": "Answer the multiple-choice question. Reason inside <think>...</think>, then give the final letter."},
],
}]
inputs = processor.apply_chat_template(
messages, add_generation_prompt=True, tokenize=True,
return_dict=True, return_tensors="pt"
).to(model.device)
out = model.generate(**inputs, max_new_tokens=1024)
print(processor.batch_decode(out[:, inputs["input_ids"].shape[1]:], skip_special_tokens=True)[0])
global_step_60 (lineage: stock 8B → OMR-SFT-770 → video-RL gs60).ngquangtrung57/verl@videorl-mods; fully-async GRPO (FSDP2 + vLLM).docs/experiments_summary_8b.md).