EcoLogits estimates GPU energy consumption per output token using a parametric model derived from the ML.ENERGY Leaderboard (specifically targeting NVIDIA H100 80GB HBM3 GPUs).
The model distinguishes between total parameters (used to calculate how many GPUs are needed to load the model) and active parameters ($P_{\text{active}}$) (used to estimate energy consumption per GPU).
For Sparse Mixture-of-Experts (SMoE) models, $P_{\text{active}}$ is the number of active experts, whereas for dense models, $P_{\text{active}} = P_{\text{total}}$.
The energy consumption per output token is modeled as a function of active parameters and batch size ($B$), following a linear-exponential form:
$$f_E(P_{\text{active}}, B) = \alpha e^{\beta B} P_{\text{active}} + \gamma$$
In current implementations, the batch size is assumed to be fixed at $B = 64$. The total GPU energy for a request is calculated as:
$$E_{\text{GPU}}(#T_{\text{out}}, P_{\text{active}}) = #T_{\text{out}} \times f_E(P_{\text{active}}, 64)$$
If the model requires multiple GPUs, this value is multiplied by the number of required GPUs.