Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Improved Algorithms for Neural Active Learning

About

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 $k$-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(\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.

Yikun Ban, Yuheng Zhang, Hanghang Tong, Arindam Banerjee, Jingrui He• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationMNIST (test)
Accuracy97.95
882
ClassificationCIFAR10 (test)
Accuracy91.8
266
ClassificationFashion (test)
Accuracy99.35
51
ClassificationLETTER (test)
Accuracy86.05
33
ClassificationPhishing (test)
Accuracy96.02
24
ClassificationIJCNN (test)
Accuracy98.75
24
Active Learning ClassificationPhishing
Total Regret689
6
Active Learning ClassificationIJCNN
Total Regret469
6
Active Learning ClassificationLetter
Total Regret2.57e+3
6
Active Learning ClassificationFashion
Total Regret118
6
Showing 10 of 12 rows

Other info

Code

Follow for update