A topological insight into restricted Boltzmann machines

D.C. Mocanu, E. Mocanu, H.P. Nguyen, M. Gibescu, A. Liotta

Research output: Contribution to journalArticleAcademicpeer-review

30 Citations (Scopus)
212 Downloads (Pure)

Abstract

Restricted Boltzmann Machines (RBMs) and models derived from them have been successfully used as basic building blocks in deep artificial neural networks for automatic features extraction, unsupervised weights initialization, but also as density estimators. Thus, their generative and discriminative capabilities, but also their computational time are instrumental to a wide range of applications. Our main contribution is to look at RBMs from a topological perspective, bringing insights from network science. Firstly, here we show that RBMs and Gaussian RBMs (GRBMs) are bipartite graphs which naturally have a small-world topology. Secondly, we demonstrate both on synthetic and real-world datasets that by constraining RBMs and GRBMs to a scale-free topology (while still considering local neighborhoods and data distribution), we reduce the number of weights that need to be computed by a few orders of magnitude, at virtually no loss in generative performance. Thirdly, we show that, for a fixed number of weights, our proposed sparse models (which by design have a higher number of hidden neurons) achieve better generative capabilities than standard fully connected RBMs and GRBMs (which by design have a smaller number of hidden neurons), at no additional computational costs.
Original languageEnglish
Pages (from-to)243-270
Number of pages28
JournalMachine Learning
Volume104
Issue number2
DOIs
Publication statusPublished - Sep 2016

Fingerprint Dive into the research topics of 'A topological insight into restricted Boltzmann machines'. Together they form a unique fingerprint.

  • Cite this