Variational Autoencoder for Semi-Supervised Text Classification

By Weidi Xu, Haoze Sun, Chao Deng, Ying Tan. 2017

236 citations recorded by the source corpus.

Although semi-supervised variational autoencoder (SemiVAE) works in image classification task, it fails in text classification task if using vanilla LSTM as its decoder. From a perspective of reinforcement learning, it is verified that the decoder's capability to distinguish between different categorical labels is essential. Therefore, Semi-supervised Sequential Variational Autoencoder (SSVAE) is proposed, which increases the capability by feeding label into its decoder RNN at each time-step. Two specific decoder structures are investigated and both of them are verified to be effective. Besides, in order to reduce the computational complexity in training, a novel optimization method is proposed, which estimates the gradient of the unlabeled objective function by sampling, along with two variance reduction techniques. Experimental results on Large Movie Review Dataset (IMDB) and AG's News corpus show that the proposed approach significantly improves the classification accuracy compared with pure-supervised classifiers, and achieves competitive performance against previous advanced methods. State-of-the-art results can be obtained by integrating other pretraining-based methods.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Mohit Bansal
  3. Chunyuan Li
  4. Luke S. Zettlemoyer
  5. Dacheng Tao
  6. Percy Liang
  7. Yejin Choi
  8. Xiang Ren

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Jianfeng Gao
  3. Chunyuan Li
  4. Luc Van Gool
  5. Mohit Bansal
  6. Trevor Darrell
  7. Ming-Hsuan Yang
  8. Xiaogang Wang

Citations in the corpus