Neural Embedding Allocation: Distributed Representations of Topic Models

Authors

  • Kamrun Naher Keya University of Maryland, Baltimore County (UMBC)
  • Yannis Papanikolaou Healx, Cambridge
  • James R. Foulds University of Maryland, Baltimore County (UMBC)

Abstract

We propose a method which uses neural embeddings to improve the performance of any given LDA-style topic model.  Our method, called neural embedding allocation (NEA), deconstructs topic models (LDA or otherwise) into interpretable vector-space embeddings of words, topics, documents, authors, and so on, by learning neural embeddings to mimic the topic model. We demonstrate that NEA improves coherence scores of the original topic model by smoothing out the noisy topics when the number of topics is large. Furthermore, we show NEA's effectiveness and generality in deconstructing and smoothing LDA, author-topic models, and the recent mixed membership skip-gram topic model and achieve better performance with the embeddings compared to several state-of-the-art models.

Author Biographies

  • Kamrun Naher Keya, University of Maryland, Baltimore County (UMBC)
    Ph.D. Student, Department of Information Systems
  • James R. Foulds, University of Maryland, Baltimore County (UMBC)

    Assistant Professor, Department of Information Systems

Published

2024-11-15