SeongryongJung's picture
Upload Physics RLSD Qwen3-4B checkpoints
df44973 verified
|
Raw
History Blame Contribute Delete
3.35 kB
metadata
license: apache-2.0
base_model: Qwen/Qwen3-4B
library_name: transformers
pipeline_tag: text-generation
tags:
  - qwen3
  - physics
  - reinforcement-learning
  - verl
  - sciknoweval
  - rlsd
  - self-distillation

Qwen3-4B Physics RLSD

This repository contains Physics fine-tuned Qwen3-4B checkpoints from the local SciKnowEval-style generalization setup.

  • Root checkpoint: final global_step_100 merged to Hugging Face safetensors.
  • best_avg16/: checkpoint with the highest validation avg@16 during training, merged to Hugging Face safetensors.

Checkpoints

Checkpoint Source step Validation avg@16 best@16 / pass@16 maj@16
Root final 100 0.700000 0.768713 0.718600
best_avg16/ 60 0.731250 0.816338 0.740938

Training Run

qwen3gen-physics-RLSD-Qwen-Qwen3-4B-mbs8-decay0-ema0.05-train256-rollout8-lr1e-6-vllm0.8

Base Model

  • Base model: Qwen/Qwen3-4B
  • Fine-tuning type: full-parameter FSDP RL training
  • Dataset: datasets/sciknoweval/physics
  • Train split: 720 examples
  • Validation split: 80 examples

Method

  • Method: RLSD
  • Config: rlsd
  • Policy loss mode: rlsd
  • Reward: local SciKnowEval multiple-choice reward checker
  • Rollout correction: token-level importance sampling, threshold 2.0

Hyperparameters

Field Value
Base model Qwen/Qwen3-4B
Training steps 100
Train batch size 256
Rollouts per prompt 8
Generations per step 2048
PPO mini batch size 8
Learning rate 1e-6
LR warmup steps 10
Weight decay 0.01
Grad clip 1.0
Max prompt length 2048
Max response length 8192
Max model length 10240
Train temperature 1.0
Train top_p 1.0
Validation generations 16
Validation temperature 0.6
Validation top_p 0.95
vLLM GPU memory utilization 0.8
GPUs 8 x NVIDIA H200
Save frequency every 10 steps
Validation frequency every 10 steps
Token reweight lambda 0.5
Token reweight eps_w 0.2
Token reweight decay steps 0
Teacher update rate 0.05
Max reprompt length 10240

Metrics

Metric Value
Final training step 100
Final critic/score/mean 0.890137
Final critic/rewards/mean 0.890137
Final validation avg@16 0.700000
Peak validation avg@16 0.731250
Peak validation step 60

Loading

Root final checkpoint:

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("SeongryongJung/Qwen3-4B-Physics-RLSD")
tokenizer = AutoTokenizer.from_pretrained("SeongryongJung/Qwen3-4B-Physics-RLSD")

Best avg@16 checkpoint:

model = AutoModelForCausalLM.from_pretrained("SeongryongJung/Qwen3-4B-Physics-RLSD", subfolder="best_avg16")
tokenizer = AutoTokenizer.from_pretrained("SeongryongJung/Qwen3-4B-Physics-RLSD", subfolder="best_avg16")

Intended Use

This model is intended for research on RL fine-tuning and self-distillation behavior on science/generalization tasks. It has not been broadly safety evaluated for production use.

Limitations

The reported scores are training-time and validation-time metrics from the local experimental setup. They should not be interpreted as broad benchmark results without independent evaluation.