CRF-OPT: an efficient high-quality conditional random field solver

  • Authors:
  • Minmin Chen;Yixin Chen;Michael R. Brent

  • Affiliations:
  • Department of Computer Science and Engineering, Washington University in St. Louis, St. Louis, MO;Department of Computer Science and Engineering, Washington University in St. Louis, St. Louis, MO;Department of Computer Science and Engineering, Washington University in St. Louis, St. Louis, MO

  • Venue:
  • AAAI'08 Proceedings of the 23rd national conference on Artificial intelligence - Volume 2
  • Year:
  • 2008

Quantified Score

Hi-index 0.00

Visualization

Abstract

Conditional random field (CRF) is a popular graphical model for sequence labeling. The flexibility of CRF poses significant computational challenges for training. Using existing optimization packages often leads to long training time and unsatisfactory results. In this paper, we develop CRFOPT, a general CRF training package, to improve the efficiency and quality for training CRFs. We propose two improved versions of the forward-backward algorithm that exploit redundancy and reduce the time by several orders of magnitudes. Further, we propose an exponential transformation that enforces sufficient step sizes for quasi-Newton methods. The technique improves the convergence quality, leading to better training results. We evaluate CRF-OPT on a gene prediction task on pathogenic DNA sequences, and show that it is faster and achieves better prediction accuracy than both the HMM models and the original CRF model without exponential transformation.