English

An Information-Theoretic Framework for Unifying Active Learning Problems

Machine Learning 2020-12-22 v1 Machine Learning

Abstract

This paper presents an information-theoretic framework for unifying active learning problems: level set estimation (LSE), Bayesian optimization (BO), and their generalized variant. We first introduce a novel active learning criterion that subsumes an existing LSE algorithm and achieves state-of-the-art performance in LSE problems with a continuous input domain. Then, by exploiting the relationship between LSE and BO, we design a competitive information-theoretic acquisition function for BO that has interesting connections to upper confidence bound and max-value entropy search (MES). The latter connection reveals a drawback of MES which has important implications on not only MES but also on other MES-based acquisition functions. Finally, our unifying information-theoretic framework can be applied to solve a generalized problem of LSE and BO involving multiple level sets in a data-efficient manner. We empirically evaluate the performance of our proposed algorithms using synthetic benchmark functions, a real-world dataset, and in hyperparameter tuning of machine learning models.

Keywords

Cite

@article{arxiv.2012.10695,
  title  = {An Information-Theoretic Framework for Unifying Active Learning Problems},
  author = {Quoc Phong Nguyen and Bryan Kian Hsiang Low and Patrick Jaillet},
  journal= {arXiv preprint arXiv:2012.10695},
  year   = {2020}
}

Comments

35th AAAI Conference on Artificial Intelligence (AAAI 2021), Extended version with derivations, 12 pages