RL4TG Qwen3-8B OPD GRPO

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.

Training configuration

Stage 1: offline policy distillation

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

Stage 2: mutation-aware GRPO

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.

Intended use

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.

Downloads last month
12
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for tomhu/RL4TG-Qwen3-8B-OPD-GRPO

Finetuned
Qwen/Qwen3-8B
Finetuned
(2160)
this model

Collection including tomhu/RL4TG-Qwen3-8B-OPD-GRPO