Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Bounds
Not on the map. Why?
Abstract (arXiv)
We design a new algorithm for batch active learning with deep neural network models. Our algorithm, Batch Active learning by Diverse Gradient Embeddings (BADGE), samples groups of points that are disparate and high-magnitude when represented in a hallucinated gradient space, a strategy designed to incorporate both predictive uncertainty and sample diversity into every selected batch. Crucially, BADGE trades off between diversity and uncertainty without requiring any hand-tuned hyperparameters. We show that while other approaches sometimes succeed for particular batch sizes or architectures, BADGE consistently performs as well or better, making it a versatile option for practical active learning problems.
Key facts. Cited by 602 papers in the map (data through 7 Oct 2026). First posted on arXiv 9 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 68; 2022: 101; 2023: 134; 2024: 112; 2025: 93; 2026: 93.
602 papers.
Sort:
By area:
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
- 2024ShieldGemma: Generative AI Content Moderation Based on Gemmacited by 224
- 2021Active Learning by Acquiring Contrastive Examplescited by 154
- 2022Active Learning on a Budget: Opposite Strategies Suit High and Low Budgetscited by 131
- 2021Batch Active Learning at Scalecited by 127
- 2022Coverage-centric Coreset Selection for High Pruning Ratescited by 115
- 2021Gone Fishing: Neural Active Learning with Fisher Embeddingscited by 88
- 2022A Comparative Survey of Deep Active Learningcited by 88
- 2021SIMILAR: Submodular Information Measures Based Active Learning In Realistic Scenarioscited by 83
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2023LLMaAA: Making Large Language Models as Active Annotatorscited by 78
38 more
- 2024Data Shapley in One Training Runcited by 76
- 2021On Statistical Bias In Active Learning: How and When To Fix Itcited by 72
- 2021RETRIEVE: Coreset Selection for Efficient and Robust Semi-Supervised Learningcited by 72
- 2023D2 Pruning: Message Passing for Balancing Diversity and Difficulty in Data Pruningcited by 68
- 2023Data Pruning via Moving-one-Sample-outcited by 67
- 2022Active Learning by Feature Mixingcited by 63
- 2023FisherRF: Active View Selection and Uncertainty Quantification for Radiance Fields using Fisher Informationcited by 63
- 2022Active Learning Through a Covering Lenscited by 62
- 2023Prediction-Oriented Bayesian Active Learningcited by 60
- 2021Towards Fewer Annotations: Active Learning via Region Impurity and Prediction Uncertainty for Domain Adaptive Semantic Segmentationcited by 56
- 2021Active Learning for Domain Adaptation: An Energy-Based Approachcited by 53
- 2022Entropy-based Active Learning for Object Detection with Progressive Diversity Constraintcited by 53
- 2021ReDAL: Region-based and Diversity-aware Active Learning for Point Cloud Semantic Segmentationcited by 52
- 2021PRISM: A Rich Class of Parameterized Submodular Information Measures for Guided Subset Selectioncited by 50
- 2024Automatic Data Curation for Self-Supervised Learning: A Clustering-Based Approachcited by 48
- 2022A Framework and Benchmark for Deep Batch Active Learning for Regressioncited by 47
- 2024Active Statistical Inferencecited by 46
- 2021Reducing Label Effort: Self-Supervised meets Active Learningcited by 42
- 2022A Survey of Active Learning for Natural Language Processingcited by 40
- 2025Prompt Curriculum Learning for Efficient LLM Post-Trainingcited by 40
- 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoningcited by 38
- 2021Effective Evaluation of Deep Active Learning on Image Classification Taskscited by 37
- 2022Submodularity In Machine Learning and Artificial Intelligencecited by 37
- 2021Stochastic Batch Acquisition: A Simple Baseline for Deep Active Learningcited by 36
- 2021Deep Active Learning by Leveraging Training Dynamicscited by 32
- 2023A comprehensive survey on deep active learning in medical image analysiscited by 32
- 2021On the Importance of Effectively Adapting Pretrained Language Models for Active Learningcited by 31
- 2024Active Generalized Category Discoverycited by 31
- 2024PINNACLE: PINN Adaptive ColLocation and Experimental points selectioncited by 31
- 2022GALAXY: Graph-based Active Learning at the Extremecited by 29
- 2023Active learning for data streams: a surveycited by 29
- 2021Active Learning at the ImageNet Scalecited by 28
- 2022Unifying Approaches in Active Learning and Active Sampling via Fisher Information and Information-Theoretic Quantitiescited by 28
- 2022Plug and Play Active Learning for Object Detectioncited by 28
- 2024Diversity Measurement and Subset Selection for Instruction Tuning Datasetscited by 28
- 2021Data Shapley Valuation for Efficient Batch Active Learningcited by 27
- 2021TALISMAN: Targeted Active Learning for Object Detection with Rare Classes and Slices using Submodular Mutual Informationcited by 27
- 2023Exploring Active 3D Object Detection from a Generalization Perspectivecited by 26
50 most cited of 602 All 602 (JSON)
- 2026From internal representations to model improvement through prediction errors28 Sep 2026
- 2026TC-ADA: One-Shot Active Domain Adaptation for Semantic Segmentation27 Sep 2026
- 2026Feature Space Guidance for Breast Cancer Classification in DCE-MRI26 Sep 2026
- 2026MICRO: Multi-Fidelity Active Search for Severe Error Discovery22 Sep 2026
- 2026Clustering-Based Balanced Sampling and Allocation with Data Parallelism for High-Performance Fine-Tuning11 Sep 2026
- 2026DISEIL: Demonstration Distillation for Sample-Efficient Imitation Learning8 Sep 2026
- 2026RAIDAL: Redundancy-Aware Information Density Active Learning for CTC-Based Continuous Sign Language Recognition6 Sep 2026
- 2026Train Smarter, Not Harder: Switching Signal-Guided Training in Active Learning6 Sep 2026
- 2026AdaptNTK: Adaptive Uncertainty Quantification and Active Learning for Neural Network Potentials31 Aug 2026
- 2026ActiveAugment: Online Active Learning for Augmentation Selection in Deep Learning28 Aug 2026
- 2026FrameScope: Temporal Data Valuation for Stream Active Learning in Autonomous Vehicle Systems24 Aug 2026
- 2026Target-Aware Calibration Data Selection for Preserving Uncertainty in Quantized Language Models21 Aug 2026
38 more
- 2026Probing Association Instability with Track-State Perturbations for Clip-Level Active Learning in Query-Propagation Multi-Object Tracking18 Aug 2026
- 2026SUGFW+: An Uncertainty-guided Feature Weighting Framework for Cold Start Active Adaptation of SAM in Medical Image Segmentation17 Aug 2026
- 2026How Many Labels Are Enough? ALDA: Active Learning Deployment Advisor for Medical Image Classification4 Aug 2026
- 2026One Knob to Rule Them All: A Unified Optimal Transport View of Cold-Start Active Learning4 Aug 2026
- 2026Distance to Class Prototypes: Active Learning for Object Detection28 Jul 2026
- 2026Test-Time Coverage: Test-Conditioned Data Curation for Deployment-Aware Learning18 Jul 2026
- 2026Cover First, Disagree Softly: Rethinking Mismatch-First Active Learning for Frame-Level Audio Classification15 Jul 2026
- 2026Greedy Volume Maximization of Gradient Embeddings for Long-Tailed Frame-Level Bioacoustic Active Learning15 Jul 2026
- 2026Active Learning for Efficient Annotation of Surgical Videos with Weak Supervision14 Jul 2026
- 2026Online Data Selection Is Implicit Alignment8 Jul 2026
- 2026Beyond Random Sampling: Distribution-Aware Alignment for Semi-Supervised Medical Image Segmentation5 Jul 2026
- 2026Coresets Before Score Sets: Evaluation-Unsupervised Prompt Subset Selection for LLM Benchmarks2 Jul 2026
- 2026Group-invariant Coresets for Data-efficient Active Learning1 Jul 2026
- 2026Active Learning for Cascaded Object Detection: Balancing Coverage and Uncertainty in Table Extraction Pipelines1 Jul 2026
- 2026A Mechanism-Driven Theory of Phase Transitions in Active Learning30 Jun 2026
- 2026LLM-ACES: Closed-Loop Discovery of Dynamical Systems with LLM-Guided Adaptive Search23 Jun 2026
- 2026FairBED: A Bayesian Experimental Design Approach to Gathering Fairer Data22 Jun 2026
- 2026ConnectomeBench2: A Unified Benchmark for Automated Connectomic Proofreading19 Jun 2026
- 2026Uncertainty Quantification for Flow-Based Vision-Language-Action Models16 Jun 2026
- 2026Active Reference Acquisition in Few-Shot Font Generation15 Jun 2026
- 2026CAAL: Contextual Bandits based Online Hand-Craft Active Learning Strategy Selection6 Jun 2026
- 2026SALT: When More Rollouts Don't Help in Group-Based Policy Optimization and How to Make Them Matter4 Jun 2026
- 2026ALINC: Active Learning for Inductive Node Classification via Graph Sampling3 Jun 2026
- 2026XSSR: Cross-Domain Self-Supervised Representative Selection for Efficient Annotation in Medical Image Segmentation3 Jun 2026
- 2026Finding Needles in the Haystack: Transductive Active Labeling in Ecology2 Jun 2026
- 2026FACT: A Simple and Efficient Framework for Active Finetuning1 Jun 2026
- 2026Active Learning with Foundation Model Priors: Efficient Learning under Class Imbalance30 May 2026
- 2026MADS: Model-Aware Diverse Core Set Selection for Instruction Tuning29 May 2026
- 2026The Long-Term Effects of Data Selection in LLM Fine-Tuning28 May 2026
- 2026Label-Efficient Dataset Pruning via Semi-Supervised Pseudo-Labeling22 May 2026
- 2026Cumulative Meta-Learning from Active Learning Queries for Robustness to Spurious Correlations20 May 2026
- 2026A Human-in-the-Loop Framework for Efficient Prompt Selection in Microscopy Vision-Language Models19 May 2026
- 2026Interaction-Aware Influence Functions for Group Attribution15 May 2026
- 2026Are Candidate Models Really Needed for Active Learning?14 May 2026
- 2026AcquisitionSynthesis: Targeted Data Generation using Acquisition Functions13 May 2026
- 2026GRACE: Gradient-aligned Reasoning Data Curation for Efficient Post-training13 May 2026
- 2026PoseCompass: Intelligent Synthetic Pose Selection for Visual Localization12 May 2026
- 2026MASS-DPO: Multi-negative Active Sample Selection for Direct Policy Optimization11 May 2026
50 newest of 602 All 602 (JSON)
- 2023Active Prompt Learning in Vision Language Modelscited by 25
- 2024DOTA: Distributional Test-Time Adaptation of Vision-Language Modelscited by 22
- 2024Active Learning for Vision-Language Modelscited by 15
- 2024Revisiting Active Learning in the Era of Vision Foundation Modelscited by 13
- 2024Post-hoc Probabilistic Vision-Language Modelscited by 13
- 2024Active Prompt Learning with Vision-Language Model Priorscited by 2
- 2025Active Learning via Vision-Language Model Adaptation with Open Datacited by 2
- 2026VIOLA: Towards Video In-Context Learning with Minimal Annotationscited by 2
- 2026Leveraging Vision-Language Models as Weak Annotators in Active Learningcited by 1
- 2025PCoreSet: Effective Active Learning through Knowledge Distillation from Vision-Language Modelscited by 0
- 2025Optimizing Active Learning in Vision-Language Models via Parameter-Efficient Uncertainty Calibrationcited by 0
- 2026Explicit Uncertainty Modeling for Active CLIP Adaptation with Dual Prompt Tuningcited by 0
12 most cited of 13 All 13 (JSON)
- 2026A Human-in-the-Loop Framework for Efficient Prompt Selection in Microscopy Vision-Language Models19 May 2026
- 2026Leveraging Vision-Language Models as Weak Annotators in Active Learning1 May 2026
- 2026Explicit Uncertainty Modeling for Active CLIP Adaptation with Dual Prompt Tuning4 Feb 2026
- 2026VIOLA: Towards Video In-Context Learning with Minimal Annotations22 Jan 2026
- 2025Optimizing Active Learning in Vision-Language Models via Parameter-Efficient Uncertainty Calibration29 Jul 2025
- 2025Active Learning via Vision-Language Model Adaptation with Open Data2 Jun 2025
- 2025PCoreSet: Effective Active Learning through Knowledge Distillation from Vision-Language Models1 Jun 2025
- 2024Post-hoc Probabilistic Vision-Language Models8 Dec 2024
- 2024Active Prompt Learning with Vision-Language Model Priors23 Nov 2024
- 2024Active Learning for Vision-Language Models29 Oct 2024
- 2024DOTA: Distributional Test-Time Adaptation of Vision-Language Models28 Sep 2024
- 2024Revisiting Active Learning in the Era of Vision Foundation Models25 Jan 2024
12 newest of 13 All 13 (JSON)
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2025Active Learning for Direct Preference Optimizationcited by 9
- 2024Out-of-Distribution Learning with Human Feedbackcited by 6
- 2024Cost-Effective Proxy Reward Model Construction with On-Policy and Active Learningcited by 5
- 2026MASS-DPO: Multi-negative Active Sample Selection for Direct Policy Optimizationcited by 1
- 2026Online Data Selection Is Implicit Alignmentcited by 1
- 2026Random Is Hard to Beat: Active Selection in online DPO with Modern LLMscited by 0
- 2026GRACE: Gradient-aligned Reasoning Data Curation for Efficient Post-trainingcited by 0
- 2026Online Data Selection Is Implicit Alignment8 Jul 2026
- 2026GRACE: Gradient-aligned Reasoning Data Curation for Efficient Post-training13 May 2026
- 2026MASS-DPO: Multi-negative Active Sample Selection for Direct Policy Optimization11 May 2026
- 2026Random Is Hard to Beat: Active Selection in online DPO with Modern LLMs3 Apr 2026
- 2025Active Learning for Direct Preference Optimization3 Mar 2025
- 2024Out-of-Distribution Learning with Human Feedback14 Aug 2024
- 2024Cost-Effective Proxy Reward Model Construction with On-Policy and Active Learning2 Jul 2024
- 2019Fine-Tuning Language Models from Human Preferences18 Sep 2019
- 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoningcited by 38
- 2026Clip-level Uncertainty and Temporal-aware Active Learning for End-to-End Multi-Object Trackingcited by 1
- 2026Distilling LLM Reasoning into Graph of Concept Predictorscited by 0
- 2026SALT: When More Rollouts Don't Help in Group-Based Policy Optimization and How to Make Them Mattercited by 0
- 2026SALT: When More Rollouts Don't Help in Group-Based Policy Optimization and How to Make Them Matter4 Jun 2026
- 2026Clip-level Uncertainty and Temporal-aware Active Learning for End-to-End Multi-Object Tracking11 May 2026
- 2026Distilling LLM Reasoning into Graph of Concept Predictors3 Feb 2026
- 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoning26 May 2025
- 2025Exploiting the Experts: Unauthorized Compression in MoE-LLMs22 Nov 2025
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
- 2024ShieldGemma: Generative AI Content Moderation Based on Gemmacited by 224
- 2021Active Learning by Acquiring Contrastive Examplescited by 154
- 2022Active Learning on a Budget: Opposite Strategies Suit High and Low Budgetscited by 131
- 2021Batch Active Learning at Scalecited by 127
- 2022Coverage-centric Coreset Selection for High Pruning Ratescited by 115
- 2021Gone Fishing: Neural Active Learning with Fisher Embeddingscited by 88
- 2022A Comparative Survey of Deep Active Learningcited by 88
- 2021SIMILAR: Submodular Information Measures Based Active Learning In Realistic Scenarioscited by 83
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2023LLMaAA: Making Large Language Models as Active Annotatorscited by 78
- 2024Data Shapley in One Training Runcited by 76
12 most cited of 565 All 565 (JSON)
- 2026From internal representations to model improvement through prediction errors28 Sep 2026
- 2026TC-ADA: One-Shot Active Domain Adaptation for Semantic Segmentation27 Sep 2026
- 2026Feature Space Guidance for Breast Cancer Classification in DCE-MRI26 Sep 2026
- 2026MICRO: Multi-Fidelity Active Search for Severe Error Discovery22 Sep 2026
- 2026Clustering-Based Balanced Sampling and Allocation with Data Parallelism for High-Performance Fine-Tuning11 Sep 2026
- 2026RAIDAL: Redundancy-Aware Information Density Active Learning for CTC-Based Continuous Sign Language Recognition6 Sep 2026
- 2026Train Smarter, Not Harder: Switching Signal-Guided Training in Active Learning6 Sep 2026
- 2026AdaptNTK: Adaptive Uncertainty Quantification and Active Learning for Neural Network Potentials31 Aug 2026
- 2026ActiveAugment: Online Active Learning for Augmentation Selection in Deep Learning28 Aug 2026
- 2026FrameScope: Temporal Data Valuation for Stream Active Learning in Autonomous Vehicle Systems24 Aug 2026
- 2026Target-Aware Calibration Data Selection for Preserving Uncertainty in Quantized Language Models21 Aug 2026
- 2026Probing Association Instability with Track-State Perturbations for Clip-Level Active Learning in Query-Propagation Multi-Object Tracking18 Aug 2026
12 newest of 565 All 565 (JSON)