BALD-VAE: Generative Active Learning based on the Uncertainties of Both Labeled and Unlabeled Data

Cited 6 time in webofscience Cited 0 time in scopus
  • Hit : 83
  • Download : 0
Deep learning has shown outstanding performances on real world problems, but acquiring sufficient labeled data to train a model is still an on-going issue. Specifically, manually labeling data is time-consuming and costly. One approach to tackle this issue is active learning. Recently, pool-based methods and generative methods are widely studied among various approaches of active learning. Especially in the uncertainty pool-based methods, a small labeled data set and a large unlabeled data set are given. A model is trained on the labeled data set and then observes the unlabeled data set. The trained model ranks the unlabeled data in order of uncertainty to select the data which has the highest uncertainty. In the generative methods, a generative model is used to generate informative samples. In the previous studies of the uncertainty pool-based active learning, however, the uncertainty of labeled data was not considered. Thus, we propose a new Bayesian active learning by disagreement with variational autoencoder (BALD-VAE), which considers the uncertainty of labeled data when generating informative samples. Basically following the uncertainty pool-based active learning with BALD, the pro-posed algorithm also utilizes the concept of generative active learning to generate informative data using VAE. Then, the generated data complement the highly uncertain labeled data. To demonstrate the effectiveness, the proposed method is tested on MNIST and CIFAR10 data sets and shown to outperform the previous algorithms.
Publisher
Institute of Electrical and Electronics Engineers Inc.
Issue Date
2019-11-01
Language
English
Citation

7th International Conference on Robot Intelligence Technology and Applications, RiTA 2019, pp.6 - 11

DOI
10.1109/RITAPP.2019.8932813
URI
http://hdl.handle.net/10203/307282
Appears in Collection
EE-Conference Papers(학술회의논문)
Files in This Item
There are no files associated with this item.
This item is cited by other documents in WoS
⊙ Detail Information in WoSⓡ Click to see webofscience_button
⊙ Cited 6 items in WoS Click to see citing articles in records_button

qr_code

  • mendeley

    citeulike


rss_1.0 rss_2.0 atom_1.0