Qwen3.6-35B-A3B-DFlash性能基准测试:从Math500到HumanEval的全面分析
【免费下载链接】Qwen3.6-35B-A3B-DFlash项目地址: https://ai.gitcode.com/hf_mirrors/z-lab/Qwen3.6-35B-A3B-DFlash
Qwen3.6-35B-A3B-DFlash是一款基于块扩散技术的高效推理加速模型,作为Qwen3.6-35B-A3B的配套 draft 模型,通过 DFlash 推测解码技术实现多token并行生成,在保持模型输出质量的同时显著提升推理速度。该模型特别适用于需要高吞吐量的文本生成场景,从数学推理到代码生成任务均表现出色。
模型性能核心优势 🚀
DFlash技术的核心创新在于采用轻量级块扩散模型并行生成多个token,其性能提升主要体现在以下方面:
- 架构优化:通过 dflash.py 实现的 Qwen3DFlashAttention 机制,结合 config.json 中配置的 target_layer_ids 目标层选择策略,实现高效的注意力计算
- 推理加速:在单NVIDIA B200环境下,对比传统自回归解码实现最高2.9倍的速度提升
- 资源效率:仅8层的 decoder 结构(config.json中num_hidden_layers=8),以较小的计算开销实现显著加速效果
测试环境与基准配置 ⚙️
所有性能测试均在统一环境下完成,确保结果的可比性:
- 硬件配置:单NVIDIA B200 GPU
- 软件栈:SGLang框架(启用thinking模式)
- 测试参数:最大输出长度4096 tokens,默认块大小(block size)16
- 评估指标:端到端吞吐量(Tokens/sec),包含prefill阶段时间
关键性能指标解析 📊
吞吐量与加速比
DFlash在不同任务和并发场景下均展现出稳定的加速效果。以下是块大小为16时的核心测试结果:
数学推理任务(Math500):
- 并发1时:从234 tokens/sec提升至682 tokens/sec,实现2.9倍加速
- 并发32时:从2755 tokens/sec提升至6520 tokens/sec,仍保持2.4倍加速
代码生成任务(HumanEval):
- 并发1时:从238 tokens/sec提升至603 tokens/sec,实现2.5倍加速
- 并发32时:从2767 tokens/sec提升至5782 tokens/sec,保持2.1倍加速
块大小对性能的影响
测试对比了块大小为8和16时的性能表现,发现:
- 小任务(Alpaca):块大小16比8仅提升1.7倍 vs 1.7倍,差异不显著
- 复杂任务(Math500):块大小16实现2.9倍加速,优于块大小8的2.6倍
- 最佳实践:对于代码和数学类复杂任务优先使用块大小16,通用对话任务可使用块大小8平衡速度与质量
接受长度分析
接受长度(Acceptance Length)反映了draft模型生成内容被目标模型接受的平均token数,是衡量加速效率的关键指标:
| 任务 | 块大小8 | 块大小16 |
|---|---|---|
| Math500 | 5.56 | 7.35 |
| HumanEval | 5.09 | 6.44 |
| GSM8K | 5.21 | 6.73 |
数据表明,块大小16在所有任务上均获得更高的接受长度,验证了更大块尺寸的优势
实际部署指南 📋
快速安装
# 安装vLLM(支持DFlash的修改版本) uv pip install vllm uv pip install -U --torch-backend=auto "vllm @ git+https://github.com/vllm-project/vllm.git@refs/pull/40898/head" # 或安装SGLang uv pip install "git+https://github.com/sgl-project/sglang.git@refs/pull/20547/head#subdirectory=python"启动服务示例
# vLLM启动命令 vllm serve Qwen/Qwen3.6-35B-A3B \ --speculative-config '{"method": "dflash", "model": "z-lab/Qwen3.6-35B-A3B-DFlash", "num_speculative_tokens": 15}' \ --attention-backend flash_attn \ --max-num-batched-tokens 32768最佳配置建议
- 长上下文场景:添加
--speculative-dflash-draft-window-size参数启用滑动窗口注意力 - 高并发服务:设置
--max-num-batched-tokens 32768优化批处理效率 - 代码生成任务:推荐块大小16,可获得最佳加速比
总结与展望
Qwen3.6-35B-A3B-DFlash通过创新的块扩散技术,在保持模型质量的前提下实现了显著的推理加速。测试结果表明,该模型在数学推理(Math500)和代码生成(HumanEval)任务上表现尤为突出,加速比分别达到2.9倍和2.5倍,为高吞吐量文本生成应用提供了高效解决方案。
随着硬件支持的不断优化和算法的持续改进,DFlash技术有望在更多场景下实现更高的性能提升。建议开发者根据具体任务类型选择合适的块大小配置,以获得最佳的速度与质量平衡。
注:所有测试数据基于SGLang框架,使用默认参数配置。实际性能可能因硬件环境和应用场景有所差异。完整测试脚本可参考项目GitHub仓库。
【免费下载链接】Qwen3.6-35B-A3B-DFlash项目地址: https://ai.gitcode.com/hf_mirrors/z-lab/Qwen3.6-35B-A3B-DFlash
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考