Affiliation:
1. School of Information Science and Engineering, Lanzhou University, No. 222 South Tianshui Road, Lanzhou, Gansu 730000, P. R. China
Abstract
Compared to cross-entropy in deep learning, triplet loss is less affected by the biased label information and widely used in fine-grained visual tasks. Especially in person re-identification (re-ID), triplet loss is improved with batch-hard sampling which only selects the hardest samples during the training process to reduce invalid triplets involved in the loss computation. The hardest samples’ loss computation can provide a more intense gradient descent than raw samples. However, the batch-hard triplet loss discards multiple samples with important information, which can negatively impact feature learning. Besides, the hardest samples cause loss stuck problems frequently in training. In this work, we propose a balanced triplet loss for comprehensive feature learning and stable model convergence. The balanced triplet loss only mines the hardest negative samples of each category within a mini-batch. Compared with batch-hard triplet loss, it preserves the features of all the negative categories rather than one negative category with the hardest negative sample. It achieves a balance between triplet selection and information loss. The experiments show that our method can produce competitive results in re-ID tasks. In addition, we analyze the correlation between the intensity of data mining and the granularity of feature learning and further adapt the balanced triplet loss to general fine-grained image classification. The experiments prove the adapted balanced triplet loss also outperforms cross-entropy in multiple datasets of different scales.
Funder
the National Key R&D Program of China
Publisher
World Scientific Pub Co Pte Ltd
Subject
Artificial Intelligence,Computer Vision and Pattern Recognition,Software
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献