arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1809.10232stat.MLcs.LG
收藏

SGD的矩阵李群预条件子

Preconditioner on Matrix Lie Group for SGD

GMEMS Technologies and Spectimbre(GMEMS技术和Spectimbre(公司));Fairview Way, Milpitas, CA 95035(366 Fairview Way, Milpitas, CA 95035(地址,非机构))

Xi-Lin Li

摘要:

我们研究了两种类型的预条件和预条件随机梯度下降(SGD)方法在一个统一的框架。第一种方法与Newton方法有密切的关系,我们称之为Newton型;第二种方法与费舍尔信息矩阵的逆有密切的关系,我们称之为Fisher型。这两个预条件可以从一个框架中导出,并使用自然或相对梯度下降最小化某些预条件估计标准在用户指定的任何矩阵李群上进行有效估计。许多现有的预处理器和方法,例如,RMSProp、Adam、KFAC、平衡SGD、批次标准化等,是Newton型或Fisher型的特例或与之密切相关。针对较大规模的机器学习问题,给出了相应的实验结果.

英文摘要:

We study two types of preconditioners and preconditioned stochastic gradient descent (SGD) methods in a unified framework. We call the first one the Newton type due to its close relationship to the Newton method, and the second one the Fisher type as its preconditioner is closely related to the inverse of Fisher information matrix. Both preconditioners can be derived from one framework, and efficiently estimated on any matrix Lie groups designated by the user using natural or relative gradient descent minimizing certain preconditioner estimation criteria. Many existing preconditioners and methods, e.g., RMSProp, Adam, KFAC, equilibrated SGD, batch normalization, etc., are special cases of or closely related to either the Newton type or the Fisher type ones. Experimental results on relatively large scale machine learning problems are reported for performance study.

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

↑