October 2023 Assigning topics to documents by successive projections
Olga Klopp, Maxim Panov, Suzanne Sigalla, Alexandre B. Tsybakov
Author Affiliations +
Ann. Statist. 51(5): 1989-2014 (October 2023). DOI: 10.1214/23-AOS2316

Abstract

Topic models provide a useful tool to organize and understand the structure of large corpora of text documents, in particular, to discover hidden thematic structure. Clustering documents from big unstructured corpora into topics is an important task in various fields, such as image analysis, e-commerce, social networks and population genetics. Since the number of topics is typically substantially smaller than the size of the corpus and of the dictionary, the methods of topic modeling can lead to a dramatic dimension reduction. We study the problem of estimating the topic-document matrix, which gives the topics distribution for each document in a given corpus, that is, we focus on the clustering aspect of the problem. We introduce an algorithm that we call Successive Projection Overlapping Clustering (SPOC) inspired by the successive projection algorithm for separable matrix factorization. This algorithm is simple to implement and computationally fast. We establish upper bounds on the performance of the SPOC algorithm for estimation of the topic-document matrix, as well as near matching minimax lower bounds. We also propose a method that achieves analogous results when the number of topics is unknown and provides an estimate of the number of topics. Our theoretical results are complemented with a numerical study on synthetic and semisynthetic data.

Funding Statement

The work of Olga Klopp was conducted as part of the project Labex MME-DII (ANR11-LBX-0023-01).
The research of Suzanne Sigalla and Alexandre B. Tsybakov was supported by the grant of French National Research Agency (ANR) “Investissements d’Avenir” LabEx Ecodec/ANR-11-LABX-0047.

Citation

Download Citation

Olga Klopp. Maxim Panov. Suzanne Sigalla. Alexandre B. Tsybakov. "Assigning topics to documents by successive projections." Ann. Statist. 51 (5) 1989 - 2014, October 2023. https://doi.org/10.1214/23-AOS2316

Information

Received: 1 August 2021; Revised: 1 July 2023; Published: October 2023
First available in Project Euclid: 14 December 2023

Digital Object Identifier: 10.1214/23-AOS2316

Subjects:
Primary: 62G05
Secondary: 60C05

Keywords: adaptive estimation , latent variable model , minimax rate of convergence , nonnegative matrix factorization , topic model

Rights: Copyright © 2023 Institute of Mathematical Statistics

Vol.51 • No. 5 • October 2023
Back to Top