Geometric Mean Metric Learning
Machine Learning
2016-07-19 v1 Machine Learning
Abstract
We revisit the task of learning a Euclidean metric from data. We approach this problem from first principles and formulate it as a surprisingly simple optimization problem. Indeed, our formulation even admits a closed form solution. This solution possesses several very attractive properties: (i) an innate geometric appeal through the Riemannian geometry of positive definite matrices; (ii) ease of interpretability; and (iii) computational speed several orders of magnitude faster than the widely used LMNN and ITML methods. Furthermore, on standard benchmark datasets, our closed-form solution consistently attains higher classification accuracy.
Cite
@article{arxiv.1607.05002,
title = {Geometric Mean Metric Learning},
author = {Pourya Habib Zadeh and Reshad Hosseini and Suvrit Sra},
journal= {arXiv preprint arXiv:1607.05002},
year = {2016}
}
Comments
7 pages, 4 figures