We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: Dropout as a Structured Shrinkage Prior

Abstract: Dropout regularization of deep neural networks has been a mysterious yet effective tool to prevent overfitting. Explanations for its success range from the prevention of "co-adapted" weights to it being a form of cheap Bayesian inference. We propose a novel framework for understanding multiplicative noise in neural networks, considering continuous distributions as well as Bernoulli noise (i.e. dropout). We show that multiplicative noise induces structured shrinkage priors on a network's weights. We derive the equivalence through reparametrization properties of scale mixtures and without invoking any approximations. Given the equivalence, we then show that dropout's Monte Carlo training objective approximates marginal MAP estimation. We leverage these insights to propose a novel shrinkage framework for resnets, terming the prior 'automatic depth determination' as it is the natural analog of automatic relevance determination for network depth. Lastly, we investigate two inference strategies that improve upon the aforementioned MAP approximation in regression benchmarks.
Comments: ICML 2019
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:1810.04045 [stat.ML]
  (or arXiv:1810.04045v3 [stat.ML] for this version)

Submission history

From: Eric Nalisnick [view email]
[v1] Tue, 9 Oct 2018 14:44:08 GMT (137kb,D)
[v2] Sun, 10 Feb 2019 14:35:37 GMT (862kb,D)
[v3] Wed, 29 May 2019 14:01:20 GMT (1170kb,D)

Link back to: arXiv, form interface, contact.