English

Improved Algorithms for Neural Active Learning

Machine Learning 2023-01-18 v3 Machine 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 kk-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 O(logT)O(\log T) 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.

Keywords

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