Robustness and Generalization for Metric Learning

Aurélien Bellet (1) , Amaury Habrard (2)

1 USC - University of Southern California
2 LHC - Laboratoire Hubert Curien

Metric learning has attracted a lot of interest over the last decade, but the generalization ability of such methods has not been thoroughly studied. In this paper, we introduce an adaptation of the notion of algorithmic robustness (previously introduced by Xu and Mannor) that can be used to derive generalization bounds for metric learning. We further show that a weak notion of robustness is in fact a necessary and sufficient condition for a metric learning algorithm to generalize. To illustrate the applicability of the proposed framework, we derive generalization results for a large family of existing metric learning algorithms, including some sparse formulations that are not covered by previous results.

Domaines: Apprentissage [cs.LG]

Aurélien Bellet, Amaury Habrard. Robustness and Generalization for Metric Learning. Neurocomputing, 2015, pp.16. ⟨10.1016/j.neucom.2014.09.044⟩. ⟨hal-01075370⟩