This is the main code of the paper "CatVRNN: Generating Category Texts via Multi-task Learning".
- pytorch 1.6.0
- tqdm 4.55.1
- Python 3.7.9
@article{CHENG2022108491,
title = {CatVRNN: Generating category texts via multi-task learning},
journal = {Knowledge-Based Systems},
volume = {244},
pages = {108491},
year = {2022},
issn = {0950-7051},
doi = {https://doi.org/10.1016/j.knosys.2022.108491},
url = {https://www.sciencedirect.com/science/article/pii/S095070512200209X},
author = {Pengsen Cheng and Jinqiao Dai and Jiayong Liu},
keywords = {Category text generation, Multi-task learning, Variational recurrent neural network, State initialization},
abstract = {Controlling the model to generate texts of different categories is a challenging task that is receiving increasing attention. Recently, generative adversarial networks (GANs) have shown promising results for category text generation. However, the texts generated by GANs usually suffer from problems of mode collapse and training instability. To avoid the above problems, in this study, inspired by multi-task learning, a novel model called category-aware variational recurrent neural network (CatVRNN) is proposed. In this model, generation and classification tasks are trained simultaneously to generate texts of different categories. The use of multi-task learning can improve the quality of the generated texts, when the classification task is appropriate. In addition, a function is proposed to initialize the hidden state of the CatVRNN to force the model to generate texts of a specific category. Experimental results on three datasets demonstrate that the model can outperform state-of-the-art text generation methods based on GAN in terms of diversity of generated texts.}
}