Strategic Data Selection in Machine Learning: Active Learning Approaches with Budget Constraints
Source: Lilian Weng19/02/2022, 21:00
Active learning is a machine learning technique that enables models to strategically select which unlabeled data examples should be manually annotated when labeling budgets are limited. This approach proves particularly valuable in scenarios where data annotation is expensive, such as medical imaging analysis.
The method identifies the most valuable examples through various sampling strategies. Uncertainty sampling prioritizes examples where models express lower confidence in predictions. Diversity-focused approaches seek examples that better represent the overall data distribution. Other strategies include Query-By-Committee, which leverages multiple models to assess agreement, and methods based on expected model change.
Advanced techniques like MC Dropout approximate Bayesian inference using dropout to estimate model uncertainty. DBAL combines this with Bayesian neural networks. More recent approaches such as VAAL employ generative adversarial architectures, while MAL introduces a minimax framework that reduces the distribution gap between labeled and unlabeled data. Complementary methods like CAL and core-set selection focus on contrastive examples and geometric approximations.
Experimental results demonstrate that these techniques outperform random baselines across various image classification and segmentation tasks, offering a cost-effective alternative for training models with limited data.