One Objective for All Models --- Self-supervised Learning for Topic Models

Zeping Luo, Cindy Weng, Shiyou Wu, Mo Zhou, Rong Ge

Keywords: self-supervised learning, topic models

Abstract: Self-supervised learning has significantly improved the performance of many NLP tasks. In this paper, we highlight a key advantage of self-supervised learning - when applied to data generated by topic models, self-supervised learning can be oblivious to the specific model, and hence is less susceptible to model mis-specification. In particular, we prove that commonly used self-supervised objectives based on reconstruction or contrastive samples can both recover useful posterior information for general topic models. Empirically, we show that the same objectives can perform competitively against posterior inference using the correct model, while outperforming posterior inference using mis-specified model.

One-sentence Summary: We study the self-supervised learning in the topic models setup and show that it can provide useful information about the topic posterior for general topic models.