arXiv:2601.08539cs.PFcs.LG2026-01被引 5

通过细粒度核级动态调压调频,大幅降低大模型运行能耗且几乎不降速。

Reducing Compute Waste in LLMs through Kernel-Level DVFS

  • 在计算核级别实施动态电压频率调节,比传统逐轮级更精细高效。
  • GPT-3训练中节能达14.6%,仅带来0.6%性能下降,优于逐轮级方案的2%节能。
  • 方法适用于数据并行与张量并行,具备良好可扩展性,适合大规模模型部署者。

人工智能的快速发展推动了基于加速器或GPU的数据中心扩张,但随之而来的高能耗已成为关键瓶颈和可持续性挑战。动态电压频率调节(DVFS)是一种成熟且无需改造硬件即可提升能效的技术。虽然已有研究证明通过DVFS或功耗限制可显著降低大语言模型(LLMs)训练与推理的能耗,但常伴随严重性能下降。本文聚焦于减少大模型运算中的资源浪费:即在不损失性能的前提下降低能耗。提出一种细粒度、核级别的DVFS方法,探索新的频率配置,证明其相比以往逐传或逐迭代级方案能实现更高节能效果。例如,在GPT-3训练任务中,传统逐轮级方法仅能节能2%(无性能损失),而本方法可达14.6%节能,仅造成0.6%性能下降。进一步研究了数据并行与张量并行的影响,表明所发现的时钟频率配置具有良好迁移性。结论表明,核级DVFS是减少大模型运算浪费的有效手段,可在极小性能损耗下实现显著节能。

原文摘要 · Abstract (English)

The rapid growth of AI has fueled the expansion of accelerator- or GPU-based data centers. However, the rising operational energy consumption has emerged as a critical bottleneck and a major sustainability concern. Dynamic Voltage and Frequency Scaling (DVFS) is a well-known technique used to reduce energy consumption, and thus improve energy-efficiency, since it requires little effort and works with existing hardware. Reducing the energy consumption of training and inference of Large Language Models (LLMs) through DVFS or power capping is feasible: related work has shown energy savings can be significant, but at the cost of significant slowdowns. In this work, we focus on reducing waste in LLM operations: i.e., reducing energy consumption without losing performance. We propose a fine-grained, kernel-level, DVFS approach that explores new frequency configurations, and prove these save more energy than previous, pass- or iteration-level solutions. For example, for a GPT-3 training run, a pass-level approach could reduce energy consumption by 2% (without losing performance), while our kernel-level approach saves as much as 14.6% (with a 0.6% slowdown). We further investigate the effect of data and tensor parallelism, and show our discovered clock frequencies translate well for both. We conclude that kernel-level DVFS is a suitable technique to reduce waste in LLM operations, providing significant energy savings with negligible slow-down.

大模型优化能效提升DVFS

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