arXiv:2512.03024cs.LGcs.AI2025-12AAAI被引 31

首个专测大模型推理耗电的开源基准,帮开发者省电降本。

TokenPowerBench: Benchmarking the Power Consumption of LLM Inference

  • 用配置化接口+无硬件依赖的测电层,轻松测出每步耗电。
  • 实测从10亿到4050亿参数模型,发现量化和批处理显著省电。
  • 适合部署大模型的团队,尤其关注能效与碳足迹者。

大型语言模型每日响应数十亿次请求,行业报告显示,推理阶段能耗占总能耗超90%。然而现有基准多聚焦训练或性能,缺乏对推理能耗的系统性测量支持。本文提出TokenPowerBench,首个轻量级、可扩展的推理能耗评测基准。该基准包含:(i) 支持模型、提示集与推理引擎配置的声明式接口;(ii) 无需专用电表即可捕获GPU、节点及系统级功耗的测量层;(iii) 将能量精确归因于每请求预填充与解码阶段的阶段对齐指标管道。用户可快速评估批量大小、上下文长度、并行策略与量化对每令牌焦耳数等能效指标的影响。我们在四种主流模型系列(Llama、Falcon、Qwen、Mistral)上进行测试,覆盖10亿至4050亿参数规模,包括前沿的Llama3-405B模型。项目已开源,助力用户测算能耗、预测运营成本并达成可持续目标。

原文摘要 · Abstract (English)

Large language model (LLM) services now answer billions of queries per day, and industry reports show that inference, not training, accounts for more than 90% of total power consumption. However, existing benchmarks focus on either training/fine-tuning or performance of inference and provide little support for power consumption measurement and analysis of inference. We introduce TokenPowerBench, the first lightweight and extensible benchmark designed for LLM-inference power consumption studies. The benchmark combines (i) a declarative configuration interface covering model choice, prompt set, and inference engine, (ii) a measurement layer that captures GPU-, node-, and system-level power without specialized power meters, and (iii) a phase-aligned metrics pipeline that attributes energy to the prefill and decode stages of every request. These elements make it straight-forward to explore the power consumed by an LLM inference run; furthermore, by varying batch size, context length, parallelism strategy and quantization, users can quickly assess how each setting affects joules per token and other energy-efficiency metrics. We evaluate TokenPowerBench on four of the most widely used model series (Llama, Falcon, Qwen, and Mistral). Our experiments cover from 1 billion parameters up to the frontier-scale Llama3-405B model. Furthermore, we release TokenPowerBench as open source to help users to measure power consumption, forecast operating expenses, and meet sustainability targets when deploying LLM services.

大模型能耗评测能效优化开源工具

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