跳到正文

AI工具Hugging Face00:52

借助 LFM2.5-DSpark 实现高达 3.2 倍的推理加速

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

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

判断

部署 LFM2.5-2.6B 做端侧 function-calling 的团队,可以按延迟降 57%、精度不变把 DSpark 默认打开;但 1.2B 必须先按自家文本分布重测,8B-A1B 先别开。

依据

LiquidAI 团队在 Hugging Face 发布了 LFM2.5 三个模型的 DSpark 草稿模型 checkpoint:1.2B-Instruct、2.6B、8B-A1B。机制是投机解码(speculative decoding):轻量草稿模型先一次前向吐出候选 token,目标模型再一口气验证全部候选,把权重的 DRAM→SRAM 搬运成本摊到多个 token 上——解码阶段本来就是 memory-bound,瓶颈在搬权重而不在算。三个草稿模型都在 300M 参数上下(295.7M 和 327.7M),显存只多这一点。质量上,贪心解码下草稿 token 必须匹配目标分布才被接受,被拒就换目标模型自己的 token,输出序列与基线逐字相同。速度:2.6B 在 H100 均值 323 → 864 tok/s(2.67x),M4 Max 61 → 139 tok/s(2.27x);1.2B 在 M4 Max 均值 138 → 350 tok/s(2.54x)。代价落在接受率上:块大小是 9,均值只接受 4.81 个,1.2B 因文本分布不同加速比能差 52%;8B-A1B 接受率更高,端侧却只快 18%,材料把原因归给 llama.cpp 的 Metal 后端 MoE 实现。

  • LFM2.5-2.6B · H100323 → 864 tok/s
  • LFM2.5-2.6B · M4 Max61 → 139 tok/s
  • LFM2.5-1.2B · M4 Max138 → 350 tok/s
  • LFM2.5-8B-A1B · 端侧18%
同一批 DSpark 草稿模型在不同硬件上的吞吐变化

没写清 材料只测了 batch size 为 1、温度 0、最多 256 输出 token 的场景,没有并发服务下的加速与显存数据。

下一步 拿自家多轮 function-calling 流量,在 batch size 大于 1 时复测 2.6B 的 57% 延迟降幅是否保持。

本期其他新闻

  1. 行业动态

    不要再做 TUI

    Simon Willison

  2. 行业动态

    引用 Matt Webb

    Simon Willison

  3. 行业动态

    从 Atari 到 EVE Online:基于 15 年游戏 AI 研究构建

    Google DeepMind

  4. AI工具

    ChatGPT 搜索现在大规模使用 site: 运算符

    Simon Willison

  5. 精选深读

    衡量语音识别中的基准优化

    Hugging Face