k-Volume: a Unified Active Learning Strategy
Organizations: SAMM, Université Paris 1 · IRIT, INP Toulouse
Abstract
In the classification context where revealing the label of a sample is costly, we propose a new Active Learning (AL) strategy named k-Volume (kVol). It randomly selects a sample to be labeled according to the volume of the smallest ball containing its k nearest neighbors. kVol is highly versatile: it can be applied either in the input space or in the output space, and it enjoys the desirable properties of the three main families of AL strategies: density, diversity, and (when performed in the output space) uncertainty. From a theoretical perspective, we establish that kVol is asymptotically equivalent to uniform sampling over the support of the (input or output) data distribution. Based on experiments on various synthetic and real-life tabular datasets and images, we observe that kVol is surprisingly stable: whatever the classifier or the experimental setup, it still lies among the best AL strategies unlike other competitors that fail dramatically. In particular, kVol outperforms state-of-the-art competitors such as CoreSet, TypiClust or Uncertainty on several real-life datasets, at a low computational cost.
Figures & tables
| Scenario | Balanced | Dimension |
|---|---|---|
| No. 1 | Yes | 2 |
| No. 2 | No | 2 |
| No. 3 | Yes | 20 |
| No. 4 | No | 20 |
| Dataset Id | Name | Number of classes | Number of points | Input dimension |
|---|---|---|---|---|
| 46 | splice | 3 | 3190 | 60 |
| 182 | satimage | 6 | 6430 | 36 |
| 846 | elevators | 2 | 16599 | 18 |
| Name | Number of classes | Number of points | Image dimension | Embedding dimension | Reference |
|---|---|---|---|---|---|
| MNIST | 10 | 60000 | 784 | 2048 | [ 24 ] |
| CIFAR10 | 10 | 50000 | 3072 | 2048 | [ 23 ] |
| CIFAR100 | 100 | 50000 | 3072 | 2048 | [ 23 ] |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Uniform | Entropy | CoreSet | TypiClust | kVolOut ( ) | kVolIn ( ) | |
|---|---|---|---|---|---|---|
| Figure 6 | 1.70e+03s | 1.71e+03s | 3.06e+03s | 1.07e+04s | 1.73e+03s | 1.90e+03s |
| Figure 14 | 6.54e+03s | 6.54e+03s | - | - | 6.58e+03s | 6.87e+03s |
| kVolOut1 | kVolOut10 | kVolOut100 | kVolOut500 | |||
| Figure 15 | 1.14e+02s | 1.37e+02s | 2.01e+02s | 3.43e+02s |
Active Learning
Active learning selects the examples, labels, features, or experiments most useful for improving a model under limited budgets.
See all 312 papers