DOE PAGES title logo U.S. Department of Energy
Office of Scientific and Technical Information

Title: A neural network for determination of latent dimensionality in Nonnegative Matrix Factorization

Abstract

Non-negative Matrix Factorization (NMF) has proven to be a powerful unsupervised learning method for uncovering hidden features in complex and noisy datasets with applications in data mining, text recognition, dimension reduction, face recognition, anomaly detection, blind source separation, and many other fields. An important input for NMF is the latent dimensionality of the data, that is, the number of hidden features, K, present in the explored dataset. Unfortunately, and this quantity is rarely known a priori. The existing methods for determining latent dimensionality, such as Automatic Relevance Determination (ARD), are mostly heuristic and utilize different characteristics to estimate the number of hidden features. However, all of them require human presence to make a final determination of K. Here we utilize a supervised machine learning approach in combination with a recent method for model determination, called NMFk, to determine the number of hidden features automatically. NMFk performs a set of NMF simulations on an ensemble of matrices, obtained by bootstrapping the initial dataset, and estimates which K produces stable groups of latent features that reconstruct the initial dataset well. We then train a Multi-Layter Perceptron (MLP) classifier network to determine the correct number of latent features utilizing the statistics and characteristicsmore » of the NMF solution, obtained from NMFk. In order to train the MLP classifier, a training set of 58,660 matrices with predetermined latent features were factorized with NMFk. The MLP classifier in conjunction with NMFk maintains a greater than 95% success rate when applied to a held out test set. Additionally, when applied to two well-known benchmark datasets, the swimmer and MIT face data, NMFk/MLP correctly recovers the established number of hidden features. Finally, we compare the accuracy of our method to the ARD, AIC and Stability-based methods.« less

Authors:
ORCiD logo [1]; ORCiD logo [1];  [1];  [1]; ORCiD logo [1]
  1. Los Alamos National Lab. (LANL), Los Alamos, NM (United States)
Publication Date:
Research Org.:
Los Alamos National Laboratory (LANL), Los Alamos, NM (United States)
Sponsoring Org.:
USDOE Laboratory Directed Research and Development (LDRD) Program; USDOE National Nuclear Security Administration (NNSA)
OSTI Identifier:
1716806
Report Number(s):
LA-UR-20-20994
Journal ID: ISSN 2632-2153
Grant/Contract Number:  
89233218CNA000001
Resource Type:
Accepted Manuscript
Journal Name:
Machine Learning: Science and Technology
Additional Journal Information:
Journal Volume: 2; Journal Issue: 2; Journal ID: ISSN 2632-2153
Publisher:
IOP Publishing
Country of Publication:
United States
Language:
English
Subject:
97 MATHEMATICS AND COMPUTING

Citation Formats

Nebgen, Benjamin Tyler, Vangara, Raviteja, Hombrados Herrera, Miguel Angel, Kuksova, Svetlana, and Alexandrov, Boian. A neural network for determination of latent dimensionality in Nonnegative Matrix Factorization. United States: N. p., 2020. Web. doi:10.1088/2632-2153/aba372.
Nebgen, Benjamin Tyler, Vangara, Raviteja, Hombrados Herrera, Miguel Angel, Kuksova, Svetlana, & Alexandrov, Boian. A neural network for determination of latent dimensionality in Nonnegative Matrix Factorization. United States. https://doi.org/10.1088/2632-2153/aba372
Nebgen, Benjamin Tyler, Vangara, Raviteja, Hombrados Herrera, Miguel Angel, Kuksova, Svetlana, and Alexandrov, Boian. Tue . "A neural network for determination of latent dimensionality in Nonnegative Matrix Factorization". United States. https://doi.org/10.1088/2632-2153/aba372. https://www.osti.gov/servlets/purl/1716806.
@article{osti_1716806,
title = {A neural network for determination of latent dimensionality in Nonnegative Matrix Factorization},
author = {Nebgen, Benjamin Tyler and Vangara, Raviteja and Hombrados Herrera, Miguel Angel and Kuksova, Svetlana and Alexandrov, Boian},
abstractNote = {Non-negative Matrix Factorization (NMF) has proven to be a powerful unsupervised learning method for uncovering hidden features in complex and noisy datasets with applications in data mining, text recognition, dimension reduction, face recognition, anomaly detection, blind source separation, and many other fields. An important input for NMF is the latent dimensionality of the data, that is, the number of hidden features, K, present in the explored dataset. Unfortunately, and this quantity is rarely known a priori. The existing methods for determining latent dimensionality, such as Automatic Relevance Determination (ARD), are mostly heuristic and utilize different characteristics to estimate the number of hidden features. However, all of them require human presence to make a final determination of K. Here we utilize a supervised machine learning approach in combination with a recent method for model determination, called NMFk, to determine the number of hidden features automatically. NMFk performs a set of NMF simulations on an ensemble of matrices, obtained by bootstrapping the initial dataset, and estimates which K produces stable groups of latent features that reconstruct the initial dataset well. We then train a Multi-Layter Perceptron (MLP) classifier network to determine the correct number of latent features utilizing the statistics and characteristics of the NMF solution, obtained from NMFk. In order to train the MLP classifier, a training set of 58,660 matrices with predetermined latent features were factorized with NMFk. The MLP classifier in conjunction with NMFk maintains a greater than 95% success rate when applied to a held out test set. Additionally, when applied to two well-known benchmark datasets, the swimmer and MIT face data, NMFk/MLP correctly recovers the established number of hidden features. Finally, we compare the accuracy of our method to the ARD, AIC and Stability-based methods.},
doi = {10.1088/2632-2153/aba372},
journal = {Machine Learning: Science and Technology},
number = 2,
volume = 2,
place = {United States},
year = {Tue Jul 07 00:00:00 EDT 2020},
month = {Tue Jul 07 00:00:00 EDT 2020}
}

Works referenced in this record:

Unsupervised learning of auditory filter banks using non-negative matrix factorisation
conference, March 2008

  • Bertrand, Alexander; Demuynck, Kris; Stouten, Veronique
  • 2008 IEEE International Conference on Acoustics, Speech and Signal Processing
  • DOI: 10.1109/icassp.2008.4518709