Projecting compute trends in Machine Learning

post by Tamay, lennart, Jsevillamol · 2022-03-07T15:32:13.278Z · LW · GW · 5 comments

Contents

  Summary
  Introduction
  When will the current scaling trend revert back to Moore’s law?
  Projecting ML compute trends
  Conclusion
  Details of the simulations
None
5 comments

Summary

Using our dataset of milestone Machine Learning models, and our recent analysis of compute trends in ML, we project forward 70 years worth of trends in the amount of compute used to train Machine Learning models. Our simulations account for (a) uncertainty in estimates of the growth rates in compute usage during the Deep Learning (DL)-era and Pre-DL era, and (b) uncertainty over the ‘reversion date’, i.e. the date when the current DL-era compute trend (with a ~6 month doubling time) will end and revert to the historically more common trend associated with Moore’s law. Assuming a reversion date of between 8 to 18 years, and without accounting for algorithmic progress, our projections suggest that the median of Cotra 2020 [LW · GW]’s biological anchors may be surpassed around August 2046 [95% CI: Jun 2039, Jul 2060]. This suggests that historical rates of compute scaling, if sustained briefly (relative to how long these trends have been around so far), could result in the emergence of transformative models.

Our work can be replicated using this Colab notebook.

Note: we present projections, not predictions. Our post answers the question of: “What would historical trends over the past 70 years when naively extrapolated forward imply about the future of ML compute?” It does not answer the question: “What should our all-things-considered best guess be about how much compute we should expect will be used in future ML experiments?

Introduction

Recently, we put together a dataset of over a hundred milestone Machine Learning models, spanning from 1952 to today, annotated with the compute required to train them. Using this data, we produce simple projections of the amount of compute that might be used to train future ML systems.

The question of how much compute we might have available to train ML systems has received some attention in the past, most notably in Cotra’s Biological Anchors report. Cotra’s report investigates TAI timelines by analyzing: (i) the training compute required for the final training run of a transformative model (using biological anchors), and (ii) the amount of effective compute available at year Y. This article replaces (ii) the compute estimate by projecting 70 years worth of trends in the amount of compute used to train Machine Learning models.

Cotra's amount of effective compute available at year Y is broken down into forecasts of (a) compute cost, (b) compute spending, and (c) algorithimic progress. By contrast, we do not decompose the estimate, and rather project it on our previous investigation of training compute of ML milestone systems. This trend includes the willingness to spend over time including the reduced compute costs over time; however, it does not address algorithmic progress. We explicitly do not forecast the cost of compute or compute spending.

 Figure 1. Contrasting our work with that of Cotra 2020 [LW · GW]

In this post, we present projections based on previously observed trends and some basic insights about how long the current 6-month doubling time can be sustained. That is, our post answers the question of: what would current trends imply about the future if you naively extrapolate them forwards.

One key reason we don’t expect these projections to be particularly good predictions is that it seems likely that Moore’s law might break down in some important way over the next few decades. We therefore might expect that that the doubling-time in compute usage, when the dollar-budgets to scale compute grow at the economic growth-rate, will be substantially longer than the historically common ~20-month doubling period.

When will the current scaling trend revert back to Moore’s law?

In our recent analysis of compute trends in ML (Sevilla et al., 2022), we find that, since the advent of Deep Learning, the amount of compute used to train ML systems has been doubling every 6 months. This is much faster than the previous historical doubling time that we find to be roughly 20 months (which is roughly in line with Moore’s law). Previous work (Carey, 2018, and Lohn and Musser, 2022) has pointed out that a scaling-rate that outstrips Moore’s law by a wide margin cannot be sustained for many years as a rate of growth in ML compute spending that far exceeds economic growth cannot be sustained for many years.

A key question, then, for projecting compute used in future ML systems, is: How long can the current fast trend continue, before it reverts to the historically much more common trend associated with Moore’s law?

To answer this question, we replicate the analysis by Carey, 2018, but instead of using the numbers from OpenAI’s AI and Compute (Amodei and Hernandez, 2018), we use the numbers from our recent analysis (summary [AF · GW]).[1] This analysis, roughly, points to three scenarios:

Roughly, we might say that these scenarios are represented by the following distributions over ‘reversion dates’, i.e. dates when the scaling trends are more similar to Moore’s law than they are to the current fast trend.

Fig 2. Distributions that roughly correspond to the three scenarios that come out of our replication of Carey, 2018.[1]

We then produce a mixture of these distributions by creating a weighted linear pool where “Bearish” is assigned 0.75, “Middle of the road” is assigned 0.20, and “Bullish” 0.05, based on our best-guesses (you can apply your own weights using this Colab notebook.)

Fig 3. our best-guess for a prior over reversion dates, formed by mixing the previous distributions

We can use this as our prior over when the fast-trend will revert to the more historically common trend associated with Moore’s law.

We simulate compute paths based on (a) our estimates of the growth rates in compute usage during the DL-era and Pre-DL era, and (b) our prior over ‘reversion date’, i.e. the date when the current DL-era compute trend will end. We account for the uncertainty in both (a) and (b) in our simulations (see details here [LW · GW]).

Fig 4. 10,000 projected compute paths. Solid line represents the median projected compute at each date, and the shaded region represents 2-standard deviations around the median.

Our simulations reveal the following projections about the amount of compute used to train ML models.

    YearProjected FLOPs used to train largest ML modelEnough for how many anchor’s median compute requirements?
2025 []0/6
2030 []0/6
2040 []1/6
2050 []3/6
2060 []5/6
2070 []5/6
2080 []6/6

Table 1: Projected FLOPs from 2025 to 2080

These projections suggest that, without accounting for algorithmic progress, the most modest of Cotra 2020 [LW · GW]’s biological anchors will be surpassed around August 2030 [95% CI: Jan 2029, May 2038], the median anchor (~ FLOPS) will be surpassed around August 2046 [95% CI: Jun 2039, Jul 2060], and the strongest of anchors will be surpassed around May 2072 [95% CI: Jan 2057, Jun 2089].

Conclusion

If we naively extrapolate the trends uncovered from 70-years worth of compute scaling in Machine Learning, we find that within roughly 25 years, large-scale ML experiments will use amounts of compute that exceed the half of the compute budgets that Cotra 2020 [LW · GW] has suggested may be sufficient for training a transformative model. This highlights the fact that historical rates of compute scaling in Machine Learning, even if sustained relatively briefly (relative to how long these trends have been around so far), could place us in novel territory where it might be likely that transformative systems would be trained. This work also suggests that understanding compute trends might be a promising direction for predicting ML progress, 


Details of the simulations

We assume compute grows exponentially in time at some rate 

In our projections, we replace  with , defined as a weighted geometric mean of our best-guess of the growth rate during Moore's law (), and the growth rate of our estimate of the growth rate during the Deep-Learning Era (): 

Here,  simply denotes the growth rate during the Deep Learning Era (2010 onwards) as estimated using OLS. In particular, we estimate the following model using our dataset: 

  is defined as follows: 

 where  is the estimated growth rate during the Pre-DL era, and  is the growth rate implied by a 20-month doubling period. The reason we take the geometric mean of the estimated growth rate, and the growth rate implied by a 20-month doubling period is because Moore's law is sufficiently well-established that the error bars around  are too large relative to how well-established Moore's law is. We therefore artificially increase our precision of the growth rate associated with Moore's law by taking an average of our estimated value and the usual growth rate implied by an ~20-month doubling-time.

Our weight function, , is constructed as follows: 

Why? Well, it's a logistic-like function with a unit-interval range, which exceeds  when , equals  when , and is less than  otherwise. This is what it looks like:

We then simulate some path  as follows: 

  1. ^

    You can find the details of this analysis and a comparison to Carey’s results here.

5 comments

Comments sorted by top scores.

comment by delton137 · 2022-03-09T00:53:15.215Z · LW(p) · GW(p)

Thanks for sharing! That's a pretty sophisticated modeling function but it makes sense. I personally think Moore's law (the FLOPS/$ version) will continue, but I know there's a lot of skepticism about that.

Could you make another graph like Fig 4 but showing projected cost, using Moore's law to estimate cost? The cost is going to be a lot, right?

Replies from: Tamay, lennart
comment by Tamay · 2022-03-12T23:23:28.859Z · LW(p) · GW(p)

Thanks! 

Could you make another graph like Fig 4 but showing projected cost, using Moore's law to estimate cost? The cost is going to be a lot, right?

Good idea. I might do this when I get the time—will let you know!

comment by lennart · 2022-03-12T15:42:01.608Z · LW(p) · GW(p)

We basically lumped the reduced cost of FLOP per $ and increased spending together.

A report from CSET on AI and Compute projects the costs by using two strongly simplified assumptions: (I) doubling every 3.4 months (based on OpenAI's previous report) and (II) computing cost stays constant. This could give you some ideas on rather upper bounds of projected costs.

Carey's previous analysis uses this dataset from AI Impacts and therefore assumes:

[..] while the cost per unit of computation is decreasing by an order of magnitude every 4-12 years (the long-run trend has improved costs by 10x every 4 years, whereas recent trends have improved costs by 10x every 12 years).

Replies from: delton137
comment by delton137 · 2022-04-01T17:40:02.910Z · LW(p) · GW(p)

Interesting, thanks. 10x reduction in cost every 4 years is roughly twice what I would have expected. But it sounds quite plausible especially considering AI accelerators and ASICs.

comment by Jotto999 · 2022-12-10T16:53:00.918Z · LW(p) · GW(p)

Sorry if I'm just misreading -- in Compute Trends Across Three eras of Machine Learning [LW · GW] it was shown that the doubling time (at that time) had slowed to every ~10 months for the large-scale projects.  In this projection you go with a 6-month doubling time for x number of years, then slowing to every 20 months.  My questions are:

  1. What would the results be like if we assumed things had already slowed to 10 months?
  2. Is 6 months likely to be a better description of the upcoming computation doubling times for the next few years, versus 10 months? If yes, why?