arXiv · 2505.06825
Active Learning for Multi-class Image Classification
Abstract
A principle bottleneck in image classification is the large number of training examples needed to train a classifier. Using active learning, we can reduce the number of training examples to teach a CNN classifier by strategically selecting examples. Assigning values to image examples using different uncertainty metrics allows the model to identify and select high-value examples in a smaller training set size. We demonstrate results for digit recognition and fruit classification on the MNIST and Fruits360 data sets. We formally compare results for four different uncertainty metrics. Finally, we observe active learning is also effective on simpler (binary) classification tasks, but marked improvement from random sampling is more evident on more difficult tasks. We show active learning is a viable algorithm for image classification problems.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Thien Nhan Vo. 2025-05-11. Active Learning for Multi-class Image Classification. https://arxiv.org/abs/2505.06825
Cite the original work for its findings. Save a collection to share your selection of sources.