Downloads · 30 days
10
21% of all-time downloads
tomhu/RL4TG-Qwen3-8B-OPD-GRPO
RL4TG-Qwen3-8B-OPD-GRPO is a text generation model from tomhu. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
This model is a Qwen3-8B checkpoint trained for Java unit-test generation with offline policy distillation (OPD), followed by mutation-aware GRPO. It is a selected release state from the GRPO run. The repository name…
Downloads · 30 days
10
21% of all-time downloads
All-time downloads
48
Public
Parameters
8.2B
16.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors16.4 GB · 100%
From the Hugging Face model README
This model is a Qwen3-8B checkpoint trained for Java unit-test generation with offline policy distillation (OPD), followed by mutation-aware GRPO. It is a selected release state from the GRPO run. The repository name intentionally does not encode an internal checkpoint number.
| Setting | Value |
|---|---|
| Student | Qwen/Qwen3-8B |
| Teacher | Qwen/Qwen3-14B |
| Objective | forward KL over teacher top-k logits |
| Teacher top-k | 256 |
| Policy-gradient term | disabled |
| Precision | BF16 |
| Learning rate | 3e-6, constant schedule |
| Epochs | 1 |
| Global batch size | 32 |
| Per-GPU micro batch | 1 |
| Gradient accumulation | 32 |
| Rollout samples | 1 |
| Temperature | 1.0 |
| Maximum prompt/response length | 4096 / 4096 tokens |
| Seed | 42 |
| Setting | Value |
|---|---|
| Starting policy | OPD model from stage 1 |
| Training examples | 1,582 mutation-applicable Defects4J training samples |
| Algorithm | GRPO, no KL term in reward |
| Configured epochs | 2 |
| Learning rate | 1e-6 |
| Global prompt batch | 32 |
| Rollouts per prompt | 8 |
| PPO prompt groups / expanded mini batch | 4 / 32 |
| Per-GPU micro batch | 2 |
| Gradient accumulation | 8 |
| Temperature | 1.0 |
| Entropy coefficient | 0.01 |
| Maximum prompt/response length | 4096 / 4096 tokens |
| Data shuffle / seed | enabled / 42 |
| Chat template | enabled; Qwen thinking disabled |
The reward is -1 for an invalid generated test. A valid test receives
0.4 * line_coverage + 0.6 * branch_coverage + mutation_score. Mutation
analysis samples up to three eligible failing mutants per rollout, with a
mutation weight of 1.0. Coverage credit is gated by focal-call, assertion, and
local focal-data provenance checks.
The model is intended for research on Java unit-test generation and reinforcement learning for software engineering. Generated tests must be compiled and executed in an isolated environment before use.