Learning Supervised Topic Models for Classification and Regression from Crowds

AUTHORS:

Filipe Rodrigues (rodr [at] dtu.dk)
Mariana Lourenço
Bernardete Ribeiro
Francisco Câmara Pereira

ABSTRACT:

The growing need to analyze large collections of documents has led to great developments in topic modeling. Since documents are frequently associated with other related variables, such as labels or ratings, much interest has been placed on supervised topic models. However, the nature of most annotation tasks, prone to ambiguity and noise, often with high volumes of documents, deem learning under a single-annotator assumption unrealistic or unpractical for most real-world applications. In this article, we propose two supervised topic models, one for classification and another for regression problems, which account for the heterogeneity and biases among different annotators that are encountered in practice when learning from crowds. We develop an efficient stochastic variational inference algorithm that is able to scale to very large datasets, and we empirically demonstrate the advantages of the proposed model over state-of-the-art approaches.

JOURNAL:

IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2017

DOI:

10.1109/TPAMI.2017.2648786

SOURCE CODE:

DATASETS:

  • 20newsgroups (classification; simulated annotators)
  • Reuters (classification; annotations from Amazon Mechanical Turk)
  • LabelMe (classification; annotations from Amazon Mechanical Turk)
  • we8there (regression; simulated annotators)
  • MovieReviews (regression; annotations from Amazon Mechanical Turk)