Autoregressive Structured Prediction with Language Models

26 Oct 2022  ยท  Tianyu Liu, Yuchen Jiang, Nicholas Monath, Ryan Cotterell, Mrinmaya Sachan ยท

Recent years have seen a paradigm shift in NLP towards using pretrained language models ({PLM}) for a wide range of tasks. However, there are many difficult design decisions to represent structures (e.g. tagged text, coreference chains) in a way such that they can be captured by PLMs. Prior work on structured prediction with PLMs typically flattens the structured output into a sequence, which limits the quality of structural information being learned and leads to inferior performance compared to classic discriminative models. In this work, we describe an approach to model structures as sequences of actions in an autoregressive manner with PLMs, allowing in-structure dependencies to be learned without any loss. Our approach achieves the new state-of-the-art on all the structured prediction tasks we looked at, namely, named entity recognition, end-to-end relation extraction, and coreference resolution.

PDF Abstract

Results from the Paper


 Ranked #1 on Relation Extraction on CoNLL04 (RE+ Micro F1 metric)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Relation Extraction ACE 2005 ASP+T5-3B RE Micro F1 72.7 # 2
NER Micro F1 91.3 # 1
RE+ Micro F1 70.5 # 2
Sentence Encoder T5-3B # 1
Cross Sentence Yes # 1
Relation Extraction CoNLL04 ASP+T0-3B RE+ Micro F1 76.3 # 1
NER Micro F1 90.3 # 1
Named Entity Recognition (NER) CoNLL 2003 (English) ASP+flan-T5-large F1 93.8 # 9
Named Entity Recognition (NER) CoNLL 2003 (English) ASP+T5-3B F1 94.1 # 4
Coreference Resolution CoNLL 2012 ASP+T0-3B Avg F1 82.3 # 3
Coreference Resolution OntoNotes ASP+T0-3B F1 82.3 # 2

Methods


No methods listed for this paper. Add relevant methods here