Conditional random field

Conditional random fields (CRFs) are a class of statistical modeling methods often applied in pattern recognition and machine learning and used for structured prediction. Whereas a classifier predicts a label for a single sample without considering "neighbouring" samples, a CRF can take context into account. To do so, the predictions are modelled as a graphical model, which represents the presence of dependencies between the predictions. What kind of graph is used depends on the application. For example, in natural language processing, "linear chain" CRFs are popular, for which each prediction is dependent only on its immediate neighbours. In image processing, the graph typically connects locations to nearby and/or similar locations to enforce that they receive similar predictions.

Other examples where CRFs are used are: labeling or parsing of sequential data for natural language processing or biological sequences,^[1] part-of-speech tagging, shallow parsing,^[2] named entity recognition,^[3] gene finding, peptide critical functional region finding,^[4] and object recognition^[5] and image segmentation in computer vision.^[6]

Description

CRFs are a type of discriminative undirected probabilistic graphical model.

Lafferty, McCallum and Pereira^[1] define a CRF on observations ${\boldsymbol {X}}$ and random variables ${\boldsymbol {Y}}$ as follows:

Let $G=(V,E)$ be a graph such that ${\boldsymbol {Y}}=({\boldsymbol {Y}}_{v})_{v\in V}$ , so that ${\boldsymbol {Y}}$ is indexed by the vertices of $G$ .
Then $({\boldsymbol {X}},{\boldsymbol {Y}})$ is a conditional random field when each random variable ${\boldsymbol {Y}}_{v}$ , conditioned on ${\boldsymbol {X}}$ , obeys the Markov property with respect to the graph; that is, its probability is dependent only on its neighbours in G:
$P({\boldsymbol {Y}}_{v}|{\boldsymbol {X}},\{{\boldsymbol {Y}}_{w}:w\neq v\})=P({\boldsymbol {Y}}_{v}|{\boldsymbol {X}},\{{\boldsymbol {Y}}_{w}:w\sim v\})$ , where ${\mathit {w}}\sim v$ means that $w$ and $v$ are neighbors in $G$ .

What this means is that a CRF is an undirected graphical model whose nodes can be divided into exactly two disjoint sets ${\boldsymbol {X}}$ and ${\boldsymbol {Y}}$ , the observed and output variables, respectively; the conditional distribution $p({\boldsymbol {Y}}|{\boldsymbol {X}})$ is then modeled.

Inference

For general graphs, the problem of exact inference in CRFs is intractable. The inference problem for a CRF is basically the same as for an MRF and the same arguments hold.^[7] However, there exist special cases for which exact inference is feasible:

If the graph is a chain or a tree, message passing algorithms yield exact solutions. The algorithms used in these cases are analogous to the forward-backward and Viterbi algorithm for the case of HMMs.
If the CRF only contains pair-wise potentials and the energy is submodular, combinatorial min cut/max flow algorithms yield exact solutions.

If exact inference is impossible, several algorithms can be used to obtain approximate solutions. These include:

Loopy belief propagation
Alpha expansion
Mean field inference
Linear programming relaxations

Parameter Learning

Learning the parameters $\theta$ is usually done by maximum likelihood learning for $p(Y_{i}|X_{i};\theta )$ . If all nodes have exponential family distributions and all nodes are observed during training, this optimization is convex.^[7] It can be solved for example using gradient descent algorithms, or Quasi-Newton methods such as the L-BFGS algorithm. On the other hand, if some variables are unobserved, the inference problem has to be solved for these variables. Exact inference is intractable in general graphs, so approximations have to be used.

Examples

In sequence modeling, the graph of interest is usually a chain graph. An input sequence of observed variables $X$ represents a sequence of observations and $Y$ represents a hidden (or unknown) state variable that needs to be inferred given the observations. The $Y_{i}$ are structured to form a chain, with an edge between each $Y_{i-1}$ and $Y_{i}$ . As well as having a simple interpretation of the $Y_{i}$ as "labels" for each element in the input sequence, this layout admits efficient algorithms for:

model training, learning the conditional distributions between the $Y_{i}$ and feature functions from some corpus of training data.
decoding, determining the probability of a given label sequence $Y$ given $X$ .
inference, determining the most likely label sequence $Y$ given $X$ .

The conditional dependency of each $Y_{i}$ on $X$ is defined through a fixed set of feature functions of the form $f(i,Y_{i-1},Y_{i},X)$ , which can be thought of as measurements on the input sequence that partially determine the likelihood of each possible value for $Y_{i}$ . The model assigns each feature a numerical weight and combines them to determine the probability of a certain value for $Y_{i}$ .

Linear-chain CRFs have many of the same applications as conceptually simpler hidden Markov models (HMMs), but relax certain assumptions about the input and output sequence distributions. An HMM can loosely be understood as a CRF with very specific feature functions that use constant probabilities to model state transitions and emissions. Conversely, a CRF can loosely be understood as a generalization of an HMM that makes the constant transition probabilities into arbitrary functions that vary across the positions in the sequence of hidden states, depending on the input sequence.

Notably, in contrast to HMMs, CRFs can contain any number of feature functions, the feature functions can inspect the entire input sequence $X$ at any point during inference, and the range of the feature functions need not have a probabilistic interpretation.

Variants

Higher-order CRFs and semi-Markov CRFs

CRFs can be extended into higher order models by making each $Y_{i}$ dependent on a fixed number $k$ of previous variables $Y_{i-k},...,Y_{i-1}$ . In conventional formulations of higher order CRFs, training and inference are only practical for small values of $k$ (such as k ≤ 5),^[8] since their computational cost increases exponentially with $k$ .

However, another recent advance has managed to ameliorate these issues by leveraging concepts and tools from the field of Bayesian nonparametrics. Specifically, the CRF-infinity approach^[9] constitutes a CRF-type model that is capable of learning infinitely-long temporal dynamics in a scalable fashion. This is effected by introducing a novel potential function for CRFs that is based on the Sequence Memoizer (SM), a nonparametric Bayesian model for learning infinitely-long dynamics in sequential observations.^[10] To render such a model computationally tractable, CRF-infinity employs a mean-field approximation^[11] of the postulated novel potential functions (which are driven by an SM). This allows for devising efficient approximate training and inference algorithms for the model, without undermining its capability to capture and model temporal dependencies of arbitrary length.

There exists another generalization of CRFs, the semi-Markov conditional random field (semi-CRF), which models variable-length segmentations of the label sequence $Y$ .^[12] This provides much of the power of higher-order CRFs to model long-range dependencies of the $Y_{i}$ , at a reasonable computational cost.

Finally, large-margin models for structured prediction, such as the structured Support Vector Machine can be seen as an alternative training procedure to CRFs.

Latent-dynamic conditional random field

Latent-dynamic conditional random fields (LDCRF) or discriminative probabilistic latent variable models (DPLVM) are a type of CRFs for sequence tagging tasks. They are latent variable models that are trained discriminatively.

In an LDCRF, like in any sequence tagging task, given a sequence of observations x = $x_{1},\dots ,x_{n}$ , the main problem the model must solve is how to assign a sequence of labels y = $y_{1},\dots ,y_{n}$ from one finite set of labels $Y$ . Instead of directly modeling $P$ (y|x) as an ordinary linear-chain CRF would do, a set of latent variables h is "inserted" between x and y using the chain rule of probability:^[13]

P(\mathbf {y} |\mathbf {x} )=\sum _{\mathbf {h} }P(\mathbf {y} |\mathbf {h} ,\mathbf {x} )P(\mathbf {h} |\mathbf {x} )

This allows capturing latent structure between the observations and labels.^[14] While LDCRFs can be trained using quasi-Newton methods, a specialized version of the perceptron algorithm called the latent-variable perceptron has been developed for them as well, based on Collins' structured perceptron algorithm.^[13] These models find applications in computer vision, specifically gesture recognition from video streams^[14] and shallow parsing.^[13]

References

^ ^a ^b Lafferty, J.; McCallum, A.; Pereira, F. (2001). "Conditional random fields: Probabilistic models for segmenting and labeling sequence data". Proc. 18th International Conf. on Machine Learning. Morgan Kaufmann. pp. 282–289.
^ Sha, F.; Pereira, F. (2003). shallow parsing with conditional random fields.
^ Settles, B. (2004). "Biomedical named entity recognition using conditional random fields and rich feature sets" (PDF). Proceedings of the International Joint Workshop on Natural Language Processing in Biomedicine and its Applications. pp. 104–107.
^ Chang KY; Lin T-p; Shih L-Y; Wang C-K (2015). "Analysis and Prediction of the Critical Regions of Antimicrobial Peptides Based on Conditional Random Fields". PLOS ONE. 10 (3): e0119490. Bibcode:2015PLoSO..1019490C. doi:10.1371/journal.pone.0119490. PMC 4372350. PMID 25803302.
^ J.R. Ruiz-Sarmiento; C. Galindo; J. Gonzalez-Jimenez (2015). "UPGMpp: a Software Library for Contextual Object Recognition.". 3rd. Workshop on Recognition and Action for Scene Understanding (REACTS).
^ He, X.; Zemel, R.S.; Carreira-Perpinñán, M.A. (2004). "Multiscale conditional random fields for image labeling". IEEE Computer Society. CiteSeerX 10.1.1.3.7826.
^ ^a ^b Sutton, Charles; McCallum, Andrew (2010). "An Introduction to Conditional Random Fields". arXiv:1011.4088v1 [stat.ML].
^ Lavergne, Thomas; Yvon, François (September 7, 2017). "Learning the Structure of Variable-Order CRFs: a Finite-State Perspective". Proceedings of the 2017 Conference on Empirical Methods in Natural Language Processing. Copenhagen, Denmark: Association for Computational Linguistics. p. 433.
^ Chatzis, Sotirios; Demiris, Yiannis (2013). "The Infinite-Order Conditional Random Field Model for Sequential Data Modeling". IEEE Transactions on Pattern Analysis and Machine Intelligence. 35 (6): 1523–1534. doi:10.1109/tpami.2012.208. hdl:10044/1/12614. PMID 23599063. S2CID 690627.
^ Gasthaus, Jan; Teh, Yee Whye (2010). "Improvements to the Sequence Memoizer" (PDF). Proc. NIPS.
^ Celeux, G.; Forbes, F.; Peyrard, N. (2003). "EM Procedures Using Mean Field-Like Approximations for Markov Model-Based Image Segmentation". Pattern Recognition. 36 (1): 131–144. Bibcode:2003PatRe..36..131C. CiteSeerX 10.1.1.6.9064. doi:10.1016/s0031-3203(02)00027-4.
^ Sarawagi, Sunita; Cohen, William W. (2005). "Semi-Markov conditional random fields for information extraction". In Lawrence K. Saul; Yair Weiss; Léon Bottou (eds.). Advances in Neural Information Processing Systems 17. Cambridge, MA: MIT Press. pp. 1185–1192. Archived from the original (PDF) on 2019-11-30. Retrieved 2015-11-12.
^ ^a ^b ^c Xu Sun; Takuya Matsuzaki; Daisuke Okanohara; Jun'ichi Tsujii (2009). Latent Variable Perceptron Algorithm for Structured Classification. IJCAI. pp. 1236–1242. Archived from the original on 2018-12-06. Retrieved 2018-12-06.
^ ^a ^b Morency, L. P.; Quattoni, A.; Darrell, T. (2007). "Latent-Dynamic Discriminative Models for Continuous Gesture Recognition" (PDF). 2007 IEEE Conference on Computer Vision and Pattern Recognition. p. 1. CiteSeerX 10.1.1.420.6836. doi:10.1109/CVPR.2007.383299. ISBN 978-1-4244-1179-5. S2CID 7117722.