|
--- |
|
language: |
|
- "en" |
|
thumbnail: "url to a thumbnail used in social sharing" |
|
tags: |
|
- 'macroeconomics' |
|
- 'automated summary evaluation' |
|
- 'wording' |
|
license: "apache-2.0" |
|
metrics: |
|
- 'mse' |
|
--- |
|
|
|
# Wording 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 Wording score. The model to predict the Content score can be found [here](https://huggingface.co./tiedaar/longformer-content-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. |
|
All summaries for 15 sources were withheld from the training set to use for testing. |
|
The model reported an R2 of 0.67 on the test set of summaries. |
|
![content scatter](wording_scatter.png) |
|
|
|
## Contact |
|
This model was developed by LEAR Lab at Vanderbilt University. |
|
For questions or comments about this model, please contact [[email protected]]([email protected]). |
|
--- |
|
|
|
|