Add metadata and improve model card

README.md

@@ -1,19 +1,21 @@
 ---
+base_model:
+- Qwen/Qwen2-VL-2B-Instruct
 datasets:
 - VLM2Vec/MMEB-V2
 language:
 - en
-base_model:
-- Qwen/Qwen2-VL-2B-Instruct
+library_name: transformers
+pipeline_tag: feature-extraction
 ---
+
 # PLUME-Qwen2-VL-2B
 
 **PLUME: Latent Reasoning Based Universal Multimodal Embedding**
 
-PLUME is a latent reasoning framework for universal multimodal embedding (UME). It replaces explicit chain-of-thought (CoT) generation with a short autoregressive rollout of continuous latent states, achieving stronger retrieval
-performance while delivering **over 30x faster inference** compared to explicit-CoT methods.
+PLUME is a latent reasoning framework for universal multimodal embedding (UME). It replaces explicit chain-of-thought (CoT) generation with a short autoregressive rollout of continuous latent states, achieving stronger retrieval performance while delivering **over 30x faster inference** compared to explicit-CoT methods.
 
-**[Project Page](https://haoxiangzhao12138.github.io/PLUME/)** | **[Paper](https://arxiv.org/abs/2507.00001)** | **[Code](https://github.com/haoxiangzhao12138/PLUME)**
+**[Project Page](https://haoxiangzhao12138.github.io/PLUME/)** | **[Paper](https://arxiv.org/abs/2604.02073)** | **[Code](https://github.com/haoxiangzhao12138/PLUME)**
 
 ## Highlights
 
@@ -43,5 +45,18 @@ huggingface-cli download CUDAOUTOFMEMORY/PLUME-Qwen2-VL-2B --local-dir /path/to/
 # Option 2: git clone (requires git-lfs)
 git lfs install
 git clone https://huggingface.co/CUDAOUTOFMEMORY/PLUME-Qwen2-VL-2B
-
+```
+
+## Citation
+
+```bibtex
+@misc{he2026plumelatentreasoningbased,
+      title={PLUME: Latent Reasoning Based Universal Multimodal Embedding},
+      author={Chenwei He and Xiangzhao Hao and Tianyu Yang and Yuxiang Ma and Yuheng Jia and Lingxiang Wu and Chaoyang Zhao and Haiyun Guo and Jinqiao Wang},
+      year={2026},
+      eprint={2604.02073},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2604.02073},
+}
 ```