vix.ing · top · new · best · stats

Tractable structured natural gradient descent using local parameterizations

2021/02/15 by Lin Wu, Wu Lin, Lin, Wu +6 · 4 citations
Computer Science · Mathematics · #Advanced Image and Video Retrieval Techniques #Advanced Vision and Imaging #FOS: Computer and information sciences #Face and Expression Recognition #Machine Learning (cs.LG) #Machine Learning (stat.ML) #cs.LG #stat.ML

paper · pdf · doi:10.48550/arxiv.2102.07405

An extended version of the ICML 2021 paper. Note: A workshop (short) paper with a focus on optimization tasks can be found at arXiv:2107.10884

openalex publication_date 2021/02/15 · openalex created_date 2021/03/01 · arxiv created 2022/01/17 · arxiv updated 2022/01/19 · openalex updated_date 2026/07/28

Abstract

Natural-gradient descent (NGD) on structured parameter spaces (e.g., low-rank covariances) is computationally challenging due to difficult Fisher-matrix computations. We address this issue by using local-parameter coordinates to obtain a flexible and efficient NGD method that works well for a wide-variety of structured parameterizations. We show four applications where our method (1) generalizes the exponential natural evolutionary strategy, (2) recovers existing Newton-like algorithms, (3) yields new structured second-order algorithms via matrix groups, and (4) gives new algorithms to learn covariances of Gaussian and Wishart-based distributions. We show results on a range of problems from deep learning, variational inference, and evolution strategies. Our work opens a new direction for scalable structured geometric methods.

Citations

Cited by

Related