Downloads · 30 days
238
4% of all-time downloads
PRIMERA-arxiv
allenai/PRIMERA-arxiv
PRIMERA-arxiv is a machine learning model from allenai. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
HF-version model for PRIMERA: Pyramid-based Masked Sentence Pre-training for Multi-document Summarization (ACL 2022).
All-time downloads
5.6K
Public
Repo size
3.6 GB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin1.8 GB · 100%
At a glance
- Library
- transformers
- License
- apache-2.0
- Model type
- led
- Access
- Public
- Created
- Mar 11, 2022
- Updated
- Jan 24, 2023
- SHA
- 9c48056d
Model card
From the Hugging Face model README
HF-version model for PRIMERA: Pyramid-based Masked Sentence Pre-training for Multi-document Summarization (ACL 2022).
The original code can be found here. You can find the script and notebook to train/evaluate the model in the original github repo.
- Note: due to the difference between the implementations of the original Longformer and the Huggingface LED model, the results of converted models are slightly different. We run a sanity check on both fine-tuned and non fine-tuned models on the MultiNews dataset, and show the results below:
| Model | Rouge-1 | Rouge-2 | Rouge-L |
|---|---|---|---|
| PRIMERA | 42.0 | 13.6 | 20.8 |
| PRIMERA-hf | 41.7 | 13.6 | 20.5 |
| PRIMERA(finetuned) | 49.9 | 21.1 | 25.9 |
| PRIMERA-hf(finetuned) | 49.9 | 20.9 | 25.8 |
You can use it by
from transformers import (
AutoTokenizer,
LEDConfig,
LEDForConditionalGeneration,
)
tokenizer = AutoTokenizer.from_pretrained('allenai/PRIMERA')
config=LEDConfig.from_pretrained('allenai/PRIMERA')
model = LEDForConditionalGeneration.from_pretrained('allenai/PRIMERA')
- Library
- transformers
- Type
- led
- License
- apache-2.0
- Created
- Mar 11, 2022
- Updated
- Jan 24, 2023