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的项目地址

SmolVLM的应用场景

  • 视频分析:在计算资源受限的情况下,SmolVLM 已经显示出做基础视频分析任务的潜力。CinePile 基准测试中它拿到 27.14% 的得分,视频理解能力有竞争力。
  • 视觉语言处理:给开发者和研究者提供了一个趁手的视觉语言处理工具,不用投入高昂的硬件费用。
  • 本地部署:小模型可以部署在浏览器或边缘设备上,本地推理成本更低,也支持用户自定义。
  • AI普及化:SmolVLM 有望扩大视觉语言模型的使用范围,让复杂的 AI 系统更普及、更易触达,把机器学习能力带给更广泛的受众。
0

相关文章

0 条评论

后参与评论

暂无评论,来抢沙发。