SmolVLM - Hugging Face推出的轻量级视觉语言模型
发布于 2024-11-27 · SmolVLM是Hugging Face推出的轻量级视觉语言模型,专为设备端推理设计。以20亿参数量,实现了高效内存占用和快速处理速度。SmolVLM提供了三个版本以满足不同需求:SmolVLM-Base:适用于下游任务的微调。SmolVLM-Synthetic:基于合成数据进行微调。SmolVLM-Instruct:指令微调版本,可直接应用于交互式应用中。
SmolVLM是什么
SmolVLM 是 Hugging Face 推出的轻量级视觉语言模型,专为设备端推理设计,以 20 亿参数量实现了高效的内存占用和快速的处理速度。它提供三个版本:SmolVLM-Base 适合下游任务的微调;SmolVLM-Synthetic 基于合成数据微调;SmolVLM-Instruct 是指令微调版本,可直接用于交互式应用。模型借鉴 Idefics3 理念,采用 SmolLM2 1.7B 作为语言主干,通过像素混洗技术提升视觉信息的压缩效率,在 Cauldron 和 Docmatix 数据集上训练,图像编码和文本处理能力都做了优化。
SmolVLM的主要功能
- 设备端推理:专为设备端推理设计,在笔记本电脑、消费级 GPU、移动设备等资源有限的环境下也能有效运行。
- 微调能力:模型提供三个版本以满足不同需求:
- SmolVLM-Base用于下游任务的微调;
- SmolVLM-Synthetic基于合成数据进行微调;
- SmolVLM-Instruct指令微调版本,可直接应用于交互式应用中。
- 优化的架构设计:借鉴 Idefics3 的理念,用 SmolLM2 1.7B 作语言主干,以像素混洗策略提高视觉信息的压缩率,视觉处理更高效。
- 处理长文本和多张图像:训练数据包括 Cauldron 和 Docmatix,并对 SmolLM2 做了上下文扩展,能处理更长的文本序列和多张图像。
- 内存占用低:把 384×384 像素的图像块编码为 81 个 tokens,相比之下 Qwen2-VL 需要 1.6 万个 tokens,内存占用低了不少。
- 高吞吐量:在多个基准测试里,SmolVLM 的预填充吞吐量比 Qwen2-VL 快 3.3 到 4.5 倍,生成吞吐量快 7.5 到 16 倍。
- 开源模型:SmolVLM 完全开源,模型检查点、VLM 数据集、训练配方和工具都在 Apache 2.0 许可证下发布。
- 训练数据集:训练覆盖 Cauldron 和 Docmatix 数据集,同时扩展了 SmolLM2 的上下文,可以处理更长的文本序列和多张图像。
SmolVLM的项目地址
- Github仓库:https://github.com/huggingface/blog/blob/main/smolvlm.md
- HuggingFace模型库:https://huggingface.co/blog/smolvlm
- 在线体验Demo:https://huggingface.co/spaces/HuggingFaceTB/SmolVLM
- 数据集完整列表:https://huggingface.co/HuggingFaceTB/SmolVLM-Instruct/blob/main/smolvlm-data.pdf
SmolVLM的应用场景
- 视频分析:在计算资源受限的情况下,SmolVLM 已经显示出做基础视频分析任务的潜力。CinePile 基准测试中它拿到 27.14% 的得分,视频理解能力有竞争力。
- 视觉语言处理:给开发者和研究者提供了一个趁手的视觉语言处理工具,不用投入高昂的硬件费用。
- 本地部署:小模型可以部署在浏览器或边缘设备上,本地推理成本更低,也支持用户自定义。
- AI普及化:SmolVLM 有望扩大视觉语言模型的使用范围,让复杂的 AI 系统更普及、更易触达,把机器学习能力带给更广泛的受众。






0 条评论
暂无评论,来抢沙发。