Re:An idea from physics helps AI see in higher dimensions
by sillysaurusx on Thursday January 09, 2020 @01:52PM
(#22004200)
https://www.docdroid.net/faDq8Bu/swarm-training-v01a.pdf
We average the weights themselves, and the efficiency seems to be similar to gradient gathering.
It’s also averaging in slices, not the full model. There’s never a full resync.
SWA is the theoretical basis for why it works, I think.
Another way of thinking about it: If the gradients can be averaged, then so can the weights.
Re:An idea from physics helps AI see in higher dimensions
by uoaei on Thursday January 09, 2020 @03:39PM
(#22005437)
If you are averaging weights often enough, then it's basically the same as averaging gradients. If you average the weights of a bunch of independently-trained models, you're going to have a rough time. Even if the function computes the exact same thing, the order of rows and columns in the intermediate matrices will totally ruin your averaging strategy.
w2tanh(w1x) = -1w2tanh(-w1*x)
But if you average the weights in those two equivalent models you get 0.
If you're talking about asynchronous data parallelism, then there can be some averaging of weights, but they all start with the same weights and are re-synched often enough that weights are never too different to break it.