Parameter free large margin nearest neighbor for distance metric learning

Kun Song, Feiping Nie, Junwei Han, Xuelong Li

科研成果: 会议稿件论文同行评审

36 引用 (Scopus)

摘要

We introduce a novel supervised metric learning algorithm named parameter free large margin nearest neighbor (PFLMNN) which can be seen as an improvement of the classical large margin nearest neighbor (LMNN) algorithm. The contributions of our work consist of two aspects. First, our method discards the cost term which shrinks the distances between inquiry input and its k target neighbors (the k nearest neighbors with same labels as inquiry input) in LMNN, and only focuses on improving the action to push the imposters (the samples with different labels form the inquiry input) apart out of the neighborhood of inquiry. As a result, our method does not have the parameter needed to tune on the validating set, which makes it more convenient to use. Second, by leveraging the geometry information of the imposters, we construct a novel cost function to penalize the small distances between each inquiry and its imposters. Different from LMNN considering every imposter located in the neighborhood of each inquiry, our method only takes care of the nearest imposters. Because when the nearest imposter is pushed out of the neighborhood of its inquiry, other imposters would be all out. In this way, the constraints in our model are much less than that of LMNN, which makes our method much easier to find the optimal distance metric. Consequently, our method not only learns a better distance metric than LMNN, but also runs faster than LMNN. Extensive experiments on different data sets with various sizes and difficulties are conducted, and the results have shown that, compared with LMNN, PFLMNN achieves better classification results.

源语言英语
2555-2561
页数7
出版状态已出版 - 2017
活动31st AAAI Conference on Artificial Intelligence, AAAI 2017 - San Francisco, 美国
期限: 4 2月 201710 2月 2017

会议

会议31st AAAI Conference on Artificial Intelligence, AAAI 2017
国家/地区美国
San Francisco
时期4/02/1710/02/17

指纹

探究 'Parameter free large margin nearest neighbor for distance metric learning' 的科研主题。它们共同构成独一无二的指纹。

引用此