Nl02znfTCT@OpenReview

Total: 1

#1 Acceleration via silver step-size on Riemannian manifolds with applications to Wasserstein space [PDF] [Copy] [Kimi] [REL]

Authors: Jiyoung Park, Abhishek Roy, Jonathan W. Siegel, Anirban Bhattacharya

There is extensive literature on accelerating first-order optimization methods in an Euclidean setting. Under which conditions such acceleration is feasible in Riemannian optimization problems is an active area of research. Motivated by the recent success of silver stepsize methods in the Euclidean setting, we undertake a study of such algorithms in the Riemannian setting. We provide the new class of algorithms determined by the choice of vector transport that allows the silver stepsize acceleration on Riemannian manifolds for the function classes associated with the corresponding vector transport. As a core application, we show that our algorithm recovers the standard Wasserstein gradient descent on the 2-Wasserstein space and, as a result, provides the first provable accelerated gradient method for potential functional optimization problems in the Wasserstein space. In addition, we validate the numerical strength of the algorithm for standard benchmark tasks on the space of symmetric positive definite matrices.

Subject: NeurIPS.2025 - Poster