HugC commited on
Commit
abe83ba
·
verified ·
1 Parent(s): 3f2bec8

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +23 -23
README.md CHANGED
@@ -13,39 +13,39 @@ library_name: transformers
13
 
14
  # VisME (Qwen2.5-VL-7B)
15
 
16
- [English](README-en.md) | 简体中文
17
 
18
- **VisME** 是我们 CVPR 2026 论文 [*Illuminating Visual Identity in Universal Multimodal Embeddings*](https://openaccess.thecvf.com/content/CVPR2026/html/Cao_Illuminating_Visual_Identity_in_Universal_Multimodal_Embeddings_CVPR_2026_paper.html) 提出的通用多模态嵌入模型。VisME 基于 [Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct) 构建,可为图片、文本以及图文对生成稠密嵌入,在通用多模态基准(**MMEB**)和视觉 ID 检索(**MVEB**)上均有出色表现。
19
 
20
  <p align="center">
21
  <img src="assets/teaser.jpg" alt="VisME teaser" width="95%">
22
  </p>
23
 
24
- ## 亮点
25
 
26
- - **ID 感知检索** — 通过 [MVEB](https://chrisclear3.github.io/MVEB/) 基准(4 类元任务、28 个测试子集)引入视觉 ID 监督进行训练。
27
- - **通用能力强** — **MMEB-v1** **MVEB** 上联合训练,在标准 MMEB 任务上保持有竞争力的表现。
28
 
29
- ## 性能
30
 
31
- 下图为 **MMEB**Cls / VQA / Ret / Grd)与 **MVEB**ID-Rec / Re-ID / ID-Grd / ID-Edit)上的结果。指标与论文一致(各任务组内所有数据集的平均 Precision@1)。
32
 
33
  <p align="center">
34
  <img src="assets/performance.jpg" alt="VisME performance on MMEB and MVEB" width="95%">
35
  </p>
36
 
37
- ## 训练数据
38
 
39
- VisME 在以下数据的混合集上训练:
40
 
41
- | 基准 | 作用 |
42
- |------|------|
43
- | [**MMEB-v1**](https://github.com/TIGER-AI-Lab/VLM2Vec) | 通用多模态嵌入任务(分类、VQA、检索、定位) |
44
- | [**MVEB**](https://chrisclear3.github.io/MVEB/) | 覆盖识别、Re-ID、定位与编辑的视觉 ID 检索 |
45
 
46
- **MVEB** 数据集已开源,训练代码即将发布。
47
 
48
- ## 使用方法
49
 
50
  ```python
51
  import torch
@@ -79,9 +79,9 @@ batch = [
79
  ]
80
 
81
  with torch.no_grad():
82
- embeddings = model.encode_input(batch) # 形状: (3, 3584),已做 L2 归一化
83
 
84
- # 两两余弦相似度
85
  # SpiderMan_comic_E616 <-> SpiderMan_promo_E199999
86
  # SpiderMan_comic_E616 <-> Toxin
87
  # SpiderMan_promo_E199999 <-> Toxin
@@ -91,13 +91,13 @@ for i in range(len(names)):
91
  print(f"cosine_similarity({names[i]}, {names[j]}) = {sim:.4f}")
92
  ```
93
 
94
- ## 评测
95
 
96
- MMEB MVEB 的基准评测代码**即将发布**。在此之前,完整的实验设置与基线请参见论文。
97
 
98
- ## 引用
99
 
100
- 如果 VisME MVEB 对您有帮助,请引用:
101
 
102
  ```bibtex
103
  @inproceedings{cao2026illuminating,
@@ -109,6 +109,6 @@ MMEB 与 MVEB 的基准评测代码**即将发布**。在此之前,完整的
109
  }
110
  ```
111
 
112
- ## 许可
113
 
114
- 本模型基于 [Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct) 构建。请遵循基础模型以及训练所用各数据集的许可条款。
 
13
 
14
  # VisME (Qwen2.5-VL-7B)
15
 
16
+ English | [简体中文](README-zh.md)
17
 
18
+ **VisME** is a universal multimodal embedding model from our CVPR 2026 paper [*Illuminating Visual Identity in Universal Multimodal Embeddings*](https://openaccess.thecvf.com/content/CVPR2026/html/Cao_Illuminating_Visual_Identity_in_Universal_Multimodal_Embeddings_CVPR_2026_paper.html). Built on [Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct), VisME produces dense embeddings for images, text, and image–text pairs, with strong performance on both general multimodal benchmarks (**MMEB**) and identity-centric retrieval (**MVEB**).
19
 
20
  <p align="center">
21
  <img src="assets/teaser.jpg" alt="VisME teaser" width="95%">
22
  </p>
23
 
24
+ ## Highlights
25
 
26
+ - **Identity-aware retrieval** — trained with visual-identity supervision via the [MVEB](https://chrisclear3.github.io/MVEB/) benchmark (4 meta-tasks, 28 test subsets).
27
+ - **Strong general capability** — jointly trained on **MMEB-v1** and **MVEB**, maintaining competitive performance on standard MMEB tasks.
28
 
29
+ ## Performance
30
 
31
+ Results on **MMEB** (Cls / VQA / Ret / Grd) and **MVEB** (ID-Rec / Re-ID / ID-Grd / ID-Edit). Metrics are reported as in the paper (average Precision@1 across datasets within each task group).
32
 
33
  <p align="center">
34
  <img src="assets/performance.jpg" alt="VisME performance on MMEB and MVEB" width="95%">
35
  </p>
36
 
37
+ ## Training Data
38
 
39
+ VisME is trained on a mixture of:
40
 
41
+ | Benchmark | Role |
42
+ |-----------|------|
43
+ | [**MMEB-v1**](https://github.com/TIGER-AI-Lab/VLM2Vec) | General multimodal embedding tasks (classification, VQA, retrieval, grounding) |
44
+ | [**MVEB**](https://chrisclear3.github.io/MVEB/) | Visual-identity retrieval across recognition, re-ID, grounding, and editing |
45
 
46
+ The **MVEB** datasets are now open-sourced. Training code will be released soon.
47
 
48
+ ## Usage
49
 
50
  ```python
51
  import torch
 
79
  ]
80
 
81
  with torch.no_grad():
82
+ embeddings = model.encode_input(batch) # shape: (3, 3584), L2-normalized
83
 
84
+ # Pairwise cosine similarity
85
  # SpiderMan_comic_E616 <-> SpiderMan_promo_E199999
86
  # SpiderMan_comic_E616 <-> Toxin
87
  # SpiderMan_promo_E199999 <-> Toxin
 
91
  print(f"cosine_similarity({names[i]}, {names[j]}) = {sim:.4f}")
92
  ```
93
 
94
+ ## Evaluation
95
 
96
+ Benchmark evaluation code for MMEB and MVEB is **coming soon**. In the meantime, please refer to the paper for full experimental settings and baselines.
97
 
98
+ ## Citation
99
 
100
+ If you find VisME or MVEB useful, please cite:
101
 
102
  ```bibtex
103
  @inproceedings{cao2026illuminating,
 
109
  }
110
  ```
111
 
112
+ ## License
113
 
114
+ This model is built upon [Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct). Please follow the license terms of the base model and the respective datasets used during training.