786 citations · 1.3k across the 45 of their papers we have counts for
11 papers · 1 filter
Rieoptax: Riemannian Optimization in JAX
Saiteja Utpala, Andi Han, Pratik Jawanpuria +1
We present Rieoptax, an open source Python library for Riemannian optimization in JAX. We show that many differential geometric primitives, such as Riemannian exponential and logar…
Riemannian accelerated gradient methods via extrapolation
Andi Han, Bamdev Mishra, Pratik Jawanpuria +1
In this paper, we propose a simple acceleration scheme for Riemannian gradient methods by extrapolating iterates on manifolds. We show when the iterates are generated from Riemanni…
Differentially private Riemannian optimization
Andi Han, Bamdev Mishra, Pratik Jawanpuria +1
In this paper, we study the differentially private empirical risk minimization problem where the parameter is constrained to a Riemannian manifold. We introduce a framework of diff…
Riemannian Hamiltonian methods for min-max optimization on manifolds
Andi Han, Bamdev Mishra, Pratik Jawanpuria +2
In this paper, we study min-max optimization problems on Riemannian manifolds. We introduce a Riemannian Hamiltonian function, minimization of which serves as a proxy for solving t…
On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein Geometry
Andi Han, Bamdev Mishra, Pratik Jawanpuria +1
In this paper, we comparatively analyze the Bures-Wasserstein (BW) geometry with the popular Affine-Invariant (AI) geometry for Riemannian optimization on the symmetric positive de…
Riemannian optimization on the simplex of positive definite matrices
Bamdev Mishra, Hiroyuki Kasai, Pratik Jawanpuria
In this work, we generalize the probability simplex constraint to matrices, i.e., , where …