Department of Information and Computer Science, Aalto University School of Science, Finland.
Department of Information and Computer Science, Aalto University School of Science, Finland.
Neural Netw. 2015 Apr;64:12-8. doi: 10.1016/j.neunet.2014.09.004. Epub 2014 Sep 28.
Restricted Boltzmann machines (RBMs) and deep Boltzmann machines (DBMs) are important models in deep learning, but it is often difficult to measure their performance in general, or measure the importance of individual hidden units in specific. We propose to use mutual information to measure the usefulness of individual hidden units in Boltzmann machines. The measure is fast to compute, and serves as an upper bound for the information the neuron can pass on, enabling detection of a particular kind of poor training results. We confirm experimentally that the proposed measure indicates how much the performance of the model drops when some of the units of an RBM are pruned away. We demonstrate the usefulness of the measure for early detection of poor training in DBMs.
受限玻尔兹曼机(RBM)和深度玻尔兹曼机(DBM)是深度学习中的重要模型,但通常难以全面衡量它们的性能,或者衡量特定情况下个别隐藏单元的重要性。我们建议使用互信息来衡量玻尔兹曼机中个别隐藏单元的有用性。该度量的计算速度很快,并且是神经元可以传递的信息量的上限,可以检测到特定类型的训练效果不佳的情况。我们通过实验证实,所提出的度量可以指示当 RBM 的某些单元被修剪掉时,模型的性能下降多少。我们证明了该度量在早期检测 DBM 中的训练不良情况方面的有用性。