mlsanityData to 7 Oct 2026

Non-negative matrix factorization algorithms generally improve topic model fits

arXiv 2105.13440 ↗First posted 27 May 2021stat.ML cs.LG stat.COcited by 2

Not on the map. Why?

Abstract (arXiv)

In an effort to develop topic modeling methods that can be quickly applied to large data sets, we revisit the problem of maximum-likelihood estimation in topic models. It is known, at least informally, that maximum-likelihood estimation in topic models is closely related to non-negative matrix factorization (NMF). Yet, to our knowledge, this relationship has not been exploited previously to fit topic models. We show that recent advances in NMF optimization methods can be leveraged to fit topic models very efficiently, often resulting in much better fits and in less time than existing algorithms for topic models. We also formally make the connection between the NMF optimization problem and maximum-likelihood estimation for the topic model, and using this result we show that the expectation maximization (EM) algorithm for the topic model is essentially the same as the classic multiplicative updates for NMF (the only difference being that the operations are performed in a different order). Our methods are implemented in the R package fastTopics.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 11 of its 100 references that are in the map. First posted on arXiv 27 May 2021.

Builds on

11 of 100 references.

References · 11

  1. 2017Implicit Regularization in Matrix Factorizationcited by 318
  2. 2015Coordinate Descent Algorithmscited by 218
  3. 2018A Continuous-Time View of Early Stopping for Least Squarescited by 67
  4. 2013Streaming Variational Bayescited by 46
  5. 2018Singularity, Misspecification, and the Convergence Rate of EMcited by 38
  6. 2014The Why and How of Nonnegative Matrix Factorizationcited by 34
  7. 2020Homeomorphic-Invariance of EM: Non-Asymptotic Convergence in KL Divergence for Exponential Families via Mirror Descentcited by 18
  8. 2020Algorithms for Nonnegative Matrix Factorization with the Kullback-Leibler Divergencecited by 17
  9. 2013Semidefinite Programming Based Preconditioning for More Robust Near-Separable Nonnegative Matrix Factorizationcited by 13
  10. 2018Accelerating Nonnegative Matrix Factorization Algorithms using Extrapolationcited by 10
  11. 2018Damped Anderson acceleration with restarts and monotonicity control for accelerating EM and EM-like algorithmscited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 1; 2026: 1.

2 papers.