File size: 1,552 Bytes
6e49dc4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
---
license: apache-2.0
base_model: Qwen/Qwen3-VL-8B-Instruct
tags:
  - video-quality
  - reward-model
  - defect-detection
  - qwen3-vl
  - grpo
language:
  - en
pipeline_tag: text-generation
---

# DefectVideoReward (Qwen3-VL-8B, GRPO step800)

A video defect detection and quality scoring reward model, fine-tuned from Qwen3-VL-8B-Instruct using two-stage training:

1. **SFT**: Supervised fine-tuning on video defect annotations
2. **GRPO**: Group Relative Policy Optimization for 800 steps on pairwise video preference data

## Task

Given a text prompt and an AI-generated video, the model:
1. Identifies visible defects (artifacts, motion issues, misalignment, etc.)
2. Assigns an overall quality score (1-10)

## Input/Output Format

**Input**: System prompt + user message with video and text prompt

**Output**: JSON with defects and score
```json
{"defects": [{"issue": "...", "weight": 1|2|3}, ...], "score": 7}
```

## Usage

```python
from transformers import Qwen3VLForConditionalGeneration, AutoProcessor

model = Qwen3VLForConditionalGeneration.from_pretrained(
    "CaoruiLi/DefectVideoReward", torch_dtype="bfloat16", device_map="auto")
processor = AutoProcessor.from_pretrained("CaoruiLi/DefectVideoReward")
```

## Training Details

- **Base model**: Qwen/Qwen3-VL-8B-Instruct
- **SFT data**: Video defect annotations with expert-labeled defect lists
- **GRPO data**: 400 text prompts × pairwise video comparisons from VideoGen-RewardBench
- **GRPO steps**: 800 (best checkpoint by validation accuracy)
- **KL coefficient**: 0.01