Timezone: »

Supervised Topic Models
David Blei · Jon McAuliffe

Mon Dec 03 10:30 AM -- 10:40 AM (PST) @ None #None

We introduce supervised latent Dirichlet allocation (sLDA), a statistical model of labelled documents. The model accommodates a variety of response types. We derive a maximum-likelihood procedure for parameter estimation, which relies on variational approximations to handle intractable posterior expectations. Prediction problems motivate this research: we use the fitted model to predict response values for new documents. We test sLDA on two real-world problems: movie ratings predicted from reviews, and web page popularity predicted from text descriptions. We illustrate the benefits of sLDA versus modern regularized regression, as well as versus an unsupervised LDA analysis followed by a separate regression.

Author Information

David Blei (Columbia University)
Jon McAuliffe (UC Berkeley and Voleon)

More from the Same Authors