JournalAI Infrastructure

Field guide / 6

PFI-aware power caps recovered 63% of the gap to an oracle in simulated AI training

A 131-run study found that training jobs respond differently to GPU power caps. An elasticity-aware scheduler preserved more throughput than equal cuts.

Sep 13, 20266By ISH Team
PFI-aware power caps recovered 63% of the gap to an oracle in simulated AI training
Advertisement

PFI-aware power caps recovered 63% of the gap to an oracle in simulated AI training

If a cluster needs to cut power by 30%, spreading the reduction evenly sounds fair. It can waste throughput. Some training jobs lose little speed under a lower GPU cap. Others slow sharply.

Characterizing Job Power Elasticity for Power-Flexible AI Training measured that variation across 131 LLM training runs on NVIDIA H200 GPUs, plus 24 H200 validation runs and 34 matched H100 runs. The authors call the sensitivity to reduced power power elasticity and use a Power Flexibility Index, or PFI, to rank it.

In simulated 100-job clusters, a PFI-aware allocator recovered about 1,500 tokens per second per job under a 30% power reduction. That was 63% of the gap between equal-weight cuts and an oracle that knew every job's full power-throughput curve.

The paper does not show that a live cluster will reproduce that gain. It shows why one cluster-wide cap conceals job-level differences.

Same cap, different slowdown

The experiments included dense and mixture-of-experts models, pretraining and rank-32 LoRA fine-tuning, and one to 32 GPUs. Each run swept GPU caps across its operating range while the team measured aggregate GPU power and tokens per second.

PFI averages the ratio of proportional power reduction to proportional throughput reduction. A high-PFI job retains more throughput when power falls. The paper describes it as more power flexible. A low-PFI job needs protection because its throughput is more sensitive to a cap.

This differs from a static joules-per-token comparison. The operational question is which job can accept a lower cap during constrained capacity without losing more work than another job would. The earlier INT4 energy result made a nearby point: a lower GPU power draw can raise total energy if the run takes longer.

Telemetry beat a large feature set

A production scheduler cannot sweep every job first. The researchers tested whether NVIDIA Data Center GPU Manager telemetry could estimate PFI from an uncapped run.

The product of DRAM activity and memory-copy utilization was their best held-out predictor. Its leave-one-cluster-out Spearman correlation was 0.515. Adding arithmetic intensity or job metadata did not improve that result. A model using every DCGM metric failed outside its training sample.

This is a warning against collecting signals without a held-out test. NVIDIA's DCGM documentation describes monitoring and power-management facilities for cluster environments.

The study controlled GPUs through nvidia-smi -pl. NVIDIA describes it as a maximum GPU power limit within device-supported bounds. A cap is not an exact draw target, so a scheduler needs realized power and throughput, not a nominal value alone.

The result comes from a simulation

The authors built 500 synthetic 100-job workloads with random and production-shaped mixes. They compared equal weighting, a hand-tuned MoE fine-tuning rule, raw DRAM-copy-product allocation, PFI-aware allocation and the oracle.

PFI-aware allocation equaled or exceeded equal weighting throughout the simulation. The production-shaped mix benefited more because small fine-tuning jobs tended to have higher PFI and large pretraining jobs lower PFI.

That pattern is not a rule for every architecture. The paper has 25 PFI data points derived from its 131 runs. Measure local workloads before assuming a category will behave the same way.

Turn flexibility into an operating rule

EPRI's DCFlex initiative runs data-center flexibility demonstrations aimed at grid reliability and faster interconnections. A cluster cannot safely respond to a grid signal by cutting every GPU equally. It needs limits on duration, acceptable throughput loss and rollback.

  1. Record realized GPU power, tokens per second, model family, training phase and cap changes.
  2. Sweep representative workloads before offering a flexible training tier.
  3. Set a conservative floor for jobs that lose throughput quickly or have risky checkpoint timing.
  4. Use telemetry estimates to rank jobs, then validate the policy against measured curves.
  5. Measure whole-system energy before claiming a carbon or cost benefit. This study excludes CPU, DRAM, networking, power supplies, fans and cooling.

The TechforHumans replay gate provides a workable discipline: replay representative runs after a hardware, training-stack, architecture or policy change. A power curve belongs to a particular setup.

Keep the safety margin visible

A flexible tier should not make a power cap invisible to the team running the job. The control plane needs to report the requested cap, realized GPU draw, estimated PFI, throughput before and after the change, and the reason for the curtailment. Those records make it possible to distinguish a useful load reduction from a job that was already slow because of data loading, a failed worker or a bad checkpoint.

The same record supports a sensible rollback rule. If a protected job loses more throughput than its measured curve predicted, restore its power before applying a deeper reduction elsewhere. If several flexible jobs are paused at once, the cluster may meet a power target but create a synchronized recovery spike later. The study does not model that behavior, so it belongs in an operator's live tests rather than an assumed benefit.

EPRI's demonstrations make the grid side concrete, while this paper supplies a workload-side measurement. Neither one removes the need for a site-specific agreement between the utility, data-center operator and training customer about curtailment duration and service obligations.

The boundaries matter

The study uses H200 GPUs on one node type, 30-minute pretraining and LoRA SFT runs, and GPU power alone. Its 32-GPU tests cover Qwen3-32B pretraining only. It excludes inference, RLHF, heterogeneous accelerators and live scheduler validation.

Fourteen configurations were repeated. Median variation was 0.4% for steady-state aggregate GPU power and 0.5% for tokens per second. That supports the controlled measurements, not the simulated scheduler under real workload arrivals and failures.

Equal cuts are a baseline rather than a policy. When a cluster must reduce power, measure which workloads bend first and protect the ones that do not.

#AI energy#GPU power#LLM training#data centers#power flexibility#PFI
Advertisement

Keep reading

Related stories

Browse the archive