Trading Off Compute in Training and Inference – Epoch
Some techniques allow to increase the performance of machine learning models at the cost of more expensive inference, or reduce inference compute at the cost of lower performance. This possibility induces a tradeoff between spending more resources on training or on inference. We explore the characteristics of this tradeoff and outline some implications for AI governance. In current machine learning systems, the performance of a system is closely related to how much compute is spent during the training process. However, it is also possible to augment the capabilities of a trained model at the cost of increasing compute usage during inference or reduce compute usage during inference at the cost of lower performance. For example, models can be pruned to reduce their inference cost, or instructed to reason via chains of thought, which increases their inference cost. Based on evidence from five concrete techniques (model scaling, Monte Carlo Tree Search, pruning, resampling, and chain of th
Trading off compute in training and inference | Epoch AI Key takeaways In current machine learning systems, the performance of a system is closely related to how much compute is spent during the training process. However, it is also possible to augment the capabilities of a trained model at the cost of increasing compute usage during inference or reduce compute usage during inference at the cost of lower performance. For example, models can be pruned to reduce their inference cost, or instructed to reason via chains of thought, which increases their inference cost. Based on evidence from five
related reading
- Optimally allocating compute between inference and trainingepoch.ai
- Optimally allocating compute between inference and training | Epoch AIepochai.org
- Spending Inference Time - Kevin Lukevinlu.ai
- Transformer Inference Arithmetic | kipply's blogkipp.ly
- Transformer Math 101 | EleutherAI Blogblog.eleuther.ai
- The Extreme Inefficiency of RL for Frontier Models - Toby Ordtobyord.com
- Efficient LLM inferencefinbarrtimbers.substack.com
- Estimating training compute of deep learning models | Epoch AIepochai.org
- [2509.14786] Pre-training under infinite computearxiv.org
- [2510.13786] The Art of Scaling Reinforcement Learning Compute for LLMsarxiv.org
- All About Transformer Inferencejax-ml.github.io
- Navigating the High Cost of AI Compute | Andreessen Horowitza16z.com