Bayesian Active Learning for Classification and Preference Learning
About
Information theoretic active learning has been widely studied for probabilistic models. For simple regression an optimal myopic policy is easily tractable. However, for other tasks and with more complex models, such as classification with nonparametric models, the optimal solution is harder to compute. Current approaches make approximations to achieve tractability. We propose an approach that expresses information gain in terms of predictive entropies, and apply this method to the Gaussian Process Classifier (GPC). Our approach makes minimal approximations to the full information theoretic objective. Our experimental performance compares favourably to many popular active learning algorithms, and has equal or lower computational complexity. We compare well to decision theoretic approaches also, which are privy to more information and require much more computational time. Secondly, by developing further a reformulation of binary preference learning to a classification problem, we extend our algorithm to Gaussian Process preference learning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Active Learning | Univariate and 1D PDE datasets (test) | Average Rank3.57 | 22 | |
| Image Classification | CIFAR10 (train test) | Execution Time5.53e+3 | 11 | |
| Selective Prediction | Diabetic Retinopathy (DR) grading patient-stratified (test) | AUSC (Critical FNR)0.436 | 10 | |
| Selective Prediction | Diabetic Retinopathy (DR) (test) | AUSC0.436 | 10 | |
| Active Learning | nnActive average | AUBC62.39 | 9 | |
| Active Learning | WeChat (Full) | NLL2.0258 | 5 | |
| Active Learning | WeChat (cold-start) | NLL2.0162 | 5 | |
| Active Learning | UBER | NLL1.7851 | 5 | |
| Active Learning | KuaiRec | NLL1.8175 | 5 | |
| OOD Detection | FashionMNIST → KMNIST (test) | Ratio5.92 | 4 |