中文

AIN:基于近似推断网络实现快速准确的序列标注

计算与语言 2020-10-13 v2 人工智能

摘要

线性链条件随机场(CRF)模型是应用最广泛的神经序列标注方法之一。CRF模型在训练与预测阶段通常采用前向-后向算法和Viterbi算法等精确概率推断算法。然而,这些算法需要顺序计算,无法并行化。本文提出将一种可并行化的近似变分推断算法用于CRF模型。基于该算法,我们设计了一个近似推断网络,可与神经CRF模型的编码器连接形成端到端网络,便于并行化以加速训练与预测。实验结果表明,相较于传统CRF方法,我们所提方法在长句上解码速度提升12.7倍且精度具有竞争力。

关键词

引用

@article{arxiv.2009.08229,
  title  = {AIN: Fast and Accurate Sequence Labeling with Approximate Inference Network},
  author = {Xinyu Wang and Yong Jiang and Nguyen Bach and Tao Wang and Zhongqiang Huang and Fei Huang and Kewei Tu},
  journal= {arXiv preprint arXiv:2009.08229},
  year   = {2020}
}

备注

Accept to Main Conference of EMNLP 2020 (Short). Camera-ready, 8 Pages