AI 模型会变快?“NF4 反量化”技术的秘密

抽象图形,描绘了正在高速处理数据的 GPU 处理器
AI Summary

介绍一种全新的 Triton 内核技术,通过改善 NF4 量化过程中的内存访问方式,大幅提升了 AI 模型推理速度。

想象一下,如果您常用的 AI 聊天机器人能比平时更快、更顺畅地输出回复会怎样?如果我们输入问题后等待回复的时间缩短,对话将会变得更加自然。最近,AI 技术领域在“提升速度”方面取得了一项有趣的突破。有一种旨在减轻庞大 AI 模型负担的技术称为“量化”,而现在,处理这种量化过程的新方法诞生了。

这为何重要?

我们日常接触的大型语言模型(LLM)拥有超过数十亿个参数(模型内部用于记忆信息的数值)。为了在普通用户的电脑或智能手机上运行这些模型,必须大幅减小模型体积,此时“量化(Quantization,一种通过降低模型数值精度来减少数据大小的技术)”就显得至关重要。其中,4位量化技术 NF4 (NormalFloat 4-bit) 是一项非常聪明的技术,它能在大幅减小模型体积的同时,将性能损失降至最低。

但这里存在一个问题。即便模型变小了,如果将压缩数据还原为可运算状态的“反量化(Dequantization,解压过程)”速度缓慢,整个 AI 的性能仍会遭遇瓶颈。这就像一辆装满货物的大卡车,如果卸货速度很慢,配送就会延迟。此次开发的优化技术正是为了让这一过程变得极其迅速,从而为缩短 AI 输出回复的总时间做出了巨大贡献。Source 11

浅显易懂的解释:有地图就不会迷路

AI 模型的参数就像图书馆里摆满的书籍。我们将这些书压缩后存放在仓库(内存)中,而 AI 若要生成回复,就必须解压这些书并读取内容。

AD

传统的“bitsandbytes(执行量化运算的代表性库)”方式在查找书中的内容时,需要逐一确认复杂的“岔路口(分支运算)”来解压。这就好比为了在图书馆找书,在迷宫般的道路中一边摸索一边确认每一条路径。

相反,此次发布的基于 Triton(一种在 PyTorch 环境下协助低级 GPU 优化的编程工具)的全新内核技术采取了完全不同的方案。Source 10

这种方式将“查找表(记录数据位置的地图)”预先复制并铺开在访问速度最快的“共享内存”中。Source 3 这样一来,就像手中握着图书馆地图一样,无需在迷宫中摸索,即可直接找到所需数据。由于共享内存比普通的全局内存快 12 到 15 倍,整体数据处理速度得到了飞跃性的提升。Source 3

现状

这项新技术的成效已得到数据证实。研究人员在 Gemma 27B、Qwen3 32B 和 Llama3.3 70B 等大型语言模型上应用了该技术。结果显示,作为运算核心的内核速度最高提升了 2.2 倍,而用户实际感受到的系统整体性能(端到端)也提升了最高 1.54 倍。Source 3 特别是它表现出比现有的标准方案“bitsandbytes”还要快 1.41 倍的速度,受到了技术界的广泛关注。Source 1, Source 8

目前,该技术已开源,正由众多开发者进行验证,无论硬件设备或模型大小如何,都能展现出一致的速度提升效果。Source 12

未来展望

这一成果再次证明了高效处理硬件资源的软件优化对于 AI 普及的重要性。预计未来会有更多的 AI 库和服务采用这种基于 Triton 的优化内核。如果我们的 AI 服务能变得更加轻快,即便没有昂贵的设备,在个人设备上直接运行比现在更智能的 AI 的时代也将更快到来。

MindTickleBytes AI 记者观察

此次案例不仅仅是提升了速度,更证明了如何高效处理硬件资源能够改变 AI 的未来。在一个 AI 模型日益庞大沉重的时代,通过“更精密的软件”而非寻找“更好的硬件”来突破极限,这种方式令人印象深刻。性能,终究要在软件的精妙中完成蜕变。

参考资料

  1. Show HN: Fast NF4 dequantization Triton kernel (1.41x faster than bitsandbytes) - https://github.com/Griffith-7/nf4-triton-kernel
  2. Accelerating NF4 Double-Dequantization within a Single Triton Kernel - https://medium.com/@samdj0245/accelerating-nf4-double-dequantization-within-a-single-triton-kernel-f26a0f35b372
  3. Fast NF4 Dequantization Kernels for Large Language Model Inference (arXiv:2604.02556) - https://arxiv.org/abs/2604.02556
  4. NF4 Dequantization Speedup in Triton with Fused Kernel (LinkedIn) - https://www.linkedin.com/posts/jash-dalvi_hi-everyone-i-wrote-a-deep-dive-on-nf4-dequantization-activity-7441491596753203200-Sxzh
  5. Fast NF4 Dequantization Kernels for Large Language Model Inference (PDF) - https://arxiv.org/pdf/2604.02556
  6. Paper page - Fast NF4 Dequantization Kernels for Large Language Model Inference (HuggingFace) - https://huggingface.co/papers/2604.02556
  7. GitHub - Griffith-7/bitnet - https://github.com/Griffith-7/bitnet/blob/main/
  8. ShowHN: Fast NF4 dequantization Triton kernel (1.41x faster than bitsandbytes) (ModernOrange) - https://modernorange.io/item/48920706
  9. VueHN2.0 ShowHN: Fast NF4 dequantization Triton kernel - https://vue-hackernews-ssr-5cavbdjcta-ew.a.run.app/item/48920706
  10. Accelerating Triton Dequantization Kernels for GPTQ – PyTorch - https://pytorch.org/blog/accelerating-triton/
  11. Fast NF4 Dequantization Kernels for Large Language Model Inference (HTML) - https://arxiv.org/html/2604.02556
  12. Fast NF4 Dequantization Kernels for Large Language Model Inference (HyperAI) - https://hyper.ai/en/papers/2604.02556
  13. Fast NF4 Dequantization Kernels for Large Language Model Inference (HTML v1) - https://arxiv.org/html/2604.02556v1
  14. From Zero to 1.55x: Writing My First Triton Kernel for NF4 - https://numb3r33.github.io/experiments/deeplearning/unsloth/triton/gpu/2025/12/21/unsloth-triton-kernels.html
AD
测试你的理解
Q1. 新的 Triton 内核技术比传统方式更快的主要原因是什么?
  • 增加了 GPU 数量
  • 将查找表置于共享内存并直接访问
  • 将 AI 模型压缩得更小
通过直接访问共享内存中的查找表(lookup table)来替代复杂的分支(branching)运算,从而提高了处理速度。
Q2. 应用该技术后,性能提升幅度约为多少?
  • 内核速度最高提升 1.41 倍
  • 准确率提升 99%
  • 功耗降低 50%
相较于 bitsandbytes 的实现,内核速度展现出最高 1.41 倍的提升。
Q3. Triton 是一种什么样的工具?
  • 用于训练 AI 模型的语言
  • 协助简化 GPU 优化的工具
  • 对数据进行加密的程序
Triton 能够在比传统 CUDA 编程更高的抽象级别上实现低级 GPU 优化。