arrow
Return

Riemannian Adaptive Optimization Algorithm and Its Application to Natural Language Processing

delete2022-08-01
delete15
delete
OA
AI
H
Hiroyuki Sakai
H
Hideaki Iiduka *
DOI:10.1109/TCYB.2021.3049845delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Y This article proposes a Riemannian adaptive optimization algorithm to optimize the parameters of deep neural networks. The algorithm is an extension of both AMSGrad in Euclidean space and RAMSGrad on a Riemannian manifold. The algorithm helps to resolve two issues affecting RAMSGrad. The first is that it can solve the Riemannian stochastic optimization problem directly, in contrast to RAMSGrad which only achieves a low regret. The other is that it can use constant learning rates, which makes it implementable in practice. Additionally, we apply the proposed algorithm to Poincare embeddings that embed the transitive closure of the WordNet nouns into the Poincare ball model of hyperbolic space. Numerical experiments show that regardless of the initial value of the learning rate, our algorithm stably converges to the optimal solution and converges faster than the existing algorithms.
Keywords:
Hyperbolic space
natural language processing
Poincare embeddings
RAdaGrad
RAdam
RAMSGrad
Riemannian adaptive optimization algorithm
Riemannian optimization
Riemannian stochastic gradient descent (RSGD)
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Transactions on Cybernetics cover
IEEE Transactions on Cybernetics
IF:
10.5
Papers:
1.1W
Citations:
5.0W

Organization

M
Meiji University
Scholars:
1.8K
Papers: 1.5K
Citations: 1.1K