How AI Uses Energy

When ChatGPT was unveiled to the public in 2022, energy usage was probably not on most people’s list of top things to wonder about. But the AI era will use a lot of power. Already, the energy usage of AI is increasing exponentially and by some estimates will surpass the energy used by Bitcoin by 2027.¹ And Bitcoin already consumes more energy than The Netherlands over the course of a year.²

There is a role for policy to help avoid a shortage of clean power for AI, but this memo focuses principally on the ways in which AI uses energy and why so much is needed.

Generative AI’s energy requirements consist of three parts: manufacturing and maintaining the equipment (“lifecycle”), training the AI model (“training”), and generating an output when prompted by a user (“inference”). This paper will focus on the latter two – training and inference – which now occur primarily in data centers.³

Training means exposing an AI model to data to improve its “understanding” in fields like language, biology, or economic statistics. Available data shows the energy usage for training AI models has increased rapidly for decades, beginning in the late 1950s with the “perceptron” artificial neuron,⁴ eventually doubling every 3.4 months between 2012 and 2018.⁵ Since the publication of open models like Mistral, Llama, and Gemma, that trend is likely not just accelerating but multiplying still further as independent developers train their own tailor-made AIs.

Inference is how most of us use AI: querying it and receiving a result (text, image, etc.). The energy cost of a single inference is minuscule, especially compared to the upfront cost of training the AI model. However, the cost is non-negligible and accumulates quickly. Every year, we perform trillions of Google searches⁶ if these searches were all performed with a chatbot like ChatGPT, they would use as much energy as Ireland.⁷

Training an AI model

Training an AI model involves feeding it data to “learn” about its domain of expertise. To better understand this process, consider three high-profile examples of AI training:

• GPT in ChatGPT stands for “Generative Pre-trained Transformer,” which tells you how important the training process is. While OpenAI reveals few details about GPT’s training, its data is known to include books, Wikipedia pages, and public websites.⁸ There is also extensive post-training testing and fine-tuning to ensure, for example, that ChatGPT does not output training data verbatim – a problem that has led to copyright infringement lawsuits.⁹
• AlphaZero, an AI developed by Google DeepMind to play the board games of chess, Go, and shogi (Japanese chess), was trained strictly by playing the board games against itself on supercomputers. It took nine hours of training to become proficient in chess, 12 hours for shogi, and 13 days for Go.¹⁰ Because it was not trained on human-played games, AlphaZero is renowned for not just playing better than humans, but creatively, deploying strategies humans have never imagined.
• AlphaFold, also by Google DeepMind,¹¹ was developed to predict the structures of protein and has revolutionized health care in the process. This process, when done experimentally, used to take months or even years. It was first trained for a week on 170,000 structures in the Protein Data Bank,¹² then trained again on 350,000 of its own predictions – a process known as self-distillation. AlphaFold has since predicted over 200 million protein structures in three years.

The more complex an AI model is, the more energy is needed to train it. In 2018, four years before ChatGPT opened to the public, OpenAI showed that the energy required to train frontier AI models was decupling (10x) each year.¹³ This trendline is even more nosebleed-inducing when compared with Moore’s Law, which has accurately projected that the computing power of an integrated circuit merely doubles every two years.

However, it is not clear whether this trendline will continue indefinitely. Hyperscalers and chip designers like Nvidia¹⁴ are continuously redesigning chips and data centers to improve energy efficiency at impressive rates. Google, for example, leverages its proprietary tensor processing units (TPUs),¹⁵ which are designed specifically for AI, unlike graphics processing units (GPUs), which are far more commonly used. With these optimizations, Google was able to train an AI model seven times more powerful than GPT-3¹⁶ with a third of the energy.¹⁷

With anxiety growing that AI developers may soon run out of internet content to train their AI on,¹⁸ there’s little basis to assume frontier AI training will necessarily grow exponentially more costly for much longer. In five years, AI labs will doubtless still be perfecting their frontier AI models, but they probably won’t be driven by exponentially increasing training anymore.

That, however, is only half the equation.

Inference from an AI

The other side of AI energy use is inference: when an AI model processes an input and infers the proper output, informed by its training.¹⁹ Even if you have never heard of it, you are familiar with the results:

• YouTube transcribing spoken text from a video into captions
• Outlook flagging incoming email as spam
• Google providing and ranking search results
• DALL-E or Midjourney making an image from a prompt
• Facebook or Instagram ranking your friends’ posts in your feed
• Alexa answering questions and fulfilling requests
• iPhone autocorrecting common typos
• Cameras detecting a human face

When you learn what AI inference is and how it is hidden in nearly every facet of our digital lives, you may notice that its energy cost is even more hidden. Of the AI inference examples above, all but the last two are performed on cloud servers rather than our own devices. That is why autocorrect still works without internet, but Alexa does not; it is why playing Xbox contributes to one’s electric bill but generating images with DALL-E or Midjourney doesn’t. When we rely on AI inference, we usually do not pay for it – the electricity comes from server farms, sometimes thousands of miles away.

Just as with training, the energy needed for inference can be thought of as a proxy for its complexity. Detecting a human face is easy but recognizing it is harder.²⁰ For a computer, processing text is trivial; reading words in an image or hearing them spoken is hard; and “understanding” words and their relationships to each other is very hard.

The energy challenge posed by inference today is that AI is performing extremely hard tasks. According to researchers,²¹ generative tasks are orders of magnitude more energy intensive than previously common AI applications like text classification (e.g., spam filtering). A Stable Diffusion image generator²² requires a whole smartphone charge of energy to produce one image; 1,000 images create as much CO2 as driving 4.1 miles. We can only imagine how much energy is needed to power OpenAI’s new video generator Sora.²³

In most cases, we can probably say inference is worth the cost. One text generation inference –one chatbot answer – only needs enough energy to power an LED lightbulb for about 17 seconds.²⁴ AlphaFold used far less energy and vastly less labor to infer 200 million protein structures from 2021-2024 than it took to determine 170,000 structures experimentally from 1976-2021.

However, AI appears positioned to outstrip the energy use of crypto soon. One study projected AI to use 0.5% of earth’s electricity by 2027²⁵ compared to 0.4% currently used for crypto mining,²⁶ and we can expect this growth will be driven primarily by increased inference. Often, as with AlphaFold, that inference can make the world and our lives vastly better, but we should be mindful that some frivolous or excessive AI usage could be a waste of resources.

What’s next?

A paradox of cloud computing is that we don’t notice how high its energy cost may be, but we also don’t notice how efficient it can be.

Google’s Tensor Processing Unit (TPU) is a great example of how hyperscalers – the largest cloud providers in the world – innovate to improve the efficiency of AI training and inference.²⁷ Non-Google developers can rent TPUs, but this specialized hardware cannot perform non-AI tasks well compared to more versatile graphics processing units (GPUs), and so only makes sense in a large, optimized data center with a constant stream of AI workloads where its efficiency gains can be maximized.

Lightweight AI is also expected to become more popular and diverse. The breakthrough of ChatGPT-3 was its size: 175 billion parameters, over 100x its predecessor (GPT-2).²⁸ While many innovators will strive to build bigger, better AI models, many will also build smaller, better models, like Google Gemma, available with either 2 billion or 7 billion parameters – small enough to run on a laptop. These models use much less energy and are ideal for narrower applications than larger, more powerful frontier models.

Nonetheless, it is an inescapable fact that significantly more energy will be needed to support the growth of AI. “The electrical grid is the largest and most complex machine humankind has ever built,” Third Way’s Shane Londagin writes in a recent paper. “But it is not prepared for the challenges of the future.”