File size: 2,366 Bytes
b5ff597 fd0be49 b5ff597 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
---
language:
- "en"
thumbnail: "url to a thumbnail used in social sharing"
tags:
- 'macroeconomics'
- 'automated summary evaluation'
- 'content'
license: "apache-2.0"
metrics:
- 'mse'
---
# Content Model
This is a longformer model with a regression head designed to predict the Content score of a summary.
By default, longformers assign global attention to only the classification token with a sliding attention window which moves across the rest of the test.
This model, however, is trained to assign global attention to the entire summary with a reduced sliding window.
When performing inference using this model, you should make sure to assign a custom global attention mask as follows:
```
def inference(summary, source, model):
combined = summary + tokenizer.sep_token + source
context = tokenizer(combined)
sep_index = context['input_ids'].index(2)
context['global_attention_mask'] = [1]*(sep_index + 1) + [0]*(len(context['input_ids'])-(sep_index + 1))
inputs = {}
for key in context:
inputs[key] = torch.tensor([context[key]])
return float(model(**inputs)['logits'][0][0])
```
## Corpus
It was trained on a corpus of 4,233 summaries of 101 sources compiled by Botarleanu et al. (2022).
The summaries were graded by expert raters on 6 criteria: Details, Main Point, Cohesion, Paraphrasing, Objective Language, and Language Beyond the Text.
A principle component analyis was used to reduce the dimensionality of the outcome variables to two.
* **Content** includes Details, Main Point, Paraphrasing and Cohesion
* **Wording** includes Objective Language, and Language Beyond the Text
## Score
This model predicts the Content score. The model to predict the Wording score can be found [here](https://huggingface.co./tiedaar/longformer-wording-global).
The following diagram illustrates the model architecture:
![model diagram](model_diagram.png)
When providing input to the model, the summary and the source should be concatenated using the seperator token \</s>.
This allows the model to have access to both the summary and the source to provide more accurate scores. The model reported an R2 of 0.82 on the test set of summaries.
![content scatter](content_scatter.png)
## Contact
For questions or comments about this model, please contact [[email protected]]([email protected]).
---
|