References & Citations
Statistics > Machine Learning
Title: DOLDA - a regularized supervised topic model for high-dimensional multi-class regression
(Submitted on 31 Jan 2016 (v1), last revised 20 Oct 2016 (this version, v2))
Abstract: Generating user interpretable multi-class predictions in data rich environments with many classes and explanatory covariates is a daunting task. We introduce Diagonal Orthant Latent Dirichlet Allocation (DOLDA), a supervised topic model for multi-class classification that can handle both many classes as well as many covariates. To handle many classes we use the recently proposed Diagonal Orthant (DO) probit model (Johndrow et al., 2013) together with an efficient Horseshoe prior for variable selection/shrinkage (Carvalho et al., 2010). We propose a computationally efficient parallel Gibbs sampler for the new model. An important advantage of DOLDA is that learned topics are directly connected to individual classes without the need for a reference class. We evaluate the model's predictive accuracy on two datasets and demonstrate DOLDA's advantage in interpreting the generated predictions.
Submission history
From: Måns Magnusson [view email][v1] Sun, 31 Jan 2016 15:33:10 GMT (368kb,D)
[v2] Thu, 20 Oct 2016 14:52:53 GMT (662kb,D)
Link back to: arXiv, form interface, contact.