English

Is margin all you need? An extensive empirical study of active learning on tabular data

Machine Learning 2022-10-11 v1 Artificial Intelligence

Abstract

Given a labeled training set and a collection of unlabeled data, the goal of active learning (AL) is to identify the best unlabeled points to label. In this comprehensive study, we analyze the performance of a variety of AL algorithms on deep neural networks trained on 69 real-world tabular classification datasets from the OpenML-CC18 benchmark. We consider different data regimes and the effect of self-supervised model pre-training. Surprisingly, we find that the classical margin sampling technique matches or outperforms all others, including current state-of-art, in a wide range of experimental settings. To researchers, we hope to encourage rigorous benchmarking against margin, and to practitioners facing tabular data labeling constraints that hyper-parameter-free margin may often be all they need.

Keywords

Cite

@article{arxiv.2210.03822,
  title  = {Is margin all you need? An extensive empirical study of active learning on tabular data},
  author = {Dara Bahri and Heinrich Jiang and Tal Schuster and Afshin Rostamizadeh},
  journal= {arXiv preprint arXiv:2210.03822},
  year   = {2022}
}
R2 v1 2026-06-28T03:02:22.919Z