Topic-VQ-VAE: Leveraging Latent Codebooks for Flexible Topic-Guided Document Generation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yoo, YoungJoon, Choi, Jongwon
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914646692200448
author Yoo, YoungJoon
Choi, Jongwon
author_facet Yoo, YoungJoon
Choi, Jongwon
contents This paper introduces a novel approach for topic modeling utilizing latent codebooks from Vector-Quantized Variational Auto-Encoder~(VQ-VAE), discretely encapsulating the rich information of the pre-trained embeddings such as the pre-trained language model. From the novel interpretation of the latent codebooks and embeddings as conceptual bag-of-words, we propose a new generative topic model called Topic-VQ-VAE~(TVQ-VAE) which inversely generates the original documents related to the respective latent codebook. The TVQ-VAE can visualize the topics with various generative distributions including the traditional BoW distribution and the autoregressive image generation. Our experimental results on document analysis and image generation demonstrate that TVQ-VAE effectively captures the topic context which reveals the underlying structures of the dataset and supports flexible forms of document generation. Official implementation of the proposed TVQ-VAE is available at https://github.com/clovaai/TVQ-VAE.
format Preprint
id arxiv_https___arxiv_org_abs_2312_11532
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Topic-VQ-VAE: Leveraging Latent Codebooks for Flexible Topic-Guided Document Generation
Yoo, YoungJoon
Choi, Jongwon
Computation and Language
Artificial Intelligence
Machine Learning
This paper introduces a novel approach for topic modeling utilizing latent codebooks from Vector-Quantized Variational Auto-Encoder~(VQ-VAE), discretely encapsulating the rich information of the pre-trained embeddings such as the pre-trained language model. From the novel interpretation of the latent codebooks and embeddings as conceptual bag-of-words, we propose a new generative topic model called Topic-VQ-VAE~(TVQ-VAE) which inversely generates the original documents related to the respective latent codebook. The TVQ-VAE can visualize the topics with various generative distributions including the traditional BoW distribution and the autoregressive image generation. Our experimental results on document analysis and image generation demonstrate that TVQ-VAE effectively captures the topic context which reveals the underlying structures of the dataset and supports flexible forms of document generation. Official implementation of the proposed TVQ-VAE is available at https://github.com/clovaai/TVQ-VAE.
title Topic-VQ-VAE: Leveraging Latent Codebooks for Flexible Topic-Guided Document Generation
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2312.11532