23 August 2024 Additive cosine margin for unsupervised softmax embedding
Author Affiliations +
Abstract

Unsupervised embedding learning aims to learn highly discriminative features of images without using class labels. Existing instance-wise softmax embedding methods treat each instance as a distinct class and explore the underlying instance-to-instance visual similarity relationships. However, overfitting the instance features leads to insufficient discriminability and poor generalizability of networks. To tackle this issue, we introduce an instance-wise softmax embedding with cosine margin (SEwCM), which for the first time adds margin in the unsupervised instance softmax classification function from the cosine perspective. The cosine margin is used to separate the classification decision boundaries between instances. SEwCM explicitly optimizes the feature mapping of networks by maximizing the cosine similarity between instances, thus learning a highly discriminative model. Exhaustive experiments on three fine-grained image datasets demonstrate the effectiveness of our proposed method over existing methods.

© 2024 SPIE and IS&T
Dan Wang, Jianwei Yang, and Cailing Wang "Additive cosine margin for unsupervised softmax embedding," Journal of Electronic Imaging 33(4), 040501 (23 August 2024). https://doi.org/10.1117/1.JEI.33.4.040501
Received: 26 March 2024; Accepted: 5 August 2024; Published: 23 August 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Feature extraction

Visualization

Education and training

Image retrieval

Machine learning

Mathematical optimization

Curium

Back to Top