Title
Efficient Optimization Methods for Extreme Similarity Learning with Nonlinear Embeddings
Abstract
ABSTRACTWe study the problem of learning similarity by using nonlinear embedding models (e.g., neural networks) from all possible pairs. This problem is well-known for its difficulty of training with the extreme number of pairs. For the special case of using linear embeddings, many studies have addressed this issue of handling all pairs by considering certain loss functions and developing efficient optimization algorithms. This paper aims to extend results for general nonlinear embeddings. First, we finish detailed derivations and provide clean formulations for efficiently calculating some building blocks of optimization algorithms such as function, gradient evaluation, and Hessian-vector product. The result enables the use of many optimization methods for extreme similarity learning with nonlinear embeddings. Second, we study some optimization methods in detail. Due to the use of nonlinear embeddings, implementation issues different from linear cases are addressed. In the end, some methods are shown to be highly efficient for extreme similarity learning with nonlinear embeddings.
Year
DOI
Venue
2021
10.1145/3447548.3467363
Knowledge Discovery and Data Mining
Keywords
DocType
Citations 
Similarity learning, Representation learning, Non-convex optimization, Newton methods, Neural networks
Conference
0
PageRank 
References 
Authors
0.34
0
4
Name
Order
Citations
PageRank
Bo-Wen Yuan141.73
Yu-Sheng Li200.34
Pengrui Quan300.34
Jen-Chih Lin4248.22