Luís A. Nunes Amaral
Professor of Engineering Sciences and Applied Mathematics
Professor of Medicine (by courtesy)
Professor of Molecular Biosciences (by courtesy)
Professor of Physics & Astronomy (by courtesy)
Chemical & Biological Engineering
2145 Sheridan Road (Room E136)
Evanston, IL 60208, US
Phone:
(847) 491-7850A new evaluation framework for topic modeling algorithms based on synthetic corpora
Proceedings of the 22nd International Conference on Artificial Intelligence and Statistics (AISTATS 2019) 89, 816-826 (2019)
Abstract
Topic models are in widespread use in natural language processing and beyond. Here, we propose a new framework for the evaluation of probabilistic topic modeling algorithms based on synthetic corpora containing an unambiguously defined ground truth topic structure. The major innovation of our approach is the ability to quantify the agreement between the planted and inferred topic structures by comparing the assigned topic labels at the level of the tokens. In experiments, our approach yields novel insights about the relative strengths of topic models as corpus characteristics vary, and the first evidence of an "undetectable phase" for topic models when the planted structure is weak. We also establish the practical relevance of the insights gained for synthetic corpora by predicting the performance of topic modeling algorithms in classification tasks in real-world corpora.