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: Improving Predictive Uncertainty Estimation using Dropout -- Hamiltonian Monte Carlo

Abstract: Estimating predictive uncertainty is crucial for many computer vision tasks, from image classification to autonomous driving systems. Hamiltonian Monte Carlo (HMC) is an sampling method for performing Bayesian inference. On the other hand, Dropout regularization has been proposed as an approximate model averaging technique that tends to improve generalization in large scale models such as deep neural networks. Although, HMC provides convergence guarantees for most standard Bayesian models, it does not handle discrete parameters arising from Dropout regularization. In this paper, we present a robust methodology for improving predictive uncertainty in classification problems, based on Dropout and Hamiltonian Monte Carlo. Even though Dropout induces a non-smooth energy function with no such convergence guarantees, the resulting discretization of the Hamiltonian proves empirical success. The proposed method allows to effectively estimate the predictive accuracy and to provide better generalization for difficult test examples.
Comments: 26 Pages, 12 Figures, version 3, to appear in Soft Computing, Author preprint
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:1805.04756 [cs.LG]
  (or arXiv:1805.04756v3 [cs.LG] for this version)

Submission history

From: Matias Valdenegro-Toro [view email]
[v1] Sat, 12 May 2018 18:11:33 GMT (1672kb,D)
[v2] Mon, 9 Jul 2018 01:17:11 GMT (3290kb,D)
[v3] Tue, 2 Jul 2019 13:49:05 GMT (5019kb,D)

Link back to: arXiv, form interface, contact.