Command Palette
Search for a command to run...
Label Verbalization and Entailment for Effective Zero- and Few-Shot Relation Extraction
Oscar Sainz Oier Lopez de Lacalle Gorka Labaka Ander Barrena Eneko Agirre

Abstract
Relation extraction systems require large amounts of labeled examples which are costly to annotate. In this work we reformulate relation extraction as an entailment task, with simple, hand-made, verbalizations of relations produced in less than 15 min per relation. The system relies on a pretrained textual entailment engine which is run as-is (no training examples, zero-shot) or further fine-tuned on labeled examples (few-shot or fully trained). In our experiments on TACRED we attain 63% F1 zero-shot, 69% with 16 examples per relation (17% points better than the best supervised system on the same conditions), and only 4 points short to the state-of-the-art (which uses 20 times more training data). We also show that the performance can be improved significantly with larger entailment models, up to 12 points in zero-shot, allowing to report the best results to date on TACRED when fully trained. The analysis shows that our few-shot systems are specially effective when discriminating between relations, and that the performance difference in low data regimes comes mainly from identifying no-relation cases.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| relation-extraction-on-tacred | NLI_RoBERTa | F1: 71.0 |
| relation-extraction-on-tacred | NLI_DeBERTa | F1: 73.9 F1 (1% Few-Shot): 63.7 F1 (10% Few-Shot): 67.9 F1 (5% Few-Shot): 69.0 F1 (Zero-Shot): 62.8 |
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.