Learning With Batch-wise Optimal Transport Loss For 3D Shape Recognition

Abstract

Deep metric learning is essential for visual recognition. The widely used pair-wise (or triplet) based loss objectives cannot make full use of semantical information in training samples or give enough attention to those hard samples during optimization. Thus, they often suffer from a slow convergence rate and inferior performance. In this paper, we show how to learn an importance-driven distance metric via optimal transport programming from batches of samples. It can automatically emphasize hard examples and lead to significant improvements in convergence. We propose a new batch-wise optimal transport loss and combine it in an end-to-end deep metric learning manner. We use it to learn the distance metric and deep feature representation jointly for recognition. Empirical results on visual retrieval and classification tasks with six benchmark datasets, i.e., MNIST, CIFAR10, SHREC13, SHREC14, ModelNet10, and ModelNet40, demonstrate the superiority of the proposed method. It can accelerate

Stats

• citations24
• S2 citations—
• github stars0
• HF likes0
• heat score10.48
• arxiv keyxu2019learning

Related papers

• Triplet-center Loss For Multi-view 3D Object Retrieval (2018)18.70
• Angular Triplet-center Loss For Multi-view 3D Shape Retrieval (2018)12.33
• Directional Statistics-based Deep Metric Learning For Image Classification And Retrieval (2018)13.05
• Constellation Loss: Improving The Efficiency Of Deep Metric Learning Loss Functions For Optimal Embedding (2019)0.00
• Shadow Loss: Memory-linear Deep Metric Learning For Efficient Training (2023)0.00
• Three Things To Know About Deep Metric Learning (2024)0.00
• Deeppoint3d: Learning Discriminative Local Descriptors Using Deep Metric Learning On 3D Point Clouds (2019)9.59
• Multi-feature Distance Metric Learning For Non-rigid 3D Shape Retrieval (2019)5.84