Optimization without Retraction on the Random Generalized Stiefel Manifold

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Vary, Simon, Ablin, Pierre, Gao, Bin, Absil, P. -A.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917067933876224
author Vary, Simon
Ablin, Pierre
Gao, Bin
Absil, P. -A.
author_facet Vary, Simon
Ablin, Pierre
Gao, Bin
Absil, P. -A.
contents Optimization over the set of matrices $X$ that satisfy $X^\top B X = I_p$, referred to as the generalized Stiefel manifold, appears in many applications involving sampled covariance matrices such as the canonical correlation analysis (CCA), independent component analysis (ICA), and the generalized eigenvalue problem (GEVP). Solving these problems is typically done by iterative methods that require a fully formed $B$. We propose a cheap stochastic iterative method that solves the optimization problem while having access only to random estimates of $B$. Our method does not enforce the constraint in every iteration; instead, it produces iterations that converge to critical points on the generalized Stiefel manifold defined in expectation. The method has lower per-iteration cost, requires only matrix multiplications, and has the same convergence rates as its Riemannian optimization counterparts that require the full matrix $B$. Experiments demonstrate its effectiveness in various machine learning applications involving generalized orthogonality constraints, including CCA, ICA, and the GEVP.
format Preprint
id arxiv_https___arxiv_org_abs_2405_01702
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Optimization without Retraction on the Random Generalized Stiefel Manifold
Vary, Simon
Ablin, Pierre
Gao, Bin
Absil, P. -A.
Machine Learning
Optimization and Control
90C26, 90C15
Optimization over the set of matrices $X$ that satisfy $X^\top B X = I_p$, referred to as the generalized Stiefel manifold, appears in many applications involving sampled covariance matrices such as the canonical correlation analysis (CCA), independent component analysis (ICA), and the generalized eigenvalue problem (GEVP). Solving these problems is typically done by iterative methods that require a fully formed $B$. We propose a cheap stochastic iterative method that solves the optimization problem while having access only to random estimates of $B$. Our method does not enforce the constraint in every iteration; instead, it produces iterations that converge to critical points on the generalized Stiefel manifold defined in expectation. The method has lower per-iteration cost, requires only matrix multiplications, and has the same convergence rates as its Riemannian optimization counterparts that require the full matrix $B$. Experiments demonstrate its effectiveness in various machine learning applications involving generalized orthogonality constraints, including CCA, ICA, and the GEVP.
title Optimization without Retraction on the Random Generalized Stiefel Manifold
topic Machine Learning
Optimization and Control
90C26, 90C15
url https://arxiv.org/abs/2405.01702