English

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.

Keywords

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

R2 v1 2026-06-22T14:57:01.441Z