3 个月前

并行迭代编辑模型用于局部序列转换

并行迭代编辑模型用于局部序列转换

摘要

我们提出了一种并行迭代编辑(Parallel Iterative Edit, PIE)模型,用于解决诸如语法错误纠正(Grammatical Error Correction, GEC)等任务中出现的局部序列转换问题。现有的方法通常基于流行的编码器-解码器(Encoder-Decoder, ED)模型进行序列到序列的学习。ED模型通过自回归方式捕捉输出标记之间的完整依赖关系,但因采用串行解码而效率较低。相比之下,PIE模型采用并行解码,虽放弃了对输出端完整依赖关系建模的优势,却在四个关键方面实现了与ED模型相当甚至更优的准确率:(1)预测编辑操作而非直接生成标记;(2)对序列进行标签化而非从头生成序列;(3)通过迭代方式逐步优化预测结果,以捕捉潜在的依赖关系;(4)将 logits 分解为编辑操作及其对应标记参数的组合,从而有效利用 BERT 等预训练语言模型的表示能力。在涵盖语法错误纠正、光学字符识别(OCR)纠错和拼写纠正等多个任务上的实验表明,PIE 模型在保持高准确率的同时,显著提升了推理速度,是一种高效且精确的局部序列转换替代方案。

代码仓库

基准测试

基准方法指标
grammatical-error-correction-on-conll-2014Sequence Labeling with edits using BERT, Faster inference
F0.5: 61.2
grammatical-error-correction-on-conll-2014Sequence Labeling with edits using BERT, Faster inference (Single Model)
F0.5: 59.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
并行迭代编辑模型用于局部序列转换 | 论文 | HyperAI超神经