CLE_LMNN: A novel framework of LMNN based on clustering labeled examples

摘要

Distance metric learning is the task that aims to automate this process of learning task-specific distance functions in a supervised manner. In this paper, we study how to learn a Mahalanobis distance metric that can improve nearest neighbor classification. Our paper makes two contributions. First, we propose a novel framework named CLE_LMNN for Mahalanobis distance learning. CLE_LMNN builds on a recently proposed framework known as large margin nearest neighbor (LMNN) classification. Compared with LMNN, CLE_LMNN learns a Mahalanobis distance in a fine-grained way by first partitioning the labeled examples into subsets. As shown by our experiments, this fine-grained learning way is inclined to obtain a Mahalanobis distance more suitable for classification. Second, we present a novel algorithm named CLE for clustering labeled examples. Different from traditional unsupervised clustering algorithms, CLE fully employ the class information of labeled examples to effectively partition the set of examples with the same class into different subsets. To evaluate our proposed framework, we conduct extensive experiments on three real datasets. The experimental results show the effectiveness of CLE_LMNN when applied to classification.