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

Handling the Positive-Definite Constraint in the Bayesian Learning Rule

University of British Columbia(英属哥伦比亚大学);Alberta Machine Intelligence Institute(阿尔伯塔机器智能研究所);RIKEN Center for Advanced Intelligence Project(理化学研究所先进智能项目中心)

Wu Lin, Mark Schmidt, Mohammad Emtiyaz Khan

英文摘要:

The Bayesian learning rule is a natural-gradient variational inference method, which not only contains many existing learning algorithms as special cases but also enables the design of new algorithms. Unfortunately, when variational parameters lie in an open constraint set, the rule may not satisfy the constraint and requires line-searches which could slow down the algorithm. In this work, we address this issue for positive-definite constraints by proposing an improved rule that naturally handles the constraints. Our modification is obtained by using Riemannian gradient methods, and is valid when the approximation attains a \emph{block-coordinate natural parameterization} (e.g., Gaussian distributions and their mixtures). We propose a principled way to derive Riemannian gradients and retractions from scratch. Our method outperforms existing methods without any significant increase in computation. Our work makes it easier to apply the rule in the presence of positive-definite constraints in parameter spaces.

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

↑