RobinY99 commited on
Commit
e8526dd
·
1 Parent(s): 7cc7be6

Upload MR-IQA 2B final model

Browse files
Files changed (2) hide show
  1. README.md +25 -23
  2. model.safetensors +1 -1
README.md CHANGED
@@ -8,40 +8,42 @@ tags:
8
  - iqa
9
  - qwen3-vl
10
  - mr-iqa
11
- - grpo
12
  ---
13
 
14
- # MR-IQA 2B
15
 
16
- This repository contains the MR-IQA 2B no-thinking model from the original 2026-06-17 training run.
17
 
18
- ## Model
19
 
20
- - Base model: Qwen3-VL-2B-Instruct
21
- - Training run: `visualquality_r1_non_thinking_2b_full_zero3_dmc_rawmargin_l2_noordinal_2b_20260617_1257`
22
- - Published checkpoint: `best_model`
23
- - Best validation epoch: 3
24
- - Validation set: KONIQ 200-sample validation split with dataset seed 42
25
- - Validation result: SRCC 0.9317891406, PLCC approximately 0.94
 
26
 
27
  ## Files
28
 
29
- This upload keeps only inference-required resources:
 
 
 
 
 
 
 
 
 
 
30
 
31
- - `model.safetensors`
32
- - `config.json`
33
- - `generation_config.json`
34
- - `processor_config.json`
35
- - `tokenizer.json`
36
- - `tokenizer_config.json`
37
- - `chat_template.jinja`
38
 
39
- Training-only artifacts such as optimizer states, scheduler states, RNG states, trainer state, DeepSpeed ZeRO folders, and local path metadata are intentionally excluded.
40
 
41
- ## Prompt Format
42
 
43
- The training/evaluation prompt asks for an overall image quality score as a float between 1 and 5, rounded to two decimal places, with the final answer in `<answer>...</answer>` tags.
44
 
45
- ## Path Hygiene
46
 
47
- The uploaded model files do not require local training paths. Provide image files through your own inference or evaluation pipeline.
 
8
  - iqa
9
  - qwen3-vl
10
  - mr-iqa
 
11
  ---
12
 
13
+ # MR-IQA
14
 
15
+ MR-IQA is a 2B vision-language model for no-reference image quality assessment. It predicts a perceptual quality score from an input image and a short instruction prompt.
16
 
17
+ This repository contains the final checkpoint from the original June 17, 2026 MR-IQA 2B training run. The training data used KONIQ-style image quality supervision, and the model was validated on a 200-image KONIQ validation split with fixed data sampling.
18
 
19
+ ## Validation
20
+
21
+ Final checkpoint validation on the 200-image KONIQ validation split:
22
+
23
+ | Split | Valid samples | SRCC | PLCC | Shards |
24
+ | --- | ---: | ---: | ---: | ---: |
25
+ | KONIQ validation | 200 | 0.9250992954 | 0.9420850686 | 8 |
26
 
27
  ## Files
28
 
29
+ The repository includes only inference-ready assets:
30
+
31
+ -
32
+ -
33
+ -
34
+ -
35
+ -
36
+ -
37
+ -
38
+
39
+ Training-only artifacts such as optimizer state, scheduler state, RNG state, trainer state, and distributed checkpoint helpers are intentionally excluded.
40
 
41
+ ## Usage
 
 
 
 
 
 
42
 
43
+ Use a standard Transformers vision-language loading flow with and the Qwen3-VL model class. Provide an image and ask the model to assess overall perceptual quality. The expected score range is 1 to 5, rounded to two decimal places.
44
 
45
+ Example prompt:
46
 
 
47
 
 
48
 
49
+ The model is released as MR-IQA. Paths used during training and evaluation are not required for loading or inference.
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9a5df83fe6a82828307b02b9e1c146b3319f017badc7639e3dc1c51f75128837
3
  size 4255140312
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:586d7cd834da987fabc2c696234526782ef7e9bb1f60bbd938bdb6752fe32a33
3
  size 4255140312