HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Structured Prediction as Translation between Augmented Natural Languages

Giovanni Paolini Ben Athiwaratkun Jason Krone Jie Ma Alessandro Achille Rishita Anubhai Cicero Nogueira dos Santos Bing Xiang Stefano Soatto

Structured Prediction as Translation between Augmented Natural Languages

Abstract

We propose a new framework, Translation between Augmented Natural Languages (TANL), to solve many structured prediction language tasks including joint entity and relation extraction, nested named entity recognition, relation classification, semantic role labeling, event extraction, coreference resolution, and dialogue state tracking. Instead of tackling the problem by training task-specific discriminative classifiers, we frame it as a translation task between augmented natural languages, from which the task-relevant information can be easily extracted. Our approach can match or outperform task-specific models on all tasks, and in particular, achieves new state-of-the-art results on joint entity and relation extraction (CoNLL04, ADE, NYT, and ACE2005 datasets), relation classification (FewRel and TACRED), and semantic role labeling (CoNLL-2005 and CoNLL-2012). We accomplish this while using the same architecture and hyperparameters for all tasks and even when training a single model to solve all tasks at the same time (multi-task learning). Finally, we show that our framework can also significantly improve the performance in a low-resource regime, thanks to better use of label semantics.

Code Repositories

amazon-science/tanl
pytorch
Mentioned in GitHub
amazon-research/tanl
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
relation-classification-on-tacred-1TANL (multi-task)
F1: 61.9
relation-classification-on-tacred-1TANL
F1: 71.9
relation-extraction-on-conll04TANL
RE+ Micro F1: 72.6
relation-extraction-on-tacredTANL
F1: 71.9

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