about
Deep Mind – On the importance of single directions for generalization (arxiv.org)
7 points by godelmachine on Mar 23, 2018 | hide | past | pdf | discuss on HN

In plain words: They measured how much trained networks lean on single units or mixes of units to decide answers, across label noise, settings, and training. Networks that generalized relied far less on such single directions, and batch normalization pushed them away from this habit more than dropout.

Abstract · On the importance of single directions for generalization

Despite their ability to memorize large datasets, deep neural networks often achieve good generalization performance. However, the differences between the learned solutions of networks which generalize and those which do not remain unclear. Additionally, the tuning properties of single directions (defined as the activation of a single unit or some linear combination of units in response to some input) have been highlighted, but their importance has not been evaluated. Here, we connect these lines of inquiry to demonstrate that a network's reliance on single directions is a good predictor of its generalization performance, across networks trained on datasets with different fractions of corrupted labels, across ensembles of networks trained on datasets with unmodified labels, across different hyperparameters, and over the course of training. While dropout only regularizes this quantity up to a point, batch normalization implicitly discourages single direction reliance, in part by decreasing the class selectivity of individual units. Finally, we find that class selectivity is a poor predictor of task importance, suggesting not only that networks which generalize well minimize their dependence on individual units by reducing their selectivity, but also that individually selective units may not be necessary for strong network performance.

Ari S. Morcos, David G. T. Barrett, Neil C. Rabinowitz, Matthew Botvinick
arXiv:1803.06959 · stat.ML, cs.AI, cs.LG, cs.NE · submitted Mar 19, 2018 · updated May 22, 2018
abstract · pdf · html · ICLR 2018 conference paper; added additional methodological details

add comment on HN
Also discussed: Mar 2018 (1 point, 0 comments) · Mar 2018 (8 points, 0 comments)