Analytic natural gradient updates for Cholesky factor in Gaussian variational approximation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autor principal: Tan, Linda S. L.
Formato: Preprint
Publicado: 2021
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910812348612608
author Tan, Linda S. L.
author_facet Tan, Linda S. L.
contents Natural gradients can improve convergence in stochastic variational inference significantly but inverting the Fisher information matrix is daunting in high dimensions. Moreover, in Gaussian variational approximation, natural gradient updates of the precision matrix do not ensure positive definiteness. To tackle this issue, we derive analytic natural gradient updates of the Cholesky factor of the covariance or precision matrix, and consider sparsity constraints representing different posterior correlation structures. Stochastic normalized natural gradient ascent with momentum is proposed for implementation in generalized linear mixed models and deep neural networks.
format Preprint
id arxiv_https___arxiv_org_abs_2109_00375
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Analytic natural gradient updates for Cholesky factor in Gaussian variational approximation
Tan, Linda S. L.
Computation
Natural gradients can improve convergence in stochastic variational inference significantly but inverting the Fisher information matrix is daunting in high dimensions. Moreover, in Gaussian variational approximation, natural gradient updates of the precision matrix do not ensure positive definiteness. To tackle this issue, we derive analytic natural gradient updates of the Cholesky factor of the covariance or precision matrix, and consider sparsity constraints representing different posterior correlation structures. Stochastic normalized natural gradient ascent with momentum is proposed for implementation in generalized linear mixed models and deep neural networks.
title Analytic natural gradient updates for Cholesky factor in Gaussian variational approximation
topic Computation
url https://arxiv.org/abs/2109.00375