Skip to main content
QUICK REVIEW

[论文解读] An Analysis of Simple Data Augmentation for Named Entity Recognition

Xiang Dai, Heike Adel|arXiv (Cornell University)|Oct 22, 2020
Topic Modeling参考文献 37被引用 9
一句话总结

本文研究了命名实体识别(NER)任务中简单数据增强技术的应用,NER是一项基于标记的序列标注任务。研究将同义词替换、提及替换和词语重排等方法适配至NER任务中,以保持实体标签的完整性。在生物医学和材料科学数据集上的实验表明,这些技术显著提升了模型性能,尤其在小规模训练集上,F1分数最高提升了4.4个百分点,超过强基线的BERT模型。

ABSTRACT

Simple yet effective data augmentation techniques have been proposed for sentence-level and sentence-pair natural language processing tasks. Inspired by these efforts, we design and compare data augmentation for named entity recognition, which is usually modeled as a token-level sequence labeling problem. Through experiments on two data sets from the biomedical and materials science domains (i2b2-2010 and MaSciP), we show that simple augmentation can boost performance for both recurrent and transformer-based models, especially for small training sets.

研究动机与目标

  • 解决生物医学和材料科学等专业领域中命名实体识别的低资源挑战,这些领域标注数据稀缺且标注成本高昂。
  • 探究以往在句子级自然语言处理任务中表现良好的简单非生成式数据增强技术,是否可有效适配至基于标记的NER任务。
  • 评估不同增强策略在循环神经网络与Transformer模型上的有效性,尤其在数据稀缺条件下的表现。
  • 确定数据增强是否能在强预训练模型(如BERT)上带来性能提升,特别是在训练数据有限的情况下。

提出的方法

  • 将源自句子级NLP的四种数据增强技术适配至NER任务:同义词替换、提及替换、段落内词语重排,以及按标签类别进行的标记替换。
  • 在标记级别应用变换操作,同时保持原始实体标签不变,确保增强后的样本仍符合序列标注任务的规范。
  • 使用WordNet进行同义词替换,并通过实体类型一致性控制提及替换,以维持语义和标签的有效性。
  • 在句子段落(如从句或短语)内部实施词语重排,生成语法上合理的变体,同时不改变标签。
  • 将所有增强方法整合为单一策略,以提升训练数据的多样性,同时监控标签一致性。
  • 在增强数据和原始数据上分别训练双向LSTM与BERT-based模型,并使用基于实体跨度的F1分数评估性能。

实验结果

研究问题

  • RQ1能否将原本为句子级NLP设计的简单非生成式数据增强技术,有效适配至基于标记的NER任务?
  • RQ2在小规模训练集上,哪种数据增强策略能带来最高的性能提升?
  • RQ3结合多种增强技术是否能带来比单一方法更好的泛化性能?
  • RQ4与从零开始训练的模型相比,这些增强技术在强预训练Transformer模型(如BERT)上的有效性如何?
  • RQ5当训练数据被限制在50或150个样本时,数据增强是否能缓解低资源环境下的性能下降?

主要发现

  • 所有数据增强技术均在循环神经网络和Transformer模型上优于基线模型,且在小规模训练集(如50个样本)上提升最为显著。
  • 在基于BERT的模型上,同义词替换带来了最高的平均F1提升(4.4个百分点);而在循环模型上,提及替换效果最佳。
  • 将所有增强方法组合使用可获得最佳平均性能,使循环模型的F1提升3.3个百分点,Transformer模型提升1.8个百分点。
  • 在完整训练集上,部分增强方法反而导致性能轻微下降,表明增强数据的多样性与样本有效性之间存在权衡。
  • 性能提升最大的情况出现在训练数据极少时,例如在MaSciP数据集上,仅使用50个训练样本时,F1分数最高提升了4.4个百分点。
  • 即使在性能优异的BERT-based模型上,数据增强仍带来了可测量的性能增益,表明数据增强在大规模预训练模型中依然具有显著价值。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。