Edit model card

paddlenlp-banner

PaddlePaddle/uie-m-large

Information extraction suffers from its varying targets, heterogeneous structures, and demand-specific schemas. The unified text-to-structure generation framework, namely UIE, can universally model different IE tasks, adaptively generate targeted structures, and collaboratively learn general IE abilities from different knowledge sources. Specifically, UIE uniformly encodes different extraction structures via a structured extraction language, adaptively generates target extractions via a schema-based prompt mechanism - structural schema instructor, and captures the common IE abilities via a large-scale pre-trained text-to-structure model. Experiments show that UIE achieved the state-of-the-art performance on 4 IE tasks, 13 datasets, and on all supervised, low-resource, and few-shot settings for a wide range of entity, relation, event and sentiment extraction tasks and their unification. These results verified the effectiveness, universality, and transferability of UIE.

UIE Paper: https://arxiv.org/abs/2203.12277

PaddleNLP released UIE model series for Information Extraction of texts and multi-modal documents which use the ERNIE 3.0 models as the pre-trained language models and were finetuned on a large amount of information extraction data.

UIE-diagram

Available Models

Model Name Usage Scenarios Supporting Tasks
uie-base
uie-medium
uie-mini
uie-micro
uie-nano
For plain text The extractive model of the scene supports Chinese Supports entity, relation, event, opinion extraction
uie-base-en An extractive model for plain text scenarios, supports English Supports entity, relation, event, opinion extraction
uie-m-base
uie-m-large
An extractive model for plain text scenarios, supporting Chinese and English Supports entity, relation, event, opinion extraction
uie-x-base An extractive model for plain text and document scenarios, supports Chinese and English Supports entity, relation, event, opinion extraction on both plain text and documents/pictures/tables

Performance on Text Dataset

We conducted experiments on the in-house test sets of the three different domains of Internet, medical care, and finance:

financehealthcareinternet
0-shot5-shot0-shot5-shot0-shot5-shot
uie-base (12L768H)46.4370.9271.8385.7278.3381.86
uie-medium (6L768H)41.1164.5365.4075.7278.3279.68
uie-mini (6L384H)37.0464.6560.5078.3672.0976.38
uie-micro (4L384H)37.5362.1157.0475.9266.0070.22
uie-nano (4L312H)38.9466.8348.2976.7462.8672.35
uie-m-large (24L1024H)49.3574.5570.5092.6678.4983.02
uie-m-base (12L768H)38.4674.3163.3787.3276.2780.13
🧾🎓uie-x-base (12L768H)48.8473.8765.6088.8179.36 81.65

0-shot means that no training data is directly used for prediction through paddlenlp.Taskflow, and 5-shot means that each category contains 5 pieces of labeled data for model fine-tuning. Experiments show that UIE can further improve the performance with a small amount of data (few-shot).

Performance on Multimodal Datasets**

We experimented on the zero-shot performance of UIE-X on the in-house multi-modal test sets in three different domains of general, financial, and medical:

General FinancialMedical
🧾🎓uie-x-base (12L768H)65.0373.5184.24

The general test set contains complex samples from different fields and is the most difficult task.

Detailed Info: https://github.com/PaddlePaddle/PaddleNLP/blob/develop/applications/information_extraction/README_en.md

Downloads last month
0
Inference API
Unable to determine this model’s pipeline type. Check the docs .