BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning
Not on the map. Why?
Abstract (arXiv)
We develop BatchBALD, a tractable approximation to the mutual information between a batch of points and model parameters, which we use as an acquisition function to select multiple informative points jointly for the task of deep Bayesian active learning. BatchBALD is a greedy linear-time 1 - frac1e-approximate algorithm amenable to dynamic programming and efficient caching. We compare BatchBALD to the commonly used approach for batch data acquisition and find that the current approach acquires similar and redundant points, sometimes performing worse than randomly acquiring data. We finish by showing that, using BatchBALD to consider dependencies within an acquisition batch, we achieve new state of the art performance on standard benchmarks, providing substantial data efficiency improvements in batch acquisition.
Key facts. Cited by 428 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 76; 2022: 85; 2023: 84; 2024: 75; 2025: 61; 2026: 46.
428 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
- 2021Deep Deterministic Uncertainty: A Simple Baselinecited by 200
- 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
- 2023Modern Bayesian Experimental Designcited by 121
- 2021Shifts: A Dataset of Real Distributional Shift Across Multiple Large-Scale Taskscited by 100
- 2021Gone Fishing: Neural Active Learning with Fisher Embeddingscited by 88
- 2022A Comparative Survey of Deep Active Learningcited by 88
- 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
38 more
- 2023Anchor Points: Benchmarking Models with Much Fewer Examplescited by 84
- 2021SIMILAR: Submodular Information Measures Based Active Learning In Realistic Scenarioscited by 83
- 2021Active Testing: Sample-Efficient Model Evaluationcited by 82
- 2021Active Learning for Deep Object Detection via Probabilistic Modelingcited by 79
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021On Statistical Bias In Active Learning: How and When To Fix Itcited by 72
- 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
- 2024Active Preference Learning for Large Language Modelscited by 55
- 2024A Survey on Deep Active Learning: Recent Advances and New Frontierscited by 53
- 2021ReDAL: Region-based and Diversity-aware Active Learning for Point Cloud Semantic Segmentationcited by 52
- 2023Curated LLM: Synergy of LLMs and Data Curation for tabular augmentation in low-data regimescited by 52
- 2021Improving black-box optimization in VAE latent space using decoder uncertaintycited by 51
- 2022Interventions, Where and How? Experimental Design for Causal Models at Scalecited by 49
- 2022A Framework and Benchmark for Deep Batch Active Learning for Regressioncited by 47
- 2022FedBalancer: Data and Pace Control for Efficient Federated Learning on Heterogeneous Clientscited by 46
- 2021Active label cleaning for improved dataset quality under resource constraintscited by 44
- 2024Rethinking Aleatoric and Epistemic Uncertaintycited by 40
- 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
- 2023Explaining Predictive Uncertainty with Information Theoretic Shapley Valuescited 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
- 2022Computer-Aided Multi-Objective Optimization in Small Molecule Discoverycited by 31
- 2024PINNACLE: PINN Adaptive ColLocation and Experimental points selectioncited by 31
- 2022Active Learning for Optimal Intervention Design in Causal Modelscited by 30
- 2021Grain: Improving Data Efficiency of Graph Neural Networks via Diversified Influence Maximizationcited by 28
- 2022Unifying Approaches in Active Learning and Active Sampling via Fisher Information and Information-Theoretic Quantitiescited by 28
- 2021Data Shapley Valuation for Efficient Batch Active Learningcited by 27
- 2021Not All Labels Are Equal: Rationalizing The Labeling Costs for Training Object Detectioncited by 27
- 2021Causal-BALD: Deep Bayesian Active Learning of Outcomes to Infer Treatment-Effects from Observational Datacited by 27
- 2021TALISMAN: Targeted Active Learning for Object Detection with Rare Classes and Slices using Submodular Mutual Informationcited by 27
- 2022Bayesian Active Learning with Fully Bayesian Gaussian Processescited by 26
- 2022In Defense of Core-set: A Density-aware Core-set Selection for Active Learningcited by 26
- 2023Exploring Active 3D Object Detection from a Generalization Perspectivecited by 26
50 most cited of 428 All 428 (JSON)
- 2026UpliftMem: Learning Set-Level Uplift for Agent Memory Retrieval29 Sep 2026
- 2026Active Data Acquisition with Side Information via Discrete Diffusion Priors26 Sep 2026
- 2026Reference-Grafting Matches Fine-Tuning at Eliciting Sandbagged Capabilities29 Aug 2026
- 2026Enhancing Bayesian Optimization and Active Learning Through Kernel Diversity25 Aug 2026
- 2026DynActiveGS: Active Gaussian Splatting for Dynamic Scene Reconstruction2 Aug 2026
- 2026Active few-shot segmentation by reinforcing data selection24 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
- 2026Slide-Level Active Learning Reduces Annotation Burden in H&E images10 Jul 2026
- 2026The Approximation Ratio for the Risk of Myopic Bayesian Active Learning for Linear Regression7 Jul 2026
- 2026A Mechanism-Driven Theory of Phase Transitions in Active Learning30 Jun 2026
- 2026Uncertainty Quantification for Flow-Based Vision-Language-Action Models16 Jun 2026
38 more
- 2026Epistemic Uncertainty Is Not the Reducible Kind10 Jun 2026
- 2026ATLAS: Active Theory Learning for Automated Science10 Jun 2026
- 2026Active Inference as Context Acquisition for AI Agents8 Jun 2026
- 2026CAAL: Contextual Bandits based Online Hand-Craft Active Learning Strategy Selection6 Jun 2026
- 2026EvoPool: Evolutionary Programmatic Annotation for Label-Efficient Specialized Supervision1 Jun 2026
- 2026Uncertainty Decomposition via Cyclical SG-MCMC and Soft-label Learning for Subjective NLP23 May 2026
- 2026Interaction-Aware Influence Functions for Group Attribution15 May 2026
- 2026A Mutual Information Lower Bound for Multimodal Regression Active Learning14 May 2026
- 2026Cognitive-Uncertainty Guided Knowledge Distillation for Accurate Classification of Student Misconceptions14 May 2026
- 2026MASS-DPO: Multi-negative Active Sample Selection for Direct Policy Optimization11 May 2026
- 2026Many Needles in a Haystack: Active Hit Discovery for Perturbation Experiments11 May 2026
- 2026Active Learning for Communication Structure Optimization in LLM-Based Multi-Agent Systems7 May 2026
- 2026Uncertainty-Guided Edge Learning for Deep Image Regression in Remote Sensing7 May 2026
- 2026RADS: Reinforcement Learning-Based Sample Selection Improves Transfer Learning in Low-resource and Imbalanced Clinical Settings22 Apr 2026
- 2026When Active Learning Falls Short: An Empirical Study on Chemical Reaction Extraction21 Apr 2026
- 2026Labeled TrustSet Guided: Batch Active Learning with Reinforcement Learning14 Apr 2026
- 2026Loss-Driven Bayesian Active Learning13 Apr 2026
- 2026Testing the Assumptions of Active Learning for Translation Tasks with Few Samples10 Apr 2026
- 2026Active In-Context Learning for Tabular Foundation Models28 Mar 2026
- 2026PruneFuse: Efficient Data Selection via Weight Pruning and Network Fusion27 Mar 2026
- 2026Select, Label, Evaluate: Active Testing in NLP23 Mar 2026
- 2026GSMem: 3D Gaussian Splatting as Persistent Spatial Memory for Zero-Shot Embodied Exploration and Reasoning19 Mar 2026
- 2026Less Data, Faster Convergence: Goal-Driven Data Optimization for Multimodal Instruction Tuning12 Mar 2026
- 2026Goal-Oriented Influence-Maximizing Data Acquisition for Learning and Optimization23 Feb 2026
- 2026Similarity-as-Evidence: Calibrating Overconfident VLMs for Interpretable and Label-Efficient Medical Active Learning21 Feb 2026
- 2026Team, Then Trim: An Assembly-Line LLM Framework for High-Quality Tabular Data Generation4 Feb 2026
- 2026Adaptive Prompt Elicitation for Text-to-Image Generation4 Feb 2026
- 2026Distilling LLM Reasoning into Graph of Concept Predictors3 Feb 2026
- 2026Active Transfer Bagging: A New Approach for Accelerated Active Learning Acquisition of Data by Combined Transfer Learning and Bagging Based Models2 Feb 2026
- 2026Beyond the Loss Curve: Scaling Laws, Active Learning, and the Limits of Learning from Exact Posteriors30 Jan 2026
- 2026From Cold Start to Active Learning: Embedding-Based Scan Selection for Medical Image Segmentation26 Jan 2026
- 2026Active learning for photonic crystals22 Jan 2026
- 2026Memorize Early, Then Query: Inlier-Memorization-Guided Active Outlier Detection16 Jan 2026
- 2026HERE: Hierarchical Active Exploration of Radiance Field with Epistemic Uncertainty Minimization12 Jan 2026
- 2025Uncertainty Quantification for Machine Learning: One Size Does Not Fit All13 Dec 2025
- 2025Hierarchical Dataset Selection for High-Quality Data Sharing11 Dec 2025
- 2025Towards Active Synthetic Data Generation for Finetuning Language Models30 Nov 2025
- 2025Active Learning with Selective Time-Step Acquisition for PDEs22 Nov 2025
50 newest of 428 All 428 (JSON)
- 2023Active Prompt Learning in Vision Language Modelscited by 25
- 2024Revisiting Active Learning in the Era of Vision Foundation Modelscited by 13
- 2023ProbVLM: Probabilistic Adapter for Frozen Vision-Language Modelscited by 7
- 2025Active Learning via Vision-Language Model Adaptation with Open Datacited by 2
- 2025PCoreSet: Effective Active Learning through Knowledge Distillation from Vision-Language Modelscited by 0
- 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
- 2024Revisiting Active Learning in the Era of Vision Foundation Models25 Jan 2024
- 2023Active Prompt Learning in Vision Language Models18 Nov 2023
- 2023ProbVLM: Probabilistic Adapter for Frozen Vision-Language Models1 Jul 2023
- 2024Shedding Light on Large Generative Networks: Estimating Epistemic Uncertainty in Diffusion Modelscited by 22
- 2025Model Already Knows the Best Noise: Bayesian Active Noise Selection via Attention in Video Diffusion Modelcited by 10
- 2025EMoE: Training-Free Expert Disagreement for Uncertainty-Aware Text-to-Image Diffusioncited by 0
- 2026Active Data Acquisition with Side Information via Discrete Diffusion Priorscited by 0
- 2026Active Data Acquisition with Side Information via Discrete Diffusion Priors26 Sep 2026
- 2025Model Already Knows the Best Noise: Bayesian Active Noise Selection via Attention in Video Diffusion Model23 May 2025
- 2025EMoE: Training-Free Expert Disagreement for Uncertainty-Aware Text-to-Image Diffusion19 May 2025
- 2024Shedding Light on Large Generative Networks: Estimating Epistemic Uncertainty in Diffusion Models5 Jun 2024
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 384
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
- 2021Deep Deterministic Uncertainty: A Simple Baselinecited by 200
- 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
- 2023Modern Bayesian Experimental Designcited by 121
- 2021Shifts: A Dataset of Real Distributional Shift Across Multiple Large-Scale Taskscited by 100
- 2021Gone Fishing: Neural Active Learning with Fisher Embeddingscited by 88
- 2022A Comparative Survey of Deep Active Learningcited by 88
- 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
12 most cited of 404 All 404 (JSON)
- 2026Reference-Grafting Matches Fine-Tuning at Eliciting Sandbagged Capabilities29 Aug 2026
- 2026Enhancing Bayesian Optimization and Active Learning Through Kernel Diversity25 Aug 2026
- 2026DynActiveGS: Active Gaussian Splatting for Dynamic Scene Reconstruction2 Aug 2026
- 2026Active few-shot segmentation by reinforcing data selection24 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
- 2026Slide-Level Active Learning Reduces Annotation Burden in H&E images10 Jul 2026
- 2026The Approximation Ratio for the Risk of Myopic Bayesian Active Learning for Linear Regression7 Jul 2026
- 2026A Mechanism-Driven Theory of Phase Transitions in Active Learning30 Jun 2026
- 2026Epistemic Uncertainty Is Not the Reducible Kind10 Jun 2026
- 2026ATLAS: Active Theory Learning for Automated Science10 Jun 2026
- 2026CAAL: Contextual Bandits based Online Hand-Craft Active Learning Strategy Selection6 Jun 2026
12 newest of 404 All 404 (JSON)