5
$\begingroup$

Is it practical/affordable to train an AlphaZero/MuZero engine using a residential gaming PC, or would it take thousands of years of training for the AI to learn enough to challenge humans?

I'm having trouble wrapping my head around how much computing power '4 hours of Google DeepMind training' equates to my residential computer running 24/7 trying to build a trained AI.

Basically, are AlphaZero or MuZero practical for indie board games that want a state of the art AI, or is it too expensive to train?

$\endgroup$

2 Answers 2

3
$\begingroup$

The vast majority of neural networks are now trained on graphics processing units (GPUs) or specialised accelerator hardware such as tensor processing units (TPUs).

In Mastering Chess and Shogi by Self-Play with a General Reinforcement Learning Algorithm, Silver et al. say that the training process involved 5,000 first-generation TPUs generating self-play games and 64 second-generation TPUs for training. This is certainly far beyond what any practical gaming computer is likely to achieve, as you'll likely only have one GPU, and that might not even rival a single TPU. Training on the CPU will be substantially slower again than either a GPU or TPU. Training would be orders of magnitude slower; you might find these benchmarks by Wang et al. of interest.

$\endgroup$
1
  • 1
    $\begingroup$ Not arguing with any part of this answer, and I may add that I think gcp (creator of Leela zero) estimated something like 10000 years in single GPU (estimated 3 years ago). However this training time depends on the game complexity a lot and also if the game has no professional players then a single GPU may well be the best in the world after a few months if not a few weeks or even days. $\endgroup$
    – davik
    Feb 15, 2021 at 19:23
2
$\begingroup$

For board games, you may not need an advanced computer to achieve very good results. TPU and high-end GPUs allow to go one step further of course.

But there is an existing implementation on github alpha-zero-general (not from me). By adding some optimizations tricks from scientific articles, I managed to train alphazero for several board games (Splendor, Santorini, Machi Koro, The Little Prince Make Me A Planet) with my i5 CPU. You may not need GPU as most of the time is spent in self-playing more than training, meaning that computer needs to do lots of network inferences more than gradient. After about a day of training, the training seemed to reach its asymptote. As a result, the computer was winning over me 100% on each game.

I don't want to promote my self github, just telling OP that is feasible.

$\endgroup$

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .