5 ms·
Understanding the generalization of ‘lottery tickets’ in neural networks
- sandoooo 7y agoThe next question would be: is it possible to compare across multiple lottery ticket networks, and look for commonality? Maybe the lottery ticket params all give rise to some small set of particularly efficient configurations that we can add directly to other networks?
- deleted 7y ago[deleted]
- gradys 7y agoFor anyone looking for a quick summary of what a lottery ticket is, how they're found, and some implications, here's what I remember from a I talk I saw about lottery tickets at EMNLP 2019: - The core hypothesis is that within "over-parameterized"[0] networks, a small subnetwork (a small set of weights) is often doing most of the work. A weight can be thought of as an edge in the neural network graph, and so subsets of weights can be thought of as subnetworks. - You can find these subnetworks by initializing weights randomly, training for some number of steps, identifying the least contributing weights, and then retraining from the same initial parameters as before, except with the least-contributing weights from the previous run zero'd out. - People have observed that you can achieve something like 99% weight pruning with relatively little loss in performance. After that, things get very unstable. - This has implications for understanding how neural nets do what they do and for shrinking model sizes. This is all from memory, so forgive any errors. [0] - Networks have grown to enormous numbers of parameters lately, and there's reason to think that even before the era of 1B+ parameter networks, neural nets were over-parameterized. Why do we use such large networks then? For a given trained neural net, there might be a much smaller one that does the same thing, but in practice, it's difficult to get the same performance by training a smaller network. This may be because our typical optimization methods aren't well suited for finding these lottery tickets.
- dmurray 7y ago> Why do we use such large networks then? For a given trained neural net, there might be a much smaller one that does the same thing, but in practice, it's difficult to get the same performance by training a smaller network. Also, because it's cheap to run a network with a billion parameters. Lots of real-world applications are constrained not by compute, but by size of the training set.
- longtom 7y agoMaybe the additional parameters give the entire network more "leeway" to find such subnetwork structures, i.e. ease gradient descent by smoothing the loss landscape?
- Akababa 7y agoThat's intuitive but doesn't support the result of the lucky subnetwork (once found and re-initialized) training faster and outperforming the original.
- longtom 7y agoThis does not seem to be a contradiction. Once you are in the right region of solution space training is expected to be faster and easier. Re-initialization could have a regularizing effect, explaining the better performance.
- Akababa 7y agoThey re-use the same initialization, so it appears that the initial weights are inherently coupled with the nonzero structure.
- edoo 7y agoIt seems like that should be expected. The point is to have a ton of data with possible patterns to find and you have no idea which data links to what other data. You'd expect most data to not correlate. You'd expect strong patterns to emerge between a subset of the parameters. You'd expect to be able to prune off the useless parameters after the fact.
- deleted 7y ago[deleted]
- closetCS 7y agoI think the main question I have about these smaller, "lottery ticket" networks is that they are being trained over and over on the same problem as the bigger network, and are being evaluated on the same dataset as the big network, which leads me to believe that the model will fail to generalize to different but still related problems. Like if the model was trained on Imagenet an the winning ticket was found that had ridiculous accuracy for a relatively small network, I would think it would be heavily overfitted
- bonoboTP 7y agoThe article discusses this as its main theme and shows generalization across datasets.