🚀 FlashVL-2B-Dynamic-ISS
FlashVL是一种优化视觉语言模型(VLMs)以用于实时应用的新方法,旨在实现超低延迟和高吞吐量,同时不牺牲准确性。它通过先进的架构增强和高效的计算策略,在多个视觉语言基准测试中保持竞争力的同时,减少处理时间以最大化吞吐量。
🚀 快速开始
环境搭建
pip install torch==2.1.2
pip install transformers==4.50.0.dev0
使用方法
import torch
from PIL import Image
import requests
from io import BytesIO
from transformers import AutoModel, AutoTokenizer, CLIPImageProcessor
model_path = "FlashVL/FlashVL-2B-Dynamic-ISS"
model = AutoModel.from_pretrained(model_path, torch_dtype=torch.bfloat16,trust_remote_code=True,device_map='cuda')
model.tokenizer = AutoTokenizer.from_pretrained(model_path,device_map='cuda')
model.im_trans = CLIPImageProcessor.from_pretrained(model_path)
image_url ="https://s3plus.meituan.net/automl-datasets/mlm/0516.png"
response = requests.get(image_url)
image_data = BytesIO(response.content)
pil_image = Image.open(image_data).convert('RGB')
messages = [{'role': 'user', 'content': "生成图中菜品的菜谱"}]
answer = model.chat(pil_image, messages, do_sample=False, max_new_tokens=256)
print(answer)
messages = [
{'role': 'user', 'content': '这是什么'},
{"role": "assistant", "content": '这是一道看起来像是银耳莲子汤的甜品。\
银耳是一种常见的食材,通常用于制作甜品和汤品,具有软糯的口感和清润的口感。莲 \
子是莲子的干燥部分,常用于中医和食疗中,具有补脾止泻的功效。图片中还可以看到 \
一些枸杞和核桃,枸杞富含维生素和抗氧化物质,核桃则提供丰富的蛋白质和健康脂肪。 \
整体来看,这道甜品不仅美味,还具有一定的营养价值。'},
{'role': 'user', 'content': '对图中菜品卡路里分析'}
]
answer = model.chat(pil_image, messages, do_sample=False, max_new_tokens=256)
print(answer)
messages = [{'role': 'user', 'content': "who are you"}]
answer = model.chat(None, messages, do_sample=False, max_new_tokens=256)
print(answer)
✨ 主要特性
我们很高兴推出 FlashVL,这是一种针对实时应用优化视觉语言模型(VLMs)的新方法,目标是在不牺牲准确性的前提下实现超低延迟和高吞吐量。Flash-VL 2B 利用先进的架构增强和高效的计算策略,通过减少处理时间来最大化吞吐量,同时在多个视觉语言基准测试中保持有竞争力的性能。我们的方法包括量身定制的架构选择、令牌压缩机制、数据整理、训练方案,以及一种名为隐式语义拼接的新型图像处理技术,该技术能有效平衡计算负载和模型性能。通过在 11 个标准 VLM 基准测试上的广泛评估,我们证明了 Flash-VL 2B 在速度和准确性方面都取得了最先进的成果,使其成为在资源受限环境和大规模实时应用中部署的有前景的解决方案。
📚 详细文档
评估
基准测试 |
Qwen2-VL-2B |
Aquila-VL-2B |
InternVL2.5-2B |
Flash-VL-2Bs |
Flash-VL-2Bd |
Flash-VL-2Bd-ISS |
MMMUval |
41.9 |
44.4 |
41.8 |
43.6 |
42.9 |
42.9 |
MMBenchen |
74.9 |
78.6 |
74.7 |
78.4 |
78.4 |
79.1 |
MMBenchcn |
73.5 |
76.3 |
71.6 |
74.7 |
74.9 |
76.7 |
MMStar |
48.0 |
54.9 |
54.1 |
53.8 |
54.4 |
54.1 |
MathVistatestmini |
43.0 |
59.4 |
50.9 |
59.3 |
58.1 |
61.5 |
AI2Dtest |
74.1 |
75.0 |
75.1 |
74.2 |
74.1 |
74.4 |
MMVet |
49.5 |
40.9 |
61.7 |
47.3 |
52.7 |
50.7 |
HallusionBench |
39.2 |
38.5 |
42.7 |
43.5 |
45.5 |
49.0 |
OCRBench |
794 |
773 |
800 |
764 |
831 |
843 |
MME |
1872 |
1813 |
2091 |
1715 |
1866 |
1850 |
SEEDBench |
71.5 |
78.9 |
73.2 |
73.6 |
73.6 |
74.5 |
平均值 |
60.2 |
62.6 |
63.6 |
62.4 |
64.0 |
64.8 |
我们使用 VLMEvalKit 来评估 FlashVL-2B-Static。
📄 许可证
本项目采用 Apache-2.0 许可证。
模型信息
属性 |
详情 |
模型类型 |
图像文本到文本 |
训练数据集 |
lmms-lab/LLaVA-OneVision-Data、BAAI/Infinity-MM |
基础模型 |
apple/aimv2-huge-patch14-448、Qwen/Qwen2-1.5B-Instruct |
库名称 |
transformers |
支持语言 |
英文、中文 |
引用
如果您在研究中发现本项目有用,请考虑引用:
@misc{zhang2025flashvl2boptimizingvisionlanguage,
title={Flash-VL 2B: Optimizing Vision-Language Model Performance for Ultra-Low Latency and High Throughput},
author={Bo Zhang and Shuo Li and Runhe Tian and Yang Yang and Jixin Tang and Jinhao Zhou and Lin Ma},
year={2025},
eprint={2505.09498},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2505.09498},
}

[📜 FlashVL]