中文

AT-BERT:用于缩略语识别的对抗训练 BERT——SDU@AAAI-21 夺冠方案

计算与语言 2021-01-13 v2

摘要

缩略语识别致力于找出缩略语及其所缩写的短语,这对科学文献理解任务至关重要。然而,人工标注数据集规模有限阻碍了该问题的进一步改进。在大型语料上预训练的语言模型近期取得的突破清楚地表明,无监督预训练可极大提升下游任务性能。本文中,我们提出一种称为 AT-BERT 的对抗训练 BERT 方法,即我们在 AAAI 2021 科学文献理解(SDU)挑战赛缩略语识别任务上的夺冠方案。具体而言,采用预训练 BERT 以捕获更好的语义表示。随后我们将 FGM 对抗训练策略融入 BERT 微调,使模型更鲁棒且具泛化性。此外,设计了一种集成机制以融合从多个 BERT 变体学到的表示。综合上述所有组件,在 SciAI 数据集上的实验结果表明,我们所提方法优于其他所有具竞争力的先进方法。

关键词

引用

@article{arxiv.2101.03700,
  title  = {AT-BERT: Adversarial Training BERT for Acronym Identification Winning Solution for SDU@AAAI-21},
  author = {Danqing Zhu and Wangli Lin and Yang Zhang and Qiwei Zhong and Guanxiong Zeng and Weilin Wu and Jiayu Tang},
  journal= {arXiv preprint arXiv:2101.03700},
  year   = {2021}
}

备注

Accepted to SDU @ AAAI 2021, 8 pages, 3 figures