SUPClust: Active Learning at the Boundaries
Machine Learning
2024-03-07 v1 Artificial Intelligence
Computer Vision and Pattern Recognition
Abstract
Active learning is a machine learning paradigm designed to optimize model performance in a setting where labeled data is expensive to acquire. In this work, we propose a novel active learning method called SUPClust that seeks to identify points at the decision boundary between classes. By targeting these points, SUPClust aims to gather information that is most informative for refining the model's prediction of complex decision regions. We demonstrate experimentally that labeling these points leads to strong model performance. This improvement is observed even in scenarios characterized by strong class imbalance.
Cite
@article{arxiv.2403.03741,
title = {SUPClust: Active Learning at the Boundaries},
author = {Yuta Ono and Till Aczel and Benjamin Estermann and Roger Wattenhofer},
journal= {arXiv preprint arXiv:2403.03741},
year = {2024}
}
Comments
Accepted at ICLR 2024 Workshop on Practical Machine Learning for Low Resource Settings (PML4LRS)