In statistics, latent Dirichlet allocation (LDA) is a generative model that allows sets of observations to be explained by unobserved groups which explain why some parts of the data are similar. For example, if observations are words collected into documents, it posits that each document is a mixture of a small number of topics and that each word's creation is attributable to one of the document's topics. LDA was first presented as a graphical model for topic discovery and was developed by David Blei, Andrew Ng, and Michael Jordan in 2002.[1]
Contents |
Topics in LDA
In LDA, each document may be viewed as a mixture of various topics. This is similar to probabilistic latent semantic analysis (pLSA), except that in LDA the topic distribution is assumed to have a Dirichlet prior. In practice, this results in more reasonable mixtures of topics in a document. It has been noted, however, that the pLSA model is equivalent to the LDA model under a uniform Dirichlet prior distribution.[2]
For example, an LDA model might have topics CAT and DOG. The CAT topic has probabilities of generating various words: the words milk, meow, kitten and of course cat will have high probability given this topic. The DOG topic likewise has probabilities of generating each word: puppy, bark and bone might have high probability. Words without special relevance, like the (see function word), will have roughly even probability between classes (or can be placed into a separate category).
A document is generated by picking a distribution over topics (ie, mostly about DOG, mostly about CAT, or a bit of both), and given this distribution, picking the topic of each specific word. Then words are generated given their topics. (Notice that words are considered to be independent given the topics. This is a standard bag of words model assumption, and makes the individual words exchangeable.)
Model
With plate notation, the dependencies among the many variables can be captured concisely. α is the parameter of the uniform Dirichlet prior on the per-document topic distributions. β is the parameter of the uniform Dirichlet prior on the per-topic word distribution. θi is the topic distribution for document i, zij is the topic for the jth word in document i, and wij is the specific word. The wij are the only observable variables, and the other variables are latent variables.
Mostly, the basic LDA model will be extended to a smoothed version to gain better result. The plate notation is shown in the right, where K denotes the number of topics considered in the model and φ is a K*V (V is the dimension of the vocabulary) Markov matrix each line of which denotes the word distribution of a topic.
Inference
Learning the various distributions (the set of topics, their associated word probabilities, the topic of each word, and the particular topic mixture of each document) is a problem of Bayesian inference. The original paper used a variational Bayes approximation of the posterior distribution[1]; alternative inference techniques use Gibbs sampling[3] and expectation propagation[4].
Followed is the derivation of the equations for collapsed Gibbs sampling, which means
s and θs will be integrated out.
According to the model, the total probability of the model is:
where the bold-font variables denote the vector version of the variables. First of all,
and
need to be integrated out.
Note that all the θs are independent to each other and the same to all the
s. So we can treat each θ and each
separately. We now focus only on the θ part.
We can further focus on only one θ as the following:
Actually, it is the hidden part of the model for the jth document. Now we replace the probabilities in the above equation by the true distribution expression to write out the explicit equation.
Let
be the number of word tokens in the jth document with the same word symbol (the rth word in the vocabulary) assigned to the ith topic. So,
is three dimensional. If any of the three dimensions is not limited to a specific value, we use a parenthesized point
to denote. For example,
denotes the number of word tokens in the jth document assigned to the ith topic. Thus, the right most part of the above equation can be rewritten as:
So the θj integration formula can be changed to:
Clearly, the equation inside the integration has the same form as the Dirichlet distribution. According to the Dirichlet distribution,
Thus,
Now we turn our attentions to the
part. Actually, the derivation of the
part is very similar to the
part. Here we only list the steps of the derivation:
For clarity, here we write down the final equation with both
and
integrated out:
The goal of Gibbs Sampling here is to approximate the distribution of
. Since
is invariable for any of Z, Gibbs Sampling equations can be derived from
directly. The key point is to derive the following conditional probability:
where Z(m,n) denotes the Z hidden variable of the nth word token in the mth document. And further we assume that the word symbol of it is the vth word in the vocabulary.
denotes all the Zs but Z(m,n). Note that Gibbs Sampling needs only to sample a value for Z(m,n), according to the above probability, we do not need the exact value of
but the ratios among the probabilities that Z(m,n) can take value. So, the above equation can be simplified as:
Finally, let
be the same meaning as
but with the Z(m,n) excluded. The above equation can be further derived to be an amazingly simple form:
Applications, extensions and similar techniques
Topic modeling is a classic problem in information retrieval. Related models and techniques are, among others, latent semantic indexing, probabilistic latent semantic indexing, non-negative matrix factorization, Gamma-Poisson.
The LDA model is highly modular and can therefore be easily extended, the main field of interest being the modeling of relations between topics. This is achieved e.g. by using another distribution on the simplex instead of the Dirichlet. The Correlated Topic Model[5] follows this approach, inducing a correlation structure between topics by using the logistic normal distribution instead of the Dirichlet. Another extension is the hierarchical LDA (hLDA)[6], where topics are joined together in a hierarchy by using the Nested Chinese Restaurant Process.
As noted earlier, PLSA is similar to LDA. The LDA model is essentially the Bayesian version of PLSA model. Bayesian formulation tends to perform better on small datasets because Bayesian methods can avoid overfitting the data. In a very large dataset, the results are probably the same. One difference is that PLSA uses a variable d to represent a document in the training set. So in PLSA, when presented with a document the model hasn't seen before, we fix
--the probability of words under topics--to be that learned from the training set and use the same EM algorithm to infer
--the topic distribution under d. Blei argues that this step is cheating because you are essentially refitting the model to the new data.
See also
Notes
- ^ a b Blei, David M.; Ng, Andrew Y.; Jordan, Michael I (January 2003). "Latent Dirichlet allocation". Journal of Machine Learning Research 3: pp. 993–1022. doi:10.1162/jmlr.2003.3.4-5.993 (inactive 2009-03-30). http://jmlr.csail.mit.edu/papers/v3/blei03a.html.
- ^ Girolami, Mark; Kaban, A. (2003). "On an Equivalence between PLSI and LDA". Proceedings of SIGIR 2003. New York: Association for Computing Machinery. ISBN 1581136463. http://www.cs.bham.ac.uk/~axk/sigir2003_mgak.pdf.
- ^ Griffiths, Thomas L.; Steyvers, Mark (April 6 2004). "Finding scientific topics". Proceedings of the National Academy of Sciences 101 (Suppl. 1): 5228–5235. doi:. PMID 14872004.
- ^ Minka, Thomas; Lafferty, John (2002). "Expectation-propagation for the generative aspect model". Proceedings of the 18th Conference on Uncertainty in Artificial Intelligence. San Francisco, CA: Morgan Kaufmann. ISBN 1-55860-897-4. https://research.microsoft.com/~minka/papers/aspect/minka-aspect.pdf.
- ^ Blei, David M.; Lafferty, John D. (2006). "Correlated topic models". Advances in Neural Information Processing Systems 18. http://www.cs.cmu.edu/~lafferty/pub/ctm.pdf.
- ^ Blei, David M.; Jordan, Michael I.; Griffiths, Thomas L.; Tenenbaum; Joshua B (2004). "Hierarchical Topic Models and the Nested Chinese Restaurant Process". Advances in Neural Information Processing Systems 16: Proceedings of the 2003 Conference. MIT Press. ISBN 0262201526. http://cocosci.berkeley.edu/tom/papers/ncrp.pdf.
External links
- LDA implementation in Java, part of the MAchine Learning for LanguagE Toolkit (MALLET).
- LDA implementation in MATLAB, by M. Steyvers and T. Griffiths.
- LDA implementation in C, by D. Blei.
- LDA implementations in C and matlab.
- LDA implementation in C++ using Gibbs Sampling.
This entry is from Wikipedia, the leading user-contributed encyclopedia. It may not have been reviewed by professional editors (see full disclaimer)


















