Interpreting pretext tasks for active learning: a reinforcement learning approach
As the amount of labeled data increases, the performance of deep neural networks tends to improve. However, annotating a large volume of data can be expensive. Active learning addresses this challenge by selectively annotating unlabeled data. There have been recent attempts to incorporate self-supervised learning into active learning, but there are issues in utilizing the results of self-supervised learning, i.e., it is uncertain how these should be interpreted in the context of active learning. To address this issue, we propose a multi-armed bandit approach to handle the information provided by self-supervised learning in active learning. Furthermore, we devise a data sampling process so that reinforcement learning can be effectively performed. We evaluate the proposed method on various image classification benchmarks, including CIFAR-10, CIFAR-100, Caltech-101, SVHN, and ImageNet, where the proposed method significantly improves previous approaches.
Authors
- Minsik Lee (ORCID: https://orcid.org/0000-0003-4941-4311)
- Dongjoo Kim (ORCID: https://orcid.org/0000-0003-0557-1366)
Institutions
- Hanyang University (KR)
- Anyang University (KR)
Publication Details
- Journal
- Scientific Reports
- Published
- 2024-10-28
- DOI
- https://doi.org/10.1038/s41598-024-76864-2
- Primary Topic
- Robot Manipulation and Learning
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Hanyang University
- Ministry of Science and ICT, South Korea
- Institute for Information and Communications Technology Promotion
- Division of Human Resource Development