arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.08562cs.LGcs.AInlin.AO
收藏

Mode-Assisted Joint Training of Deep Boltzmann Machines

University of California, San Diego(加州大学圣地亚哥分校)

Haik Manukian, Massimiliano Di Ventra

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑