References & Citations
Computer Science > Computation and Language
Title: AdaVAE: Exploring Adaptive GPT-2s in Variational Auto-Encoders for Language Modeling
(Submitted on 12 May 2022 (v1), last revised 19 Nov 2022 (this version, v3))
Abstract: Variational Auto-Encoder (VAE) has become the de-facto learning paradigm in achieving representation learning and generation for natural language at the same time. Nevertheless, existing VAE-based language models either employ elementary RNNs, which is not powerful to handle complex works in the multi-task situation, or fine-tunes two pre-trained language models (PLMs) for any downstream task, which is a huge drain on resources. In this paper, we propose the first VAE framework empowered with adaptive GPT-2s (AdaVAE). Different from existing systems, we unify both the encoder\&decoder of the VAE model using GPT-2s with adaptive parameter-efficient components, and further introduce Latent Attention operation to better construct latent space from transformer models. Experiments from multiple dimensions validate that AdaVAE is competent to effectively organize language in three related tasks (language modeling, representation modeling and guided text generation) even with less than $15\%$ activated parameters in training. Our code is available at \url{this https URL}.
Submission history
From: Haoqin Tu [view email][v1] Thu, 12 May 2022 03:22:07 GMT (46kb,D)
[v2] Fri, 20 May 2022 05:52:50 GMT (48kb,D)
[v3] Sat, 19 Nov 2022 12:41:26 GMT (261kb,D)
Link back to: arXiv, form interface, contact.