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

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

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

Computer Science > Machine Learning

Title: Advanced Dropout: A Model-free Methodology for Bayesian Dropout Optimization

Abstract: Due to lack of data, overfitting ubiquitously exists in real-world applications of deep neural networks (DNNs). We propose advanced dropout, a model-free methodology, to mitigate overfitting and improve the performance of DNNs. The advanced dropout technique applies a model-free and easily implemented distribution with parametric prior, and adaptively adjusts dropout rate. Specifically, the distribution parameters are optimized by stochastic gradient variational Bayes in order to carry out an end-to-end training. We evaluate the effectiveness of the advanced dropout against nine dropout techniques on seven computer vision datasets (five small-scale datasets and two large-scale datasets) with various base models. The advanced dropout outperforms all the referred techniques on all the datasets.We further compare the effectiveness ratios and find that advanced dropout achieves the highest one on most cases. Next, we conduct a set of analysis of dropout rate characteristics, including convergence of the adaptive dropout rate, the learned distributions of dropout masks, and a comparison with dropout rate generation without an explicit distribution. In addition, the ability of overfitting prevention is evaluated and confirmed. Finally, we extend the application of the advanced dropout to uncertainty inference, network pruning, text classification, and regression. The proposed advanced dropout is also superior to the corresponding referred methods. Codes are available at this https URL
Comments: Accepted by IEEE TPAMI, 2021
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
DOI: 10.1109/TPAMI.2021.3083089
Cite as: arXiv:2010.05244 [cs.LG]
  (or arXiv:2010.05244v2 [cs.LG] for this version)

Submission history

From: Jiyang Xie [view email]
[v1] Sun, 11 Oct 2020 13:19:58 GMT (5002kb,D)
[v2] Tue, 10 Aug 2021 08:04:11 GMT (11424kb,D)

Link back to: arXiv, form interface, contact.