arXiv:2411.00853cs.LGcs.AI2024-11被引 1

动态执行让AI推理更快更省电,按输入难易自动调整计算量。

Accelerated AI Inference via Dynamic Execution Methods

  • 根据输入复杂度动态调整计算流程,简单任务少算
  • 在不降质前提下,显著降低延迟和提升吞吐量
  • 适合追求低延迟、高能效的生成式AI部署场景

本文聚焦动态执行技术,根据输入特征优化计算流程,识别可低资源求解的简单问题,类比人类认知。涵盖深度网络提前退出、语言模型推测性采样及扩散模型自适应步数等方法。实验表明,这些动态策略可在不牺牲质量的前提下显著提升延迟与吞吐。结合量化等模型优化技术,形成多维度推理优化方案。针对生成式AI高算力需求趋势,提出更高效的采样机制与基于提示难度预测的扩散模型早停方法。相关技术已集成至Intel性能库与Hugging Face Optimum,推动落地应用。性能提升带来成本与能耗双下降,具备实际部署价值。

原文摘要 · Abstract (English)

In this paper, we focus on Dynamic Execution techniques that optimize the computation flow based on input. This aims to identify simpler problems that can be solved using fewer resources, similar to human cognition. The techniques discussed include early exit from deep networks, speculative sampling for language models, and adaptive steps for diffusion models. Experimental results demonstrate that these dynamic approaches can significantly improve latency and throughput without compromising quality. When combined with model-based optimizations, such as quantization, dynamic execution provides a powerful multi-pronged strategy to optimize AI inference. Generative AI requires a large amount of compute resources. This is expected to grow, and demand for resources in data centers through to the edge is expected to continue to increase at high rates. We take advantage of existing research and provide additional innovations for some generative optimizations. In the case of LLMs, we provide more efficient sampling methods that depend on the complexity of the data. In the case of diffusion model generation, we provide a new method that also leverages the difficulty of the input prompt to predict an optimal early stopping point. Therefore, dynamic execution methods are relevant because they add another dimension of performance optimizations. Performance is critical from a competitive point of view, but increasing capacity can result in significant power savings and cost savings. We have provided several integrations of these techniques into several Intel performance libraries and Huggingface Optimum. These integrations will make them easier to use and increase the adoption of these techniques.

AI推理动态执行生成模型能效优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。