vix.ing · top · new · best · stats · spec

AIN: Fast and Accurate Sequence Labeling with Approximate Inference Network

2020/09/17 by Xinyu Wang, Wang, Xinyu, Yong Jiang +11
Computer Science · #Artificial Intelligence (cs.AI) #Computation and Language (cs.CL) #FOS: Computer and information sciences #Handwritten Text Recognition Techniques #Music and Audio Processing #Neural Networks and Applications

paper · pdf · doi:10.48550/arxiv.2009.08229

openalex publication_date 2020/09/17 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

The linear-chain Conditional Random Field (CRF) model is one of the most widely-used neural sequence labeling approaches. Exact probabilistic inference algorithms such as the forward-backward and Viterbi algorithms are typically applied in training and prediction stages of the CRF model. However, these algorithms require sequential computation that makes parallelization impossible. In this paper, we propose to employ a parallelizable approximate variational inference algorithm for the CRF model. Based on this algorithm, we design an approximate inference network that can be connected with the encoder of the neural CRF model to form an end-to-end network, which is amenable to parallelization for faster training and prediction. The empirical results show that our proposed approaches achieve a 12.7-fold improvement in decoding speed with long sentences and a competitive accuracy compared with the traditional CRF approach.

Citations

Related