Improved Algorithms for Neural Active Learning
Abstract
We improve the theoretical and empirical performance of neural-network(NN)-based active learning algorithms for the non-parametric streaming setting. In particular, we introduce two regret metrics by minimizing the population loss that are more suitable in active learning than the one used in state-of-the-art (SOTA) related work. Then, the proposed algorithm leverages the powerful representation of NNs for both exploitation and exploration, has the query decision-maker tailored for -class classification problems with the performance guarantee, utilizes the full feedback, and updates parameters in a more practical and efficient manner. These careful designs lead to an instance-dependent regret upper bound, roughly improving by a multiplicative factor and removing the curse of input dimensionality. Furthermore, we show that the algorithm can achieve the same performance as the Bayes-optimal classifier in the long run under the hard-margin setting in classification problems. In the end, we use extensive experiments to evaluate the proposed algorithm and SOTA baselines, to show the improved empirical performance.
Cite
@article{arxiv.2210.00423,
title = {Improved Algorithms for Neural Active Learning},
author = {Yikun Ban and Yuheng Zhang and Hanghang Tong and Arindam Banerjee and Jingrui He},
journal= {arXiv preprint arXiv:2210.00423},
year = {2023}
}
Comments
Published on NeurIPS 2022