Neural nonnegative matrix factorization for hierarchical multilayer topic modeling
Author(s)
Haddock, Jamie; Will, Tyler; Vendrow, Joshua; Zhang, Runyu; Molitor, Denali; Needell, Deanna; Gao, Mengdi; Sadovnik, Eli; ... Show more Show less
Download43670_2023_Article_77.pdf (1.534Mb)
Publisher with Creative Commons License
Publisher with Creative Commons License
Creative Commons Attribution
Terms of use
Metadata
Show full item recordAbstract
We introduce a new method based on nonnegative matrix factorization, Neural NMF, for detecting latent hierarchical structure in data. Datasets with hierarchical structure arise in a wide variety of fields, such as document classification, image processing, and bioinformatics. Neural NMF recursively applies NMF in layers to discover overarching topics encompassing the lower-level features. We derive a backpropagation optimization scheme that allows us to frame hierarchical NMF as a neural network. We test Neural NMF on a synthetic hierarchical dataset, the 20 Newsgroups dataset, and the MyLymeData symptoms dataset. Numerical results demonstrate that Neural NMF outperforms other hierarchical NMF methods on these data sets and offers better learned hierarchical structure and interpretability of topics.
Date issued
2023-12-19Department
Massachusetts Institute of Technology. Department of Electrical Engineering and Computer SciencePublisher
Springer International Publishing
Citation
Sampling Theory, Signal Processing, and Data Analysis. 2023 Dec 19;22(1):4
Version: Final published version