Siamese Network Representation for Active Learning
Jianing Li, Yuan Du, Li Du
-
SPS
IEEE Members: $11.00
Non-members: $15.00
Active learning is an important part of machine learning aiming to reduce the amount of labeled data by selecting the most informative data to be annotated. Most of the previous proposed active learning methods are based on uncertainty obtained by learning models while ignoring relationships within the data itself. We propose an efficient similarity-based active learning method using siamese convolutional neural networks. Pairs of image data are sent into the siamese network and similarity between them is computed on their output features. We evaluate our method on image classification, and validate the method on CIFAR10/100 and Caltech101 dataset. Our method outperforms at most 3.17% accuracy than Bayesian-based method and 6.31% than random sample. Additionally, We propose a hierarchical clustering method for pool-based sampling strategy, which boosts the representation stage of our method. We also conduct ablation study to fully explore the efficiency of our method.