Active Learning for Convolutional Neural Networks: A Core-Set Approach
Ozan SenerSilvio Savarese
Proposes a geometric core-set formulation for batch active learning in convolutional neural networks, providing theoretical performance bounds and an effective subset selection algorithm that significantly reduces labeling requirements for image classification.
Labeling large image datasets for training convolutional neural networks remains expensive and time-consuming, creating a practical need for methods that select the most informative points to label under a fixed budget. Existing active learning heuristics often prove ineffective for CNNs when labels are acquired in batches rather than one at a time, largely because batch sampling introduces correlations that reduce diversity.
The article sets out to reframe active learning as a core-set selection task: choosing a modest subset of unlabeled points such that a model trained on the labeled subset performs nearly as well on the remaining data as a model trained on the full set.
The authors derive a geometric upper bound on the difference in average loss between any chosen subset and the full dataset, showing that this core-set loss is controlled by the covering radius of the selected points. They convert the resulting optimization into the classic k-Center problem and solve it with a greedy 2-OPT algorithm followed by a mixed-integer program that improves robustness to outliers. Experiments compare the method against random sampling, uncertainty-based baselines, Bayesian approaches, and prior batch-mode algorithms on CIFAR-10, CIFAR-100, and SVHN under both fully supervised and weakly supervised training regimes.
The proposed core-set method consistently delivers higher classification accuracy than all baselines, often by substantial margins, and performs especially well when combined with weakly supervised models that produce richer feature spaces. Uncertainty-driven strategies, by contrast, frequently underperform random sampling once batch correlations are taken into account. The geometric coverage produced by the algorithm visibly spreads selected points more evenly across the data manifold than uncertainty sampling.
These results indicate that organizations can achieve target accuracy levels with far fewer labeled examples, lowering labeling costs and shortening project timelines without sacrificing model quality. The approach also supplies a practical way to decide how many additional labels are worth acquiring at each iteration.
Practitioners should adopt the core-set procedure, preferring the full mixed-integer solution when dataset size permits and falling back to the fast greedy approximation otherwise. Future work could explore principled ways to blend uncertainty signals with the geometric criterion and test scaling behavior on datasets larger than 50 000 images.
The main limitations are the zero-training-error assumption used to simplify the bound, which is only approximately true, and the fact that the performance guarantee weakens as the number of classes grows. Results are nevertheless consistent across three standard benchmarks and multiple random initial pools, supporting moderate-to-high confidence for image-classification tasks of similar scale.
- Paper: Active Learning with Statistical Models, D. Cohn et al. (1996). This seminal paper introduces statistically optimal variance-reduction frameworks for active data selection, establishing the foundational theoretical principles for querying informative samples.
- Paper: Support Vector Machine Active Learning with Applications to Text Classification, Simon Tong et al. (2001). This foundational work establishes pool-based active learning via margin and version-space query heuristics, providing essential background for understanding uncertainty-based sample selection.
- Paper: Query by committee, H. Seung et al. (1992). This classic work formalizes pool-based query-by-committee active learning, which serves as a primary conceptual baseline for disagreement- and uncertainty-driven active data acquisition.
- Paper: Combining active learning and semi-supervised learning using Gaussian fields and harmonic functions, Xiaojin Zhu et al. (2003). This paper establishes the core principles of combining graph-based semi-supervised structure with active learning to minimize expected classification risk.
- Paper: Neural Network Ensembles, Cross Validation, and Active Learning, Anders Krogh et al. (1994). This paper derives the relationship between ensemble ambiguity and active sample selection, motivating modern disagreement-based active learning strategies.
No sufficiently relevant recommendations were found.
