跳到正文

AI工具Hugging Face00:52

LFM2.5-DSpark 推理速度提升高达 3.2 倍

Up to 3.2x Faster Inference with LFM2.5-DSpark

速度提升3.2倍,但需验证是否以精度或资源为代价。

判断

端侧或单卡部署 LFM2.5 的人,现在要按内存预算而非精度来决定挂不挂 DSpark:约 300M 参数的 draft 模型换 2.27x 到 2.67x 吞吐,代价是一小块常驻显存,而 8B-A1B 端侧只多 18%。

依据

LiquidAI 的 Leonie Monigatti、Fernando Fernandes Neto 等作者在 Hugging Face 发文,为 LFM2.5-1.2B-Instruct、LFM2.5-2.6B、LFM2.5-8B-A1B 发布 DSpark draft model checkpoint。机制上,LLM 推理的 decode 阶段受显存带宽限制(memory-bound),延迟主要来自把权重从 DRAM 搬进 SRAM,而不是算力;speculative decoding(投机解码)先用轻量 draft 模型猜若干 token,再由 target 模型一次前向全部验证,把同一份权重加载成本摊到多个 token 上。DSpark 由三部分组成:DFlash 式并行骨干、一个 Markov 链形式的轻量顺序头(补相邻 token 依赖,抬高后位接受率)、以及按置信度剪掉低置信后缀的 verifier。draft 模型各约 300M 参数、5 层、block size 9,训练时按接受率而非 loss 挑 epoch。作者给出的对照:H100 上平均 2.67x(323 → 864 tok/s),M4 Max 上平均 2.27x(61 → 139 tok/s),2.6B 在多工具场景延迟平均降 57%;但 8B-A1B 端侧平均只提升 18%,作者把它归因于 llama.cpp 的 Metal 后端 MoE 实现,以及验证 k 个 token 会激活更多专家。

  • H100 平均吞吐(5 个数据集)323 → 864 tok/s
  • M4 Max 平均吞吐61 → 139 tok/s
  • 2.6B 多工具场景延迟降 57%
  • 8B-A1B 端侧平均提升18%
DSpark 在三款 LFM2.5 上的吞吐与延迟变化

没写清 材料只写「a minimal memory increase」,没给 draft 模型的显存增量绝对值,所以这块钱到底多大无法替它核算。

下一步 等 llama.cpp 的 Metal 后端更新 MoE 实现后,核对 LFM2.5-8B-A1B 端侧平均提升是否仍停在 18%。

本期其他新闻

  1. 行业动态

    AI 期货问世

    OpenAI

  2. 行业动态

    Stampli 使用 ChatGPT Work 将启动时间缩短 68%

    OpenAI

  3. AI工具

    ChatGPT 搜索现大规模支持 site: 运算符

    Simon Willison

  4. AI工具

    基于 Bun 1.4 新 Bun.WebView 的 shot-scraper 风格 JSON API

    Simon Willison

  5. AI工具

    smolmachines / smolvm 作为不受信任 Python 与 JavaScript 的沙箱

    Simon Willison