Semi-Supervised Learning via Triplet Network Based Active Learning (Student Abstract)

Authors

  • Divyanshu Sundriyal IIIT Delhi
  • Soumyadeep Ghosh IIIT Delhi
  • Mayank Vatsa IIT Jodhpur
  • Richa Singh IIT Jodhpur

Keywords:

Active Learning, Classification, Computer Vision

Abstract

In recent years, deep learning models have pushed state-of-the-art accuracies for several machine learning tasks. However, such models require a large amount of (supervised) data for training. While unlabelled data is available in abundance, manually labeling them is very costly. Active learning techniques helps in utilizing unlabelled data which may result in an improved classification model. In this research, we present an active learning algorithm which can help in increasing performance of deep learning models by using large amount of unlabelled data. A novel active learning algorithm, Triplet AL is proposed which uses a triplet network to select samples from an unlabelled data set. Previous active learning methods rely on classification model's final prediction scores as a measure of confidence for an unlabelled sample. We propose a more reliable confidence measure, termed as Top-Two-Margin which is given by the Triplet Network. The proposed algorithm shows improved performance compared to other active learning approaches.

Downloads

Published

2021-05-18

How to Cite

Sundriyal, D., Ghosh, S., Vatsa, M., & Singh, R. (2021). Semi-Supervised Learning via Triplet Network Based Active Learning (Student Abstract). Proceedings of the AAAI Conference on Artificial Intelligence, 35(18), 15903-15904. Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/17948

Issue

Section

AAAI Student Abstract and Poster Program