8
$\begingroup$

I have read somewhere on the web (I lost the reference) that the number of units (or neurons) in a hidden layer should be a power of 2 because it helps the learning algorithm to converge faster.

Is this a fact? If it is, why is this true? Does it have something to do with how the memory is laid down?

$\endgroup$
0

2 Answers 2

14
$\begingroup$

I have read somewhere on the web (I lost the reference) that the number of units (or neurons) in a hidden layer should be a power of 2 because it helps the learning algorithm to converge faster.

I would quite like to see a reference to this suggestion, in case it has been misunderstood.

As far as I know, there is no such effect in normal neural networks. In convolutional neural networks, it might potentially be true in a minor way because some FFT approaches work better with $2^n$ items.

Is this a fact? If it is, why is this true? Does it have something to do with how the memory is laid down?

I would say that this is not a general fact. Instead, it seems like misunderstood advice to search some hyperparameters such as number of neurons in each layer, by increasing or decreasing by a factor of 2. Doing this and trying layer sizes of 32, 64, 128 etc should increase the speed of finding a good layer size compared to trying sizes 32, 33, 34 etc.

The main reason to pick powers of 2 is tradition in computer science. Provided there is no driver to pick other specific numbers, may as well pick a power of 2 . . . but equally you will see researchers picking multiples of 10, 100 or 1000 as "round numbers", for a similar reason.

One related factor: If a researcher presents a result for some new technique where the hidden layer sizes were tuned to e.g. 531, 779, 282 etc, then someone reviewing the work would ask the obvious question "Why?" - such numbers might imply the new technique is not generic or requires large amounts of hyperparameter tuning, neither of which would be seen as positive traits. Much better to be seen using an obvious "simple" number . . .

$\endgroup$
0
1
$\begingroup$

There is a hardware based reasoning. Matrix multiplication is one of the central computations in deep learning. SIMD operations in CPUs happen in batch sizes, which are powers of 2.

Here is a good reference about speeding up neural networks on CPUs by leveraging SIMD instructions:

Improving the speed of neural networks on CPUs

You will notice batch sizes that are powers of 2. This is a good paper to read about implementing neural networks using SIMD instructions.

$\endgroup$
1
  • 1
    $\begingroup$ Do all sizes for the batches really need to be a power of 2 in the case of SIMD operations? It would be nice if you can provide a link to a reliable reference that supports this claim or just for us to read more about the topic. $\endgroup$
    – nbro
    Jan 20, 2021 at 23:45

You must log in to answer this question.

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