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
Explore this link on the map →related reading
- Optimally allocating compute between inference and training | Epoch AIepochai.org
- How To Scale Your Modeljax-ml.github.io
- Composer2.pdfcursor.com
- Spending Inference Time - Kevin Lukevinlu.ai
- How Well Does RL Scale? - Toby Ordtobyord.com
- Distinguish between inference scaling and "larger tasks use more compute" — AI Alignment Forumalignmentforum.org
- The Scaling Hypothesis · Gwern.netgwern.net
- o3 — LessWronglesswrong.com
- My picture of the present in AI — LessWronglesswrong.com
- LLM Inference Performance Engineering: Best Practices | Databricks Blogdatabricks.com
- Reiner Pope – The math behind how LLMs are trained and serveddwarkesh.com
- Inference Scaling Reshapes AI Governance - Toby Ordtobyord.com