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
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