arXiv · 1806.09856
Dropout-based Active Learning for Regression
Abstract
Active learning is relevant and challenging for high-dimensional regression models when the annotation of the samples is expensive. Yet most of the existing sampling methods cannot be applied to large-scale problems, consuming too much time for data processing. In this paper, we propose a fast active learning algorithm for regression, tailored for neural network models. It is based on uncertainty estimation from stochastic dropout output of the network. Experiments on both synthetic and real-world datasets show comparable or better performance (depending on the accuracy metric) as compared to the baselines. This approach can be generalized to other deep learning architectures. It can be used to systematically improve a machine-learning model as it offers a computationally efficient way of sampling additional data.
Explore related subjects
Keep this discovery
Evgenii Tsymbalov, Maxim Panov, Alexander Shapeev. 2018-06-26. Dropout-based Active Learning for Regression. https://doi.org/10.1007/978-3-030-11027-7_24
Cite the original work for its findings. Save a collection to share your selection of sources.