Learning to Optimize Decision Boundary for One-Shot One-Class Image Recognition

Linkai Chen,Hongjin Zhu,Congzhe You,Honghui Fan,Yaduan Ruan
DOI: https://doi.org/10.21203/rs.3.rs-1848299/v1
2022-01-01
Abstract:Abstract In this paper, we study One-Shot One-Class classification (OSOC), where the aim is to identify the objects of a specific class amongst all objects by using only one single sample of that class as training data. The key challenge of this task is how to learn a robust decision boundary to distinguish a class in such an extreme case. Towards this goal, we propose a novel meta-learning strategy to meta learn a mapping function from the exemplar image of a target class to the categorical classifier as well as a reference score for this class. The classifier is applied to a testing sample to predict the classification score, which will be compared against the reference score to decide whether this sample belongs to the target class or not. To realize this idea, in the meta training phase, we propose a ranking loss that employs the reference score as the threshold to distinguish samples both from the target class and other non-target classes. On top of this basic loss function, we develop another two terms: a fidelity term to ensure the learned classifier has high fidelity to the exemplar and a learnable slack term to adaptively adjust the reference score on the fly. These two terms compete against each other and consequently seek a balance to optimize the decision boundary. Comprehensive experimental results on five OSOC datasets demonstrate the proposed method can obviously outperform state-of-the-art
What problem does this paper attempt to address?