arXiv · 1909.09237
Improved Variational Neural Machine Translation by Promoting Mutual Information
Abstract
Posterior collapse plagues VAEs for text, especially for conditional text generation with strong autoregressive decoders. In this work, we address this problem in variational neural machine translation by explicitly promoting mutual information between the latent variables and the data. Our model extends the conditional variational autoencoder (CVAE) with two new ingredients: first, we propose a modified evidence lower bound (ELBO) objective which explicitly promotes mutual information; second, we regularize the probabilities of the decoder by mixing an auxiliary factorized distribution which is directly predicted by the latent variables. We present empirical results on the Transformer architecture and show the proposed model effectively addressed posterior collapse: latent variables are no longer ignored in the presence of powerful decoder. As a result, the proposed model yields improved translation quality while demonstrating superior performance in terms of data efficiency and robustness.
Explore related subjects
Keep this discovery
Arya D. McCarthy, Xian Li, Jiatao Gu, Ning Dong. 2019-09-19. Improved Variational Neural Machine Translation by Promoting Mutual Information. https://arxiv.org/abs/1909.09237
Cite the original work for its findings. Save a collection to share your selection of sources.