跳转至

推理优化

推理优化的目标是在质量可接受的前提下降低延迟和成本。不同优化手段作用在不同层:模型结构、数值精度、运行时、调度和硬件。

优化方向

精度与量化

  • FP16 / BF16:常用混合精度格式。
  • INT8 / INT4:降低显存和带宽压力。
  • AWQ / GPTQ:常见权重量化方法。

内存优化

  • KV Cache 管理。
  • Paged Attention。
  • 权重共享和加载策略。
  • CPU/GPU offload。

解码优化

  • Continuous Batching。
  • Speculative Decoding。
  • Prefix Cache。
  • Early Exit 或约束解码。

算子优化

  • FlashAttention。
  • Kernel Fusion。
  • TensorRT / ONNX Runtime / vLLM / TGI。

取舍

  • 量化可能影响质量。
  • 更大的 batch 提高吞吐,但可能增加延迟。
  • Prefix Cache 依赖请求模式。
  • Speculative Decoding 依赖小模型命中率。