Improved Algorithms for Neural Active Learning
Yikun Ban, Yuheng Zhang, Hanghang Tong, Arindam Banerjee, Jingrui He
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers8
- Deep Active Learning by Leveraging Training DynamicsHaonan Wang, Wei Huang, Ziwei Wu, Hanghang Tong et al.NeurIPS 2022 · 49 citations
- Streaming Active Learning with Deep Neural NetworksAkanksha Saran, Safoora Yousefi, Akshay Krishnamurthy, John Langford et al.ICML 2023 · 26 citations
- PageRank Bandits for Link PredictionYikun Ban, Jiaru Zou, Zihao Li, Yunzhe Qi et al.NeurIPS 2024 · 20 citations
- Neural Active Learning Beyond BanditsYikun Ban, Ishika Agarwal, Ziwei Wu, Yada Zhu et al.ICLR 2024 · 14 citations
- Adaptive Batch-Wise Sample Scheduling for Direct Preference OptimizationZixuan Huang, Yikun Ban, Lean Fu, Xiaojie Li et al.NeurIPS 2025 · 14 citations
Builds on18
- Deep Batch Active Learning by Diverse, Uncertain Gradient Lower BoundsJordan T. Ash, Chicheng Zhang, Akshay Krishnamurthy, John Langford et al.ICLR 2020 · 974 citations
- Neural Contextual Bandits with UCB-based ExplorationDongruo Zhou, Lihong Li, Quanquan GuICML 2020 · 329 citations
- Batch Active Learning at ScaleGui Citovsky, Giulia DeSalvo, Claudio Gentile, Lazaros Karydas et al.NeurIPS 2021 · 220 citations
- Confidence-Aware Learning for Deep Neural NetworksJooyoung Moon, Jihyo Kim, Younghak Shin, Sangheum HwangICML 2020 · 184 citations
- Neural Thompson SamplingWeitong Zhang, Dongruo Zhou, Lihong Li, Quanquan GuICLR 2021 · 152 citations
Related papers
- Neural Active Learning with Performance GuaranteesZhilei Wang, Pranjal Awasthi, Christoph Dann, Ayush Sekhari et al.NeurIPS 2021 · 26 citations
- Gone Fishing: Neural Active Learning with Fisher EmbeddingsJordan T. Ash, Surbhi Goel, Akshay Krishnamurthy, Sham M. KakadeNeurIPS 2021 · 124 citations
- Provably Neural Active Learning Succeeds via Prioritizing Perplexing SamplesDake Bu, Wei Huang, Taiji Suzuki, Ji Cheng et al.ICML 2024 · 5 citations
- Online Active Learning with Surrogate Loss FunctionsGiulia DeSalvo, Claudio Gentile, Tobias Sommer ThuneNeurIPS 2021 · 9 citations
- Active Learning with Neural Networks: Insights from Nonparametric StatisticsYinglun Zhu, Robert NowakNeurIPS 2022 · 15 citations
