Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD)
Distance metric learning (DML) is an important task that has found applications in many domains. The high computational cost of DML arises from the large number of variables to be determined and the constraint that a distance metric has to be a positive semi-definite (PSD) matrix. Although stochasti...
Main Authors: | , , , , |
---|---|
Format: | Text |
Language: | unknown |
Published: |
2013
|
Subjects: | |
Online Access: | http://arxiv.org/abs/1304.1192 |
id |
ftarxivpreprints:oai:arXiv.org:1304.1192 |
---|---|
record_format |
openpolar |
spelling |
ftarxivpreprints:oai:arXiv.org:1304.1192 2023-09-05T13:19:04+02:00 Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) Qian, Qi Jin, Rong Yi, Jinfeng Zhang, Lijun Zhu, Shenghuo 2013-04-03 http://arxiv.org/abs/1304.1192 unknown http://arxiv.org/abs/1304.1192 Computer Science - Machine Learning text 2013 ftarxivpreprints 2023-08-16T12:59:39Z Distance metric learning (DML) is an important task that has found applications in many domains. The high computational cost of DML arises from the large number of variables to be determined and the constraint that a distance metric has to be a positive semi-definite (PSD) matrix. Although stochastic gradient descent (SGD) has been successfully applied to improve the efficiency of DML, it can still be computationally expensive because in order to ensure that the solution is a PSD matrix, it has to, at every iteration, project the updated distance metric onto the PSD cone, an expensive operation. We address this challenge by developing two strategies within SGD, i.e. mini-batch and adaptive sampling, to effectively reduce the number of updates (i.e., projections onto the PSD cone) in SGD. We also develop hybrid approaches that combine the strength of adaptive sampling with that of mini-batch online learning techniques to further improve the computational efficiency of SGD for DML. We prove the theoretical guarantees for both adaptive sampling and mini-batch based approaches for DML. We also conduct an extensive empirical study to verify the effectiveness of the proposed algorithms for DML. Text DML ArXiv.org (Cornell University Library) |
institution |
Open Polar |
collection |
ArXiv.org (Cornell University Library) |
op_collection_id |
ftarxivpreprints |
language |
unknown |
topic |
Computer Science - Machine Learning |
spellingShingle |
Computer Science - Machine Learning Qian, Qi Jin, Rong Yi, Jinfeng Zhang, Lijun Zhu, Shenghuo Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
topic_facet |
Computer Science - Machine Learning |
description |
Distance metric learning (DML) is an important task that has found applications in many domains. The high computational cost of DML arises from the large number of variables to be determined and the constraint that a distance metric has to be a positive semi-definite (PSD) matrix. Although stochastic gradient descent (SGD) has been successfully applied to improve the efficiency of DML, it can still be computationally expensive because in order to ensure that the solution is a PSD matrix, it has to, at every iteration, project the updated distance metric onto the PSD cone, an expensive operation. We address this challenge by developing two strategies within SGD, i.e. mini-batch and adaptive sampling, to effectively reduce the number of updates (i.e., projections onto the PSD cone) in SGD. We also develop hybrid approaches that combine the strength of adaptive sampling with that of mini-batch online learning techniques to further improve the computational efficiency of SGD for DML. We prove the theoretical guarantees for both adaptive sampling and mini-batch based approaches for DML. We also conduct an extensive empirical study to verify the effectiveness of the proposed algorithms for DML. |
format |
Text |
author |
Qian, Qi Jin, Rong Yi, Jinfeng Zhang, Lijun Zhu, Shenghuo |
author_facet |
Qian, Qi Jin, Rong Yi, Jinfeng Zhang, Lijun Zhu, Shenghuo |
author_sort |
Qian, Qi |
title |
Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
title_short |
Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
title_full |
Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
title_fullStr |
Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
title_full_unstemmed |
Efficient Distance Metric Learning by Adaptive Sampling and Mini-Batch Stochastic Gradient Descent (SGD) |
title_sort |
efficient distance metric learning by adaptive sampling and mini-batch stochastic gradient descent (sgd) |
publishDate |
2013 |
url |
http://arxiv.org/abs/1304.1192 |
genre |
DML |
genre_facet |
DML |
op_relation |
http://arxiv.org/abs/1304.1192 |
_version_ |
1776199881000484864 |