Incorporating Non-local Information into Information Extraction Systems by Gibbs Sampling
Jenny Rose FinkelTrond GrenagerChristopher D. Manning
Proposes using Gibbs sampling with simulated annealing during inference to incorporate long-range consistency constraints into conditional random fields, boosting information extraction accuracy on standard benchmarks without requiring structural retraining.
Statistical models for natural language processing tasks such as information extraction have long been limited to local features because these permit efficient exact inference through dynamic programming. This constraint prevents them from capturing the long-distance regularities common in text, such as consistent labeling of repeated entity names across a document. The work addresses this gap by demonstrating a practical way to relax the locality requirement while retaining usable inference procedures.
The authors set out to evaluate whether Gibbs sampling combined with simulated annealing could serve as a drop-in replacement for Viterbi decoding in existing sequence models, thereby allowing non-local consistency constraints to be added without making inference intractable. They started from trained conditional random field models and augmented them with simple penalty-based models that discourage inconsistent entity labels or mismatched template fields. Inference was performed by running a Markov chain that repeatedly resamples individual positions conditioned on the rest of the sequence and the non-local penalties; a linear cooling schedule gradually sharpened the distribution toward high-probability assignments.
On the CoNLL 2003 named-entity recognition task the non-local model raised per-entity F1 from 85.51 to 86.86, an absolute gain of 1.35 points that corresponds to roughly a 9 percent relative error reduction. On the CMU Seminar Announcements template-filling task the same approach produced a smaller but still statistically significant improvement, moving overall token F1 from 91.85 to 92.29. Both gains were obtained against already competitive baselines and were stable across repeated runs. The method also proved flexible: the same sampling machinery worked on top of the original CRF without retraining and could enforce constraints that are difficult to encode in skip-chain or relational Markov network alternatives.
These results indicate that modest, linguistically motivated non-local constraints can measurably improve extraction accuracy on standard benchmarks without requiring entirely new model architectures. The computational price is a roughly thirty-fold increase in test-time cost, which may be acceptable when accuracy matters more than throughput. Because the technique is model-agnostic, it can be layered onto hidden Markov models, conditional Markov models, or other sequence taggers already in production.
The main limitations are the task-specific design of the penalty terms and the lack of a systematic exploration of cooling schedules or sampling budgets. Readers should therefore treat the reported gains as evidence of feasibility rather than as guaranteed improvements for every new domain. Further work to reduce sampling cost or to learn the non-local penalties automatically would strengthen the practical case for wider adoption.
- Paper: Conditional Random Fields: Probabilistic Models for Segmenting and Labeling Sequence Data, J. Lafferty et al. (2001). This foundational paper introduces Conditional Random Fields, the sequence modeling framework that the source paper augments with Gibbs sampling.
- Paper: Bidirectional LSTM-CRF Models for Sequence Tagging, Zhiheng Huang et al. (2015). This paper extends sequence tagging by replacing traditional graphical models like CRFs with bidirectional LSTM-neural architectures.
