TRACE Qwen2.5-VL 3B

Name: trace-qwen2.5-vl-3b
Availability: InStock

TRACE Qwen2.5-VL 3B is a GRPO checkpoint derived from Qwen/Qwen2.5-VL-3B-Instruct and trained on 64,000 grounded visual-reasoning examples spanning 1,000 tasks from maveryn/trace. This repository contains the merged step-500 inference checkpoint.

Paper · Project page · GitHub · Collection · Training configuration · Evaluation suite · Run artifacts

Training and provenance

Field	Value
Base model	`Qwen/Qwen2.5-VL-3B-Instruct@66285546d2b821cf421d4f5eb2576359d3770cd3`
Dataset	`maveryn/trace@4e5b54361360296a855542b40cfd8b7f81b355fe`
Method	GRPO, 500 steps, global/rollout batch 128, 8 responses per prompt
Optimization	AdamW, learning rate `1e-6`, constant schedule, KL disabled
Reward	`0.95 × exact JSON answer + 0.05 × valid JSON format`; annotation reward `0`
Reference run	8 NVIDIA H100 80GB GPUs
Released artifact	Merged step-500 inference weights

The released training profile reads prompt_answer, scores answer_gt, and does not use the advisory trace_supervision_mode column. The consumed fields, embedded image bytes, and row order in the current dataset release were verified identical to the original training input in the public equivalence receipt.

The canonical output ends with {"answer": ...}. Exact hashes, source revision, and run provenance are recorded in trace_training_provenance.json and .trace_model_revision.json. The repository does not include optimizer, scheduler, trainer, or FSDP state for continuation.

Evaluation

trace_eval_v1 evaluates 24 external benchmarks and 32,805 examples per model and decoding seed. Scores below are the unweighted macro mean of the 24 benchmark percentages, reported as mean ± sample standard deviation across seeds 42, 43, and 44.

Model	Overall score
Qwen2.5-VL-3B-Instruct	39.34 ± 0.63
TRACE Qwen2.5-VL 3B	42.85 ± 0.39
Paired improvement	+3.51 ± 0.25

Per-benchmark scores, model revisions, aggregation, and benchmark provenance are available in the public results.json and evaluation documentation.

Usage

python

from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration

model_id = "maveryn/trace-qwen2.5-vl-3b"
revision = "2ec2374d5c219e6b12e26bda93d3b3adeb1e30c5"
image_url = "https://raw.githubusercontent.com/maveryn/trace/main/docs/assets/paper-domain-montage/trace-paper-domain-montage.png"

processor = AutoProcessor.from_pretrained(model_id, revision=revision)
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
    model_id,
    revision=revision,
    dtype="auto",
    device_map="auto",
)

messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "url": image_url},
            {"type": "text", "text": "How many visual domains are shown with example images? Respond with only a JSON object using the key \"answer\"."},
        ],
    }
]
inputs = processor.apply_chat_template(
    messages,
    tokenize=True,
    add_generation_prompt=True,
    return_dict=True,
    return_tensors="pt",
).to(model.device)
output_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids = [
    output[len(prompt) :] for prompt, output in zip(inputs.input_ids, output_ids)
]
print(processor.batch_decode(generated_ids, skip_special_tokens=True)[0])

Verified reference answer

For the published montage used above, the expected output is:

{"answer": 11}

The value is backed by the committed montage manifest (layout.panel_count). This is reference ground truth, not a claim about a particular decoding run; publish qualitative model outputs only with their recorded inference settings.

The pinned revision is the checkpoint used by the canonical evaluation; later repository heads may update documentation without changing the model weights.

Intended use and limitations

This checkpoint is intended for research on multimodal reasoning and verifiable-reward post-training. It can produce incorrect answers, unsupported reasoning, or unreliable grounding. It was trained on synthetic tasks and has not been validated for safety-critical, medical, legal, or autonomous decision-making uses. Results depend on the recorded prompts, decoding, parsers, and scorers.

License

This checkpoint is subject to the upstream Qwen Research License.