Back to AI News
arxiv

SIMPLE Law: A Closed-Form Analytical Model for LLM Inference Energy Cost

This paper introduces SIMPLE, a four-parameter analytical law that predicts LLM inference energy from architectural hyperparameters without per-model calibration. It separates prefill from decode and KV-cache costs, achieving R²=0.969 and 11.97% MAPE across 13 dense LLMs on H100 GPUs, outperforming SOTA by 3.1-3.5x. The law reveals that prefill energy grows quadratically with hidden dimension, GQA reduces per-token energy up to 3x, and reasoning budgets cause super-linear energy growth with an optimal Intelligence per Joule. Agentic serving raises energy per request 13x, and prefix caching only mitigates prefill cost.

AI Brief

Provides a novel, accurate analytical model for LLM energy costs, offering actionable insights for optimizing inference efficiency and cost.

arXiv:2611.03330v1 Announce Type: new Abstract: Generative AI is billed per token, yet tokens are not equally expensive. Each token enters the KV-cache and raises the cost of those that follow, so energy grows non-linearly along a sequence whose tokens look identical. No analytical model states what a token costs, and the most accurate SOTA model is a fit, not a law: it absorbs architecture into fitted coefficients, requiring per-LLM fitting. We derive closed-form expressions of the computational and memory-access complexity of LLM inference in terms of architectural hyperparameters, covering MHA and GQA, and compare a family of analytical models of increasing granularity. From these, we obtain the SIMPLE Inference Model to Predict LLMs Energy (SIMPLE), a four-parameter law that separates prefill from decode, and decode into KV-cache cost and next-token computation. Over traces of 13 dense LLMs from six families on NVIDIA H100 GPUs, it attains R2 = 0.969 and 11.97% MAPE, 3.1-3.5x below the SOTA, without per-LLM calibration. The SIMPLE law explains as well as predicts: prefill energy grows quadratically with the hidden dimension, so larger LLMs suffer most from long inputs; and more aggressive GQA KV grouping reduces the per-token energy by up to 3x relative to standard MHA. Increasing the reasoning token budget pushes output length beyond the analytically derived energy minimum, driving super-linear energy growth; although accuracy grows with the budget, energy grows faster, so Intelligence per Joule peaks at an intermediate budget and falls by up to 3.14x across a 4.0x reasoning-budget range. Agentic serving raises energy per request by 13.0x relative to non-agentic ones, driven by the growth of both the prefill and the KV-cache terms with input length; while prefix caching removes most of the prefill cost (-93%), the KV-cache term is unaffected, so even a cached agentic request costs 4.8x more.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!