Contrastive estimation reveals topic posterior information to linear models
Authors: Christopher Tosh, Akshay Krishnamurthy, Daniel Hsu
Keywords: contrastive learning, self-supervised learning, representation learning, theory
Abstract: Contrastive learning is an approach to representation learning that utilizes naturally occurring similar and dissimilar pairs of data points to find useful embeddings of data. In the context of document classification under topic modeling assumptions, we prove that contrastive learning is capable of recovering a representation of documents that reveals their underlying topic posterior information to linear models. We apply this procedure in a semi-supervised setup and demonstrate empirically that linear classifiers with these representations perform well in document classification tasks with very few training examples.