HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Autoregressive Structured Prediction with Language Models

Tianyu Liu Yuchen Jiang Nicholas Monath Ryan Cotterell Mrinmaya Sachan

Autoregressive Structured Prediction with Language Models

Abstract

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.

Code Repositories

lyutyuh/asp
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
coreference-resolution-on-conll-2012ASP+T0-3B
Avg F1: 82.3
coreference-resolution-on-ontonotesASP+T0-3B
F1: 82.3
named-entity-recognition-ner-on-conll-2003ASP+flan-T5-large
F1: 93.8
named-entity-recognition-ner-on-conll-2003ASP+T5-3B
F1: 94.1
relation-extraction-on-ace-2005ASP+T5-3B
Cross Sentence: Yes
NER Micro F1: 91.3
RE Micro F1: 72.7
RE+ Micro F1: 70.5
Sentence Encoder: T5-3B
relation-extraction-on-conll04ASP+T0-3B
NER Micro F1: 90.3
RE+ Micro F1: 76.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Autoregressive Structured Prediction with Language Models | Papers | HyperAI