arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1301.3545cs.LGcs.NEstat.ML
收藏

Boltzmann机联合训练的无度量自然梯度

Metric-Free Natural Gradient for Joint-Training of Boltzmann Machines

Université de Montréal(蒙特利尔大学)

Guillaume Desjardins, Razvan Pascanu, Aaron Courville, Yoshua Bengio

摘要:

本文介绍了用于训练玻尔兹曼机的无度量自然梯度(MFNG)算法。与Martens [8]的Hessian-Free方法的精神相似,我们的算法属于截断牛顿方法家族,并利用有效的矩阵向量积来避免显式存储自然梯度度量$L$。该度量被示出为对数配分函数的预期二阶导数(在模型分布下),或者等价地,能量函数的偏导数的向量的方差。我们在联合训练3层深度玻尔兹曼机的任务上评估了我们的方法,并表明MFNG确实比具有中心的随机最大似然函数具有更快的每历元收敛,尽管挂钟性能目前没有竞争力。

英文摘要:

This paper introduces the Metric-Free Natural Gradient (MFNG) algorithm for training Boltzmann Machines. Similar in spirit to the Hessian-Free method of Martens [8], our algorithm belongs to the family of truncated Newton methods and exploits an efficient matrix-vector product to avoid explicitely storing the natural gradient metric $L$. This metric is shown to be the expected second derivative of the log-partition function (under the model distribution), or equivalently, the variance of the vector of partial derivatives of the energy function. We evaluate our method on the task of joint-training a 3-layer Deep Boltzmann Machine and show that MFNG does indeed have faster per-epoch convergence compared to Stochastic Maximum Likelihood with centering, though wall-clock performance is currently not competitive.

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

↑