Semi-supervised Nonnegative Matrix Factorization for Document Classification
Semi-supervised Nonnegative Matrix Factorization for Document Classification
复制标题
DOI:
10.1109/ieeeconf53345.2021.9723109
复制
发表时间:
2021-10
期刊:
影响因子:
--
通讯作者:
Jamie Haddock;Lara Kassab;Sixian Li;Alona Kryshchenko;Rachel Grotheer;Elena Sizikova;Chuntian Wang;Thomas Merkh;R. W. M. A. Madushani;Miju Ahn;D. Needell;Kathryn Leonard
中科院分区:
文献类型:
--
作者:
Jamie Haddock;Lara Kassab;Sixian Li;Alona Kryshchenko;Rachel Grotheer;Elena Sizikova;Chuntian Wang;Thomas Merkh;R. W. M. A. Madushani;Miju Ahn;D. Needell;Kathryn Leonard
We propose new semi-supervised nonnegative matrix factorization (SSNMF) models for document classification and provide motivation for these models as maximum likelihood estimators. The proposed SSNMF models simultaneously provide both a topic model and a model for classification, thereby offering highly interpretable classification results. We derive training methods using multiplicative updates for each new model, and demonstrate the application of these models to single-label and multi-label document classification, although the models are flexible to other supervised learning tasks such as regression. We illustrate the promise of these models and training methods on document classification datasets (e.g., 20 Newsgroups, Reuters).