Native visual reasoning, i.e., reasoning through visual generation, has recently emerged as a promising direction for studying visual intelligence beyond language. Yet progress remains bottlenecked by the lack of scalable training tasks, reliable feedback, and controlled comparisons across generative substrates. In this work, we introduce VBVR-Pro, a closed-loop testbed that makes native visual reasoning through generation trainable, verifiable, optimizable, and experimentally controllable. 1) Task scaling. VBVR-Pro turns visual reasoning into a controlled task space of 300 procedurally generated tasks. Models trained on VBVR-Pro show strong transfer beyond the proposed suite across six held-out visual reasoning benchmarks such as RISE-Video, MME-CoF-Pro, and BabyVision. Further analysis validates that these gains reflect visual reasoning rather than instruction-pattern fitting. 2) Verifiable rewards. VBVR-Pro provides verifiable reward scorers for task-grounded evaluation. Through a systematic study of leading MLLMs as judges, we identify recurring failure modes of the prevalent VLM-as-a-judge paradigm. In contrast, the proposed scorers are grounded on verifiable task-specific rules, achieve fine-grained alignment with human judgments. Importantly, they serve as reliable reward signals for large-scale multi-task reinforcement learning and demonstrate stronger post-RL performance across visual reasoning tasks. 3) Mechanism study. VBVR-Pro enables controlled modality studies across more than 30 image, video, and interleaved generators. Our analysis shows that video generation remains strongest for tasks requiring persistent spatiotemporal state tracking, while interleaved generation provides a compute-efficient alternative by externalizing intermediate visual states. Critically, ablations and probing confirm the presence of vision-native trajectories, that are a more crucial substrate than explicit linguistic chains of thought for visual reasoning. We release all data, models, scorers, and code to facilitate future research.
This repository contains a VBVR-Pro fine-tune of ThinkMorph-7B. It takes an initial image and a text instruction, reasons with interleaved text and images, and can generate one or more sequential images before returning its final answer.
The released weights are the EMA checkpoint from training step 25,000. The training checkpoint was stored in FP32; this Hugging Face export is converted to BF16, matching the precision of the original ThinkMorph-7B release.
Repository contents
File
Purpose
ema.safetensors
Complete BF16 fine-tuned ThinkMorph model state
ae.safetensors
Image autoencoder used for encoding and generation
Run the following from the ThinkMorph repository. The same downloaded directory is passed as both model_path and config_path because this release is self-contained.
python
1import random
23import numpy as np
4import torch
5from PIL import Image
67from inferencer import InterleaveInferencer
8from scripts.cza.inference import load_model
910model_path ="models/VBVR-Pro-ThinkMorph"1112random.seed(42)13np.random.seed(42)14torch.manual_seed(42)15torch.cuda.manual_seed_all(42)1617model, vae, tokenizer, token_ids, vae_transform, vit_transform = load_model(18 model_path,19 model_path,20)21inferencer = InterleaveInferencer(22 model, vae, tokenizer, vae_transform, vit_transform, token_ids
23)2425system_prompt =(26"Let's think step by step to answer the question. For text-based thinking, "27"enclose the process within <think> </think>. For visual thinking, enclose "28"the content within <image_start> </image_end>. Finally conclude with the "29"final answer wrapped in <answer> </answer>"30)3132input_image = Image.open("first_frame.png").convert("RGB")33output = inferencer(34 image=input_image,35 text="Move the object to the requested destination while preserving the scene.",36 understanding_output=False,37 think=True,38 system_prompt=system_prompt,39 max_think_token_n=4096,40 do_sample=True,41 text_temperature=0.3,42 cfg_text_scale=4.0,43 cfg_img_scale=2.0,44 cfg_interval=[0.0,1.0],45 timestep_shift=3.0,46 num_timesteps=50,47 cfg_renorm_min=0.0,48 cfg_renorm_type="text_channel",49 max_rounds=10,50)5152for index, item inenumerate(output):53ifisinstance(item, Image.Image):54 item.save(f"generated_{index}.png")55else:56print(item, end="")
The settings above reproduce the first-attempt settings used by the existing VBVR-Pro evaluator:
1git clone https://github.com/Video-Reason/VBVR-Pro.git
2cd VBVR-Pro/
3uv sync --extra cu124 # or one of [cu118|cu121|cu124|cu126|cu128|cu129]4source .venv/bin/activate
The model uses the custom ThinkMorph/BAGEL architecture and is not directly loadable with transformers.AutoModel or a Diffusers pipeline.
Inputs should follow the same image-plus-instruction format used by ThinkMorph. Generated images appear as PIL.Image.Image objects interleaved with text strings.
More generated images, larger resolutions, and longer reasoning traces increase runtime and memory use.
This checkpoint may inherit limitations and biases from its base model and fine-tuning data. Validate outputs for your application and use generated content responsibly.
Citation
bibtex
1@misc{xu2026vbvrproscalableverifiablesuite,
2 title={VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning},
3 author={Junxiang Xu and Ruisi Wang and Fanyi Pu and Maijunxian Wang and Ran Ji and Tongxi Zhou and Chenyang Gu and Jing Zuo and Hongcan Xiao and Yimeng Geng and Wanqi Yin and Wei Chen and Oscar Qian and Zhengan Yan and Ziqi Huang and Haiwen Diao and Liang Pan and Bo Li and Xiangyu Fan and Dezhi Luo and Fengyuan Yu and Zehong Zhao and Qingying Gao and Tinghui Zhu and Yilan Zhang and Jingqi Tong and Pinyuan Feng and Zhengze Jiang and Letian Wang and Ziyu Guo and Renrui Zhang and Jieneng Chen and Sonia Joseph and Constantin Venhoff and Saman Motamed and Mengyue Yang and Chandra Sripada and Alan Yuille and Philip Torr and Lvmin Zhang and Vikash Kumar and Daniel Khashabi and Nikolaus Kriegeskorte and Raphaël Millière and Vincent C. Müller and Anyi Rao and Quan Wang and Ziwei Liu and Dahua Lin and Lei Yang and Hokin Deng and Zhongang Cai},
4 year={2026},
5 eprint={2608.26105},
6 archivePrefix={arXiv},
7 primaryClass={cs.CV},
8 url={https://arxiv.org/abs/2608.26105},
9}
License
This model is released under the Apache License 2.0. See the base model repository for additional context.