Cargando…

Mode-assisted joint training of deep Boltzmann machines

The deep extension of the restricted Boltzmann machine (RBM), known as the deep Boltzmann machine (DBM), is an expressive family of machine learning models which can serve as compact representations of complex probability distributions. However, jointly training DBMs in the unsupervised setting has...

Descripción completa

Detalles Bibliográficos
Autores principales: Manukian, Haik, Di Ventra, Massimiliano
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8463543/
https://www.ncbi.nlm.nih.gov/pubmed/34561505
http://dx.doi.org/10.1038/s41598-021-98404-y
_version_ 1784572418435579904
author Manukian, Haik
Di Ventra, Massimiliano
author_facet Manukian, Haik
Di Ventra, Massimiliano
author_sort Manukian, Haik
collection PubMed
description The deep extension of the restricted Boltzmann machine (RBM), known as the deep Boltzmann machine (DBM), is an expressive family of machine learning models which can serve as compact representations of complex probability distributions. However, jointly training DBMs in the unsupervised setting has proven to be a formidable task. A recent technique we have proposed, called mode-assisted training, has shown great success in improving the unsupervised training of RBMs. Here, we show that the performance gains of the mode-assisted training are even more dramatic for DBMs. In fact, DBMs jointly trained with the mode-assisted algorithm can represent the same data set with orders of magnitude lower number of total parameters compared to state-of-the-art training procedures and even with respect to RBMs, provided a fan-in network topology is also introduced. This substantial saving in number of parameters makes this training method very appealing also for hardware implementations.
format Online
Article
Text
id pubmed-8463543
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-84635432021-09-27 Mode-assisted joint training of deep Boltzmann machines Manukian, Haik Di Ventra, Massimiliano Sci Rep Article The deep extension of the restricted Boltzmann machine (RBM), known as the deep Boltzmann machine (DBM), is an expressive family of machine learning models which can serve as compact representations of complex probability distributions. However, jointly training DBMs in the unsupervised setting has proven to be a formidable task. A recent technique we have proposed, called mode-assisted training, has shown great success in improving the unsupervised training of RBMs. Here, we show that the performance gains of the mode-assisted training are even more dramatic for DBMs. In fact, DBMs jointly trained with the mode-assisted algorithm can represent the same data set with orders of magnitude lower number of total parameters compared to state-of-the-art training procedures and even with respect to RBMs, provided a fan-in network topology is also introduced. This substantial saving in number of parameters makes this training method very appealing also for hardware implementations. Nature Publishing Group UK 2021-09-24 /pmc/articles/PMC8463543/ /pubmed/34561505 http://dx.doi.org/10.1038/s41598-021-98404-y Text en © The Author(s) 2021 https://creativecommons.org/licenses/by/4.0/Open AccessThis article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Manukian, Haik
Di Ventra, Massimiliano
Mode-assisted joint training of deep Boltzmann machines
title Mode-assisted joint training of deep Boltzmann machines
title_full Mode-assisted joint training of deep Boltzmann machines
title_fullStr Mode-assisted joint training of deep Boltzmann machines
title_full_unstemmed Mode-assisted joint training of deep Boltzmann machines
title_short Mode-assisted joint training of deep Boltzmann machines
title_sort mode-assisted joint training of deep boltzmann machines
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8463543/
https://www.ncbi.nlm.nih.gov/pubmed/34561505
http://dx.doi.org/10.1038/s41598-021-98404-y
work_keys_str_mv AT manukianhaik modeassistedjointtrainingofdeepboltzmannmachines
AT diventramassimiliano modeassistedjointtrainingofdeepboltzmannmachines