AIcell commited on
Commit
2dc0442
·
verified ·
1 Parent(s): 3f6f1fc

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +141 -0
README.md ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen2.5-VL-3B-Instruct
4
+ tags:
5
+ - multimodal
6
+ - vision-language
7
+ - visual-reasoning
8
+ - reinforcement-learning
9
+ - qwen2.5-vl
10
+ - math
11
+ - reasoning
12
+ datasets:
13
+ - OpenMMReasoner-Data
14
+ language:
15
+ - en
16
+ pipeline_tag: image-text-to-text
17
+ library_name: transformers
18
+ ---
19
+
20
+ # Frankenstein-RL
21
+
22
+ **Frankenstein-RL** is the reinforced (reinforcement training after cold-start initialization) model from the paper:
23
+
24
+ > **[What does RL improve for Visual Reasoning? A Frankenstein-Style Analysis](https://arxiv.org/abs/2602.12395)**
25
+ >
26
+ > Xirui Li\*, Ming Li\*, Tianyi Zhou
27
+ >
28
+ > University of Maryland  |  Mohamed bin Zayed University of Artificial Intelligence
29
+ >
30
+ > *(\* Co-first Authors)*
31
+
32
+ This model serves as the **IN (Instruction-tuned) checkpoint** before reinforcement learning, built on the [OpenMMReasoner](https://arxiv.org/abs/2511.16334) training recipe with [Qwen2.5-VL-3B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-3B-Instruct) as the base model.
33
+
34
+ ## Overview
35
+
36
+ Our paper introduces a **Frankenstein-style analysis framework** to understand *what* reinforcement learning (RL) actually improves in vision-language models (VLMs) for visual reasoning. Rather than relying on end-to-end benchmark scores, we decompose VLMs at the granularity of transformer layers and probe their functional roles through:
37
+
38
+ 1. **Functional Localization via Causal Probing** — localizing vision- and reasoning-related computations along transformer depth
39
+ 2. **Update Characterization via Parameter Comparison** — showing that IN and RL differ systematically in update magnitude and geometry
40
+ 3. **Transferability Test via Model Merging** — transplanting RL-refined regions into IN models to test causal contributions
41
+
42
+ ### Key Findings
43
+
44
+ - RL does **not** uniformly improve visual perception or standalone reasoning
45
+ - RL induces **structured refinements concentrated in mid-to-late layers**, improving vision-to-reasoning alignment
46
+ - These mid-to-late refinements are both **transferable** (via merging) and **necessary** (via freezing) for RL gains
47
+ - Freezing **late layers** during RL training leads to a pronounced drop in reasoning performance
48
+
49
+ ## Evaluation Results
50
+
51
+ ### Fine-grained and Benchmark Metrics
52
+
53
+ | Model | Vision (M_vis) | Vision-to-Reasoning (M_v2r) | Reasoning (M_rea) | MathVista | MathVision | MathVerse |
54
+ |:---|:---:|:---:|:---:|:---:|:---:|:---:|
55
+ | **Frankenstein-IN** (this model) | 34.0 | 21.0 | 26.0 | 46.5 | 18.4 | 37.0 |
56
+ | Frankenstein-RL | 33.0 | 29.0 | 34.0 | 48.1 | 14.1 | 37.8 |
57
+
58
+ ### Parameter Freezing Analysis (RL Training)
59
+
60
+ | Model | Vision (M_vis) | Vision-to-Reasoning (M_v2r) | Reasoning (M_rea) | MathVista | MathVision | MathVerse |
61
+ |:---|:---:|:---:|:---:|:---:|:---:|:---:|
62
+ | RL - Frozen **Early** Block | **35.0** | **31.0** | 36.0 | **48.2** | **21.0** | 34.5 |
63
+ | RL - Frozen **Mid** Block | 25.0 | 29.0 | **38.0** | 46.5 | 15.5 | **35.7** |
64
+ | RL - Frozen **Late** Block | 30.0 | 27.0 | 34.0 | 47.9 | 16.8 | 35.0 |
65
+
66
+ ## Quick Start
67
+
68
+ ### Installation
69
+
70
+ ```bash
71
+ pip install transformers accelerate
72
+ pip install qwen-vl-utils[decord]==0.0.8
73
+ ```
74
+
75
+ ### Inference
76
+
77
+ ```python
78
+ from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
79
+ from qwen_vl_utils import process_vision_info
80
+
81
+ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
82
+ "AIcell/Frankenstein-IN",
83
+ torch_dtype="auto",
84
+ device_map="auto",
85
+ )
86
+
87
+ processor = AutoProcessor.from_pretrained("AIcell/Frankenstein-IN")
88
+
89
+ messages = [
90
+ {
91
+ "role": "user",
92
+ "content": [
93
+ {"type": "image", "image": "https://your-image-url.jpg"},
94
+ {"type": "text", "text": "Please solve this math problem step by step."},
95
+ ],
96
+ }
97
+ ]
98
+
99
+ text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
100
+ image_inputs, video_inputs = process_vision_info(messages)
101
+ inputs = processor(
102
+ text=[text],
103
+ images=image_inputs,
104
+ videos=video_inputs,
105
+ padding=True,
106
+ return_tensors="pt",
107
+ ).to(model.device)
108
+
109
+ generated_ids = model.generate(**inputs, max_new_tokens=2048)
110
+ generated_ids_trimmed = [
111
+ out_ids[len(in_ids):] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
112
+ ]
113
+ output_text = processor.batch_decode(
114
+ generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
115
+ )
116
+ print(output_text[0])
117
+ ```
118
+
119
+ ## Related Resources
120
+
121
+ | Resource | Link |
122
+ |:---|:---|
123
+ | Paper | [arXiv:2602.12395](https://arxiv.org/abs/2602.12395) |
124
+ | Frankenstein-RL Model | [AIcell/Frankenstein-RL](https://huggingface.co/AIcell/Frankenstein-RL) |
125
+ | Base Model | [Qwen/Qwen2.5-VL-3B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-3B-Instruct) |
126
+ | OpenMMReasoner | [arXiv:2511.16334](https://arxiv.org/abs/2511.16334) |
127
+
128
+ ## Citation
129
+
130
+ ```bibtex
131
+ @article{li2026frankenstein,
132
+ title={What does RL improve for Visual Reasoning? A Frankenstein-Style Analysis},
133
+ author={Li, Xirui and Li, Ming and Zhou, Tianyi},
134
+ journal={arXiv preprint arXiv:2602.12395},
135
+ year={2026}
136
+ }
137
+ ```
138
+
139
+ ## License
140
+
141
+ This model is released under the [Apache 2.0 License](https://www.apache.org/licenses/LICENSE-2.0).