文章基本信息

标题：Mode-assisted joint training of deep Boltzmann machines
本地全文：下载
作者：Haik Manukian ; Massimiliano Di Ventra
期刊名称：Scientific Reports
电子版ISSN：2045-2322
出版年度：2021
卷号：11
DOI：10.1038/s41598-021-98404-y
语种：English
出版社：Springer Nature
摘要：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.