MindCube — Plain CogMap + Free-Form Reasoning (checkpoint-57)

Full-parameter SFT of Qwen/Qwen2.5-VL-3B-Instruct on the plain_cgmap_ffr_out setting of MindCube: the model first emits a plain cognitive map of the scene, then free-form reasoning, then the answer.

This repo holds the checkpoint-57 snapshot of that run.

Training setup

Base model Qwen/Qwen2.5-VL-3B-Instruct
Task / dataset plain_cgmap_ffr_out
Learning rate 1e-5
Epochs 3
Max sequence length 8192
Image pixels min 784 / max 90000
Precision bfloat16

Usage

from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor

model_id = "Inevitablevalor/MindCube-plain-cgmap-ffr-out-ckpt57"
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
    model_id, torch_dtype="bfloat16", device_map="auto"
)
processor = AutoProcessor.from_pretrained(model_id, min_pixels=784, max_pixels=90000)

Inference follows the standard Qwen2.5-VL multi-image chat format; see the MindCube repo for the exact prompts and evaluation scripts used for this setting.

Citation

@article{yin2025mindcube,
  title   = {Spatial Mental Modeling from Limited Views},
  author  = {Yin, Baiqiao and Wang, Qineng and Zhang, Pingyue and Zhang, Jianshu and
             Wang, Kangrui and Wang, Zihan and Zhang, Jieyu and Chandrasegaran, Keshigeyan and
             Liu, Han and Krishna, Ranjay and Xie, Saining and Li, Manling and
             Wu, Jiajun and Fei-Fei, Li},
  journal = {arXiv preprint arXiv:2506.21458},
  year    = {2025}
}
Downloads last month
11
Safetensors
Model size
4B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Inevitablevalor/MindCube-plain-cgmap-ffr-out-ckpt57

Finetuned
(847)
this model

Dataset used to train Inevitablevalor/MindCube-plain-cgmap-ffr-out-ckpt57

Paper for Inevitablevalor/MindCube-plain-cgmap-ffr-out-ckpt57