5 minrss原文 ↗

Hugging Face 发布 LFM2.5 系列 DSpark 草稿模型,推理速度最高提升 3.18 倍

Hugging Face 发布 LFM2.5 系列 DSpark 草稿模型,推理速度最高提升 3.18 倍Hugging Face 发布 LFM2.5 系列三款模型的 DSpark 草稿模型检查点,通过投机解码在不改变输出质量的前提下,GPU 吞吐最高提升 3.18 倍,端侧最高 2.87 倍。草稿模型约 300M 参数,LFM2.5-2.6B 函数调用延迟平均降低 57%,已开源支持 llama.cpp 和 SGLang。推荐理由约300M参数的draft模型换最高3.18倍吞吐,贪心输出不变,LFM2.5-2.6B在M4 Max达139 tok/s,本地agent门槛被拉低。本文目录DSpark 是如何工作的训练与架构质量对齐CPU 和 GPU 上的推理加速如何使用 LFM2.5-DSpark标签Hugging Face 发布 LFM2.5 系列 DSpark 草稿模型,推理速度最高提升 3.18 倍Hugging Face 发布 LFM2.5 系列三款模型的 DSpark 草稿模型检查点,通过投机解码在不改变输出质量的前提下,GPU 吞吐最高提升 3.18 倍,端侧最高 2.87 倍。草稿模型约 300M 参数,LFM2.5-2.6B 函数调用延迟平均降低 57%,已开源支持 llama.cpp 和 SGLang。约300M参数的draft模型换最高3.18倍吞吐,贪心输出不变,LFM2.5-2.6B在M4 Max达139 tok/s,本地agent门槛被拉低。今天,我们发布了针对 LFM2.5 系列中三款模型的 DSpark 草稿模型检查点:LFM2.5-1.2B-Instruct、LFM2.5-2.6B 和 LFM2.5-8B-A1B。这些模型增加了一条投机解码路径,以极小的内存开销换取大幅度的解码速度提升,且不改变输出质量:更快的推理:在 GPU 上吞吐量最高提升 3.18 倍,在端侧设备上最高提升 2.87 倍。迈向端侧智能体推理:LFM2.5-2.6B 的函数调用延迟平均降低 57%。首日支持 llama.cpp 和 SGLang:兼容 LFM 的 DSpark 集成已在上游开源。DSpark 是如何工作的大语言模型推理中的解码阶段传统上受内存带宽限制。大部分延迟来自将权重从 DRAM 流式加载到 SRAM,而非密集计算。投机解码通过使用轻量级草稿模型生成候选 token,然后由目标模型在单次前向传播中一次性验证所有候选 token 来解决这一问题,从而将加载权重的成本分摊到我们验证的所有 token 上。多年来,业界提出了多种投机方法,其中最著名的是 EAGLE-3、DFlash,以及最新的 DSpark,后者结合了三个组件:基于 DFlash 风格的并行主干网络,以目标模型的上下文特征为条件,在单次前向传播中为所有草稿 token 生成隐藏状态。一个轻量级的顺序头,以相邻 token 之间的马尔可夫链为模型,增加了 token 间的依赖关系,从而提高了后续位置的接受率。一个基于置信度调度的验证器,用于预测每个 token 的存活概率,并在验证成本高于其节省成本时,剪除低置信度的后缀。训练与架构我们遵循 DSpark 的方法,但使用了更大、更多样化的数据混合,涵盖 SFT、聊天、代码和函数调用数据。根据我们的消融实验,草稿模型的初始版本是简化的仅注意力(attention-only)草稿模型,包含 5 层和 9 个块。对于每个草稿模型,我们在整个数据集上运行了 15 个 epoch,并选择了接受率最高而非损失最低的那个 epoch。由此产生的草稿模型相对较小,每个模型约有 ~3 亿参数。质量对齐在贪心解码下,草稿 token 只有在与目标模型的分布匹配时才会被接受。被拒绝时,目标模型自身的 token 会取而代之。因此,生成的序列在构造上与基线贪心解码完全一致,所以基准测试的准确率(pass@1 或精确匹配)保持不变。CPU 和 GPU 上的推理加速我们为 LFM2.5 推出的 DSpark 草稿模型在发布首日即支持 llama.cpp(实现基于官方代码库构建,我们使用实验性的 Metal 内核运行)和 SGLang(实现基于官方 SGLang 对 DSpark 的实现构建)。我们使用 llama.cpp 和 Metal 在 M4 Max MacBook Pro 上测量端侧吞吐量,采用 FP16 GGUF 权重,最多生成 256 个输出 token。我们使用 SGLang 在单块 H100 80 GB 上以 BF16 测量 GPU 吞吐量。两种配置均使用 DSpark 块大小 9、批大小 1 和温度 0。我们在五个基准数据集上进行了评估。三个草稿模型在大型加速器(H100)和边缘部署(M4 Max MacBook)上都带来了显著的吞吐量提升。对于 LFM2.5-2.6B,MacBook 上的加速尤为明显,它将用户可享受的交互性水平推向了远超大多数专有云模型(约 ~140 tok/s,视数据集而定)的吞吐量。在各种多工具场景中,DSpark 为 LFM2.5-2.6B 平均降低了 57% 的延迟。对于 LFM2.5-1.2B-Instruct,我们看到数据集接受率的波动要大得多,因此加速效果会因底层文本分布的不同而出现高达 52% 的差异。对于 LFM2.5-8B-A1B,其接受率相比两个稠密模型有所提升,但在端侧设备上我们平均仅获得 18% 的改进。这一差距源于 llama.cpp 的 Metal 后端中当前 MoE 实现的限制,以及验证 k 个 token 会比单次解码步骤激活更多专家,从而产生更大的权重流量。如何使用 LFM2.5-DSpark使用 SGLang 运行 DSpark 草稿模型,需要构建支持 LFM2 目标 DSpark 功能的 SGLang 版本(PR #31041)。启动目标模型并附加草稿模型:然后查询位于 http://localhost:30000/v1 的 OpenAI 兼容端点。块大小从草稿模型的 config.json 中读取;基线是去掉三个 --speculative-* 标志后的相同命令。使用 llama.cpp 运行它们需要相应的 llama.cpp 构建版本(PR#27383)。块大小从 sidecar 元数据中读取(n-max 会被限制在该值内)。投机解码是精确的:目标模型会验证每一个提议的 token,因此贪婪解码的输出与单独运行目标模型完全一致;每次响应的计时报告会显示 draft_n / draft_n_accepted。DSpark 草稿模型检查点已在 Hugging Face 上以 Safetensors 和 GGUF 格式提供:Safetensors:LFM2.5-2.6B-DSpark、LFM2.5-1.2B-Instruct-DSpark 和 LFM2.5-8B-A1B-DSparkGGUF:LFM2.5-2.6B-DSpark-GGUF、LFM2.5-1.2B-Instruct-DSpark-GGUF、LFM2.5-8B-A1B-DSpark-GGUF我们迫不及待想看到你的成果。引用如需引用,请使用以下参考文献或 BibTeX:Liquid AI 发布“LFM2.5-DSpark:从 H100 到 MacBook,推理速度最高提升 3.2 倍”,Liquid AI 博客,2026 年 8 月。来源:Hugging Face:Blog(RSS)· huggingface.co

如果可以重来,你还愿意花这段时间读它吗?

· 匿名阅读记录只用于改进推荐