Text Classification from Labeled and Unlabeled Documents using EM
K. NigamA. McCallumS. ThrunTom Michael Mitchell
Demonstrates how combining Expectation-Maximization with naive Bayes leverages abundant unlabeled text to significantly reduce classification error and labeled data requirements, while introducing practical extensions to address violated generative model assumptions.
This paper addresses the high cost of obtaining labeled training examples for statistical text classifiers, which must be hand-labeled by people, while unlabeled documents are often easy to collect in large quantities from sources such as the web, newsgroups, and news feeds. The work evaluates whether a large pool of unlabeled documents can be combined with a small number of labeled ones to produce more accurate classifiers.
The authors combine the Expectation-Maximization algorithm with a naive Bayes generative model to iteratively assign probabilistic labels to the unlabeled documents and re-estimate classifier parameters until convergence. They test the approach on three real-world collections—20 Newsgroups articles, university web pages, and Reuters newswire stories—using training sets that range from a few dozen to several thousand labeled documents and up to 10,000 unlabeled ones. They also examine two practical extensions: a tunable weight on the unlabeled data and the use of multiple mixture components per class.
The experiments show that unlabeled data can reduce classification error by up to 30 percent. With only 300 labeled documents on 20 Newsgroups, accuracy rises from 52 percent to 66 percent; with just 40 labeled documents, accuracy improves from 27 percent to 43 percent. In one case the method achieves the same 70 percent accuracy target with roughly one-third as many labeled examples. When the data fit the model assumptions well, basic EM works reliably; when the assumptions are violated, performance can drop slightly, but the two extensions largely eliminate the degradation while preserving most of the gains.
These results matter because they demonstrate a practical way to lower the labeling burden that currently limits the deployment of text classification systems in many domains. The approach is especially valuable in settings where only a few dozen or a few hundred labeled examples can be obtained.
The authors recommend applying the basic EM procedure when labeled data are scarce, selecting the unlabeled-data weight and the number of mixture components per class by cross-validation on the labeled set, and falling back to ordinary naive Bayes when abundant labeled data are already available. They note that the method’s success depends on how closely the data match the assumed mixture model and that cross-validation can under-select components when the labeled set is extremely small; additional labeled data or more robust model-selection techniques would increase reliability in those edge cases.
- Paper: A comparison of event models for naive bayes text classification, Andrew McCallum et al. (1998). Reading this comparison of multinomial and binary event models for naive Bayes text classification provides essential baseline context for the generative text model used in the source.
- Paper: Latent Dirichlet Allocation, David M. Blei et al. (2003). This work generalizes document modeling beyond class-conditional mixture components to Latent Dirichlet Allocation, representing a major evolution in generative text corpora analysis.
