Timezone: »
Low-rank metric learning aims to learn better discrimination of data subject to low-rank constraints. It keeps the intrinsic low-rank structure of datasets and reduces the time cost and memory usage in metric learning. However, it is still a challenge for current methods to handle datasets with both high dimensions and large numbers of samples. To address this issue, we present a novel fast low-rank metric learning (FLRML) method. FLRML casts the low-rank metric learning problem into an unconstrained optimization on the Stiefel manifold, which can be efficiently solved by searching along the descent curves of the manifold. FLRML significantly reduces the complexity and memory usage in optimization, which makes the method scalable to both high dimensions and large numbers of samples. Furthermore, we introduce a mini-batch version of FLRML to make the method scalable to larger datasets which are hard to be loaded and decomposed in limited memory. The outperforming experimental results show that our method is with high accuracy and much faster than the state-of-the-art methods under several benchmarks with large numbers of high-dimensional data. Code has been made available at https://github.com/highan911/FLRML.
Author Information
Han Liu (Tsinghua University)
Zhizhong Han (University of Maryland, College Park)
Yu-Shen Liu (Tsinghua University)
Ming Gu (Tsinghua University)
More from the Same Authors
-
2023 Poster: NeuralGF: Unsupervised Point Normal Estimation by Learning Neural Gradient Function »
Qing Li · Huifang Feng · Kanle Shi · Yue Gao · Yi Fang · Yu-Shen Liu · Zhizhong Han -
2023 Poster: Differentiable Registration of Images and LiDAR Point Clouds with VoxelPoint-to-Pixel Matching »
Junsheng Zhou · Baorui Ma · Wenyuan Zhang · Yi Fang · Yu-Shen Liu · Zhizhong Han -
2022 Poster: HSurf-Net: Normal Estimation for 3D Point Clouds by Learning Hyper Surfaces »
Qing Li · Yu-Shen Liu · Jin-San Cheng · Cheng Wang · Yi Fang · Zhizhong Han -
2022 Poster: Learning Consistency-Aware Unsigned Distance Functions Progressively from Raw Point Clouds »
Junsheng Zhou · Baorui Ma · Yu-Shen Liu · Yi Fang · Zhizhong Han