arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.02645cs.CLcs.LGstat.ML

SentenceMIM: A Latent Variable Language Model

University of Toronto(多伦多大学);Vector Institute(向量研究所);Google Research(谷歌研究院)

Micha Livne, Kevin Swersky, David J. Fleet

英文摘要:

SentenceMIM is a probabilistic auto-encoder for language data, trained with Mutual Information Machine (MIM) learning to provide a fixed length representation of variable length language observations (i.e., similar to VAE). Previous attempts to learn VAEs for language data faced challenges due to posterior collapse. MIM learning encourages high mutual information between observations and latent variables, and is robust against posterior collapse. As such, it learns informative representations whose dimension can be an order of magnitude higher than existing language VAEs. Importantly, the SentenceMIM loss has no hyper-parameters, simplifying optimization. We compare sentenceMIM with VAE, and AE on multiple datasets. SentenceMIM yields excellent reconstruction, comparable to AEs, with a rich structured latent space, comparable to VAEs. The structured latent representation is demonstrated with interpolation between sentences of different lengths. We demonstrate the versatility of sentenceMIM by utilizing a trained model for question-answering and transfer learning, without fine-tuning, outperforming VAE and AE with similar architectures.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑