SZhanZ commited on
Commit
5cc47fc
·
verified ·
1 Parent(s): bf6f943

docs: add probe model card and feature-layer clarification

Browse files
Files changed (1) hide show
  1. README.md +76 -0
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - spatial-intelligence
6
+ - frozen-feature-probing
7
+ - semantic-tagging
8
+ - instance-grouping
9
+ - 3d-geometry
10
+ ---
11
+
12
+ # Probing-VLM-VGM Checkpoints
13
+
14
+ [Paper](https://arxiv.org/abs/2605.28132) | [GitHub](https://github.com/om-ai-lab/Probing-VLM-VGM)
15
+
16
+ Official trained probe checkpoints for **Which Pretraining Paradigm Better Serves Spatial Intelligence? An Empirical Comparison of Vision-Language and Video Generation Models**.
17
+
18
+ ## About
19
+
20
+ This repository provides **54 trained probes covering 18 foundation models and three tasks**: semantic tagging and instance grouping on ScanNet, and 3D geometry prediction on DL3DV. The probes read frozen VLM or VGM features to assess their semantic, object-level, and geometric information.
21
+
22
+ These checkpoints contain the **probe weights**, not the frozen foundation-model weights. They are compact Lightning checkpoints containing the model state dictionary and Lightning version, without optimizer or other resumable training state. Datasets and extracted features are not included.
23
+
24
+ For installation, data preparation, feature extraction, and download/evaluation commands, see the [GitHub README](https://github.com/om-ai-lab/Probing-VLM-VGM#-training-and-evaluation).
25
+
26
+ ## Repository Layout
27
+
28
+ ```text
29
+ .
30
+ |-- README.md
31
+ |-- manifest.json
32
+ |-- semantic_tagging/
33
+ | `-- <model>/
34
+ | |-- probe.ckpt
35
+ | |-- probe_config.json
36
+ | |-- metadata.json
37
+ | |-- clip_class_embeds.npy
38
+ | `-- class_names.json
39
+ |-- instance_grouping/
40
+ | `-- <model>/
41
+ | |-- probe.ckpt
42
+ | |-- probe_config.json
43
+ | `-- metadata.json
44
+ `-- dl3dv_geometry/
45
+ `-- <model>/
46
+ |-- probe.ckpt
47
+ |-- probe_config.json
48
+ `-- metadata.json
49
+ ```
50
+
51
+ Each task directory contains one subdirectory per model, such as `qwen3-vl-8b` or `wan-t2v-14b`. `probe_config.json` specifies the probe architecture; `metadata.json` records the checkpoint's actual feature settings and provenance. `manifest.json` indexes all checkpoints and their SHA-256 checksums. The semantic-tagging directories additionally include CLIP class-name embeddings and the ordered class vocabulary.
52
+
53
+ ## Feature-Layer Clarification
54
+
55
+ The paper's implementation details specify **layer 21 for Qwen2.5-VL** and **layer 22 for Qwen3-VL** in the main 3D experiments. The following uploaded checkpoints were instead trained with **layer 18** features:
56
+
57
+ | Model | Layer stated in the paper's main 3D settings | Uploaded checkpoints' feature layer |
58
+ | --- | --- | --- |
59
+ | Qwen2.5-VL-3B | 21 | 18 |
60
+ | Qwen2.5-VL-7B | 21 | 18 |
61
+ | Qwen3-VL-2B | 22 | 18 |
62
+
63
+ For these three models, the uploaded checkpoints use layer 18 across **all three tasks**. Please follow the settings in each checkpoint's `metadata.json` rather than assuming that the release configuration defaults apply: extract features with `--output-layers 18` and evaluate with `feat_postfix=_layer18`, together with the packaged probe architecture. Features from different layers are not interchangeable, even when their tensor shapes match.
64
+
65
+ Our spot checks of the uploaded probes reproduce the paper's reported metrics under matching evaluation settings. In particular, the uploaded **Qwen2.5-VL-7B layer-18 DL3DV probe** achieves **0.234 P-map Err., 0.127 depth AbsRel, and 0.310 camera AUC@30**, matching the paper's displayed values. This verifies the tested checkpoint against the reported results; it is not a claim that every layer variant has been evaluated or that different layers give identical results. Exact reproduction also depends on evaluation sampling and dependency versions.
66
+
67
+ ## Citation
68
+
69
+ ```bibtex
70
+ @article{shen2026probingvlmvgm,
71
+ title={Which Pretraining Paradigm Better Serves Spatial Intelligence? An Empirical Comparison of Vision-Language and Video Generation Models},
72
+ author={Shen, Haozhan and Zhao, Tiancheng and Zhao, Kangjia and Yin, Jianwei},
73
+ journal={arXiv preprint arXiv:2605.28132},
74
+ year={2026}
75
+ }
76
+ ```