VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning

Project Page Code Code arXiv Dataset Bench Data Leaderboard

Overview

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.

The models are presented in the paper VBVR-Pro: A Scalable and Verifiable Suite for Native Visual Reasoning.

Models Zoo

Model Base Architecture Other Remarks
Image Generation Models
VBVR-Pro-BAGELBAGEL-7B-MoTComplete model
VBVR-Pro-FLUX2-devFLUX.2-devComplete model, Diffusers format
VBVR-Pro-FLUX2-dev-diffsynthFLUX.2-devLoRA model, DiffSynth format
VBVR-Pro-Qwen-Image-EditQwen-Image-Edit-2511Complete model, Diffusers format
VBVR-Pro-Qwen-Image-Edit-diffsynthQwen-Image-Edit-2511LoRA model, DiffSynth format
Interleaved Image Generation Models
VBVR-Pro-ThinkMorphThinkMorph-7BComplete model
VBVR-Pro-SenseNova-U1SenseNova-U1-8B-MoTComplete model
Video Generation Models
VBVR-Pro-LTX2.3LTX-Video-2.3Complete model, Diffusers format
VBVR-Pro-LTX2.3-diffsynthLTX-Video-2.3LoRA model, DiffSynth format
VBVR-Pro-Wan2.1-I2V-14BWan2.1-I2V-14B-720PComplete model, Diffusers format
VBVR-Pro-Wan2.1-I2V-14B-diffsynthWan2.1-I2V-14B-720PLoRA model, DiffSynth format
VBVR-Pro-Wan2.2-I2V-A14BWan2.2-I2V-A14BComplete model, Diffusers format
VBVR-Pro-Wan2.2-I2V-A14B-diffsynthWan2.2-I2V-A14BLoRA model, DiffSynth format
VBVR-Pro-Wan2.2-TI2V-5BWan2.2-TI2V-5BComplete model, Diffusers format
VBVR-Pro-Wan2.2-TI2V-5B-diffsynthWan2.2-TI2V-5BLoRA model, DiffSynth format

Release Information

This repository contains the Hugging Face export of the EMA checkpoint at training step 30,000 from neo_old_script_vbvr_pro. The model takes an initial image and a text instruction and generates one or more sequential keyframes.

The checkpoint uses custom Transformers code. Loading it therefore requires trust_remote_code=True. Review the Python files in this repository before loading code from an untrusted copy.

In this release, we present all models presented in paper VBVR-Pro-Trained-Models, VBVR-Pro-Dataset-Video, VBVR-Pro-Dataset-Image, VBVR-Pro-Bench, VBVR-Pro-Code and VBVR-Bench-Leaderboard.

VBVR-Pro Benchmark Results

Models Overall In-Domain by Category Out-of-Domain by Category
Avg.Abst.Know.Perc.Spat.Trans. Avg.Abst.Know.Perc.Spat.Trans.
Image Generation Models
Proprietary Models
Qwen-Image-2.00.3130.2480.2690.1960.2250.1700.1320.3780.3410.2350.3910.3840.080
Seedream-5.0-Pro0.5570.4850.5180.3120.5090.4010.2170.6290.5070.4550.6610.5590.202
Open-source Models
BAGEL-7B-MoT0.0890.0660.0390.0850.0670.0460.0270.1110.2010.0310.0730.0280.121
FLUX.2-dev0.1570.1080.0880.1090.0720.1000.0660.2060.1970.1650.1840.2410.077
Qwen-Image-Edit0.1340.1080.0920.0820.1000.1090.0560.1590.1760.0630.1410.1820.082
Strong Baselines
VBVR-Pro-BAGEL0.1720.1680.1990.1050.1100.2130.0550.1760.2540.1040.1480.0150.145
VBVR-Pro-FLUX.20.4070.4840.4830.3230.3670.4490.3360.3300.3610.2720.2550.4540.128
VBVR-Pro-Qwen-Image0.3220.3320.2980.2170.1930.4310.2220.3110.3410.2390.2330.4130.181
Interleaved Image Generation Models
Proprietary Models
GPT-Image-20.5070.4280.4560.3180.4280.2060.3000.5870.3980.4130.6330.4800.303
Nano Banana Pro0.5640.4800.5180.4220.5120.2850.1740.6480.5530.4990.6570.5850.220
Open-source Models
ThinkMorph-7B0.1540.1130.1000.0820.1010.1480.0310.1950.1760.1660.1630.2530.103
VBVR-SenseNova-U10.4080.4690.3560.3130.3730.3860.4770.3470.2910.3170.2750.4800.238
SenseNova-U1-8B-MoT0.5650.5330.5010.3950.5440.3550.3490.5970.4480.4950.5330.7170.401
Strong Baselines
VBVR-Pro-ThinkMorph0.3730.4020.4030.3440.2380.4540.1840.3440.3670.2240.2380.5350.257
VBVR-Pro-SenseNova-U10.6380.8110.6480.6950.6210.7700.5410.4640.4800.3280.3440.5580.408
Video Generation Models
Proprietary Models
Veo 3.10.3090.3120.2750.2990.2520.2670.1570.3050.3050.2330.2520.3120.219
Kling V30.3920.3560.2130.3260.3200.3550.2290.4270.2940.5640.3750.2420.412
SeedDance 2.00.4990.4510.3380.3610.3530.4680.3080.5470.3690.5110.4780.5380.532
Open-source Models
HunyuanVideo-I2V0.0540.0540.0230.0640.0150.0840.0320.0530.0880.0140.0280.0620.055
CogVideoX1.5-5B-I2V0.0850.1000.0610.1180.0690.0920.0600.0700.1250.0380.0510.0400.024
Wan2.1-I2V-14B0.1000.1050.0520.1250.0910.1020.0520.0950.1120.0730.0710.1230.044
Wan2.2-TI2V-5B0.0940.0660.0290.0730.0500.0830.0310.1220.1560.0520.1060.0630.099
Wan2.2-I2V-14B-720P0.1820.1570.0820.1310.1100.1610.1560.2070.2240.1390.1400.1950.273
LTX2.3-I2AV0.1120.1060.0620.1090.0700.1330.0550.1190.1610.1350.0860.0910.050
VBVR-Wan2.20.5170.5480.2370.4990.3340.5660.5910.4860.3100.3430.3450.7320.684
Strong Baselines
VBVR-Pro-LTX2.30.4250.5270.4090.5100.3460.4600.3900.3240.3810.1080.2010.4770.386
VBVR-Pro-Wan2.1-I2V-14B0.5620.7300.6170.5800.4520.6760.6230.3950.4100.3050.2300.6170.439
VBVR-Pro-Wan2.2-TI2V-5B0.4700.6410.5280.5560.3730.5650.5570.3000.3330.1270.1610.5050.409
VBVR-Pro-Wan2.2-I2V-14B0.6700.8080.6320.6850.5560.7510.6360.5320.4790.4180.3500.6790.690

Quick Start

Method 1: Standalone Transformers inference

Requirements

  • Linux and a CUDA-capable GPU
  • Python 3.10 or newer
  • PyTorch 2.5 or newer, with a matching TorchVision build
  • Transformers 4.51 or newer (Transformers 5 is not currently supported)

Install PyTorch and TorchVision for your CUDA version first, then install the remaining packages:

pip install "transformers>=4.51,<5" safetensors pillow numpy pandas

The checkpoint contains about 36 GB of bfloat16 weights. An 80 GB-class GPU is recommended for straightforward single-GPU inference; activation memory depends strongly on output resolution and the number of generated frames.

Run inference

Download this repository or pass its Hugging Face repository ID directly to the included script:

python inference.py \
  --model Video-Reason/VBVR-Pro-SenseNova-U1 \
  --input first_frame.png \
  --prompt "Move the object to the requested destination while preserving the scene." \
  --num-images 3 \
  --width 512 \
  --height 512 \
  --output-dir outputs

This writes frame_1.png, frame_2.png, and frame_3.png under outputs/. Both output dimensions must be positive multiples of 32.

The equivalent core API is:

import numpy as np
import torch
from PIL import Image
from transformers import AutoModel, AutoTokenizer

model_id = "Video-Reason/VBVR-Pro-SenseNova-U1"
device = "cuda:0"

torch.manual_seed(42)
torch.cuda.manual_seed_all(42)

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModel.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
    trust_remote_code=True,
).to(device).eval()

first_frame = Image.open("first_frame.png").convert("RGB")
num_images = 3

with torch.inference_mode():
    frames = model.interleave_gen_image_only(
        tokenizer,
        "Move the object to the requested destination while preserving the scene.",
        gt_text="<image>" * num_images,
        images=[first_frame],
        image_size=(512, 512),  # (width, height)
        max_images=num_images,
        num_steps=50,
        cfg_scale=1.0,
        img_cfg_scale=1.0,
        timestep_shift=1.0,
    )

for index, frame in enumerate(frames, start=1):
    image = (frame.float() * 0.5 + 0.5).clamp(0, 1)
    array = (
        image[0].permute(1, 2, 0).cpu().numpy() * 255.0
    ).round().astype(np.uint8)
    Image.fromarray(array).save(f"frame_{index}.png")

gt_text controls how many image slots are generated: use one <image> token per requested output frame. Generated frames are sequential: each generated frame is encoded back into the context before the next frame is produced.

Method 2: Unified VBVR-Pro inference

Clone Video-Reason/VBVR-Pro and create its unified inference environment:

git clone https://github.com/Video-Reason/VBVR-Pro.git
cd VBVR-Pro/
uv sync --extra cu124 # or one of [cu118|cu121|cu124|cu126|cu128|cu129]
source .venv/bin/activate

Then run the unified inference script:

python example.py \
  --model_path Video-Reason/VBVR-Pro-SenseNova-U1 \
  --image_paths first_frame.png \
  --prompt "Show the next three steps of the action." \
  --num_images 3 --width 512 --height 512 \
  --output outputs

Settings used by the existing VBVR-Pro evaluator

The previous evaluation path loads this EMA export with AutoModel and AutoTokenizer, then calls interleave_gen_image_only with the following defaults:

Setting Value
Denoising steps 50
Text CFG scale 1.0
Image CFG scale 1.0
Timestep shift 1.0
Seed 42
Input first_frame.png plus prompt.txt
Output count Number of reference frame_N.png files

For benchmark evaluation, the evaluator removes literal <image> placeholders from the prompt, uses the reference keyframe dimensions after resizing them to multiples of 32, and requests one output image per reference keyframe. The included CLI exposes the same generation API but uses one explicit output size for all frames.

Notes

  • This is a custom Neo-Unify image-generation checkpoint, not a Diffusers or Wan checkpoint.
  • The model is intended for bfloat16 CUDA inference. CPU inference is not supported by the included script.
  • Higher resolutions and additional output frames increase runtime and memory use substantially.
  • Use generated content responsibly and follow the terms that accompany the eventual Hugging Face repository release.
Downloads last month
-
Safetensors
Model size
19B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Paper for Video-Reason/VBVR-Pro-SenseNova-U1