Björn Barz commited on
Commit
f3b1f9f
1 Parent(s): 873f0af

Initial model

Browse files
Files changed (6) hide show
  1. README.md +18 -0
  2. config.json +158 -0
  3. pytorch_model.bin +3 -0
  4. special_tokens_map.json +1 -0
  5. tokenizer_config.json +1 -0
  6. vocab.txt +0 -0
README.md ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - summarization
6
+ license: apache-2.0
7
+ datasets:
8
+ - arxiv_dataset
9
+ metrics:
10
+ - rouge
11
+ ---
12
+
13
+ # Paper Title Generator
14
+
15
+ Generates titles for computer science papers given an abstract.
16
+
17
+ The model is a BERT2BERT Encoder-Decoder using the official `bert-base-uncased` checkpoint as initialization for the encoder and decoder.
18
+ It was fine-tuned on 318,500 computer science papers posted on arXiv.org between 2007 and 2020 and achieved a 26.3% Rouge2 F1-Score on held-out validation data.
config.json ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "EncoderDecoderModel"
4
+ ],
5
+ "decoder": {
6
+ "_name_or_path": "bert-base-uncased",
7
+ "add_cross_attention": true,
8
+ "architectures": [
9
+ "BertForMaskedLM"
10
+ ],
11
+ "attention_probs_dropout_prob": 0.1,
12
+ "bad_words_ids": null,
13
+ "bos_token_id": null,
14
+ "chunk_size_feed_forward": 0,
15
+ "decoder_start_token_id": null,
16
+ "diversity_penalty": 0.0,
17
+ "do_sample": false,
18
+ "early_stopping": false,
19
+ "encoder_no_repeat_ngram_size": 0,
20
+ "eos_token_id": null,
21
+ "finetuning_task": null,
22
+ "gradient_checkpointing": false,
23
+ "hidden_act": "gelu",
24
+ "hidden_dropout_prob": 0.1,
25
+ "hidden_size": 768,
26
+ "id2label": {
27
+ "0": "LABEL_0",
28
+ "1": "LABEL_1"
29
+ },
30
+ "initializer_range": 0.02,
31
+ "intermediate_size": 3072,
32
+ "is_decoder": true,
33
+ "is_encoder_decoder": false,
34
+ "label2id": {
35
+ "LABEL_0": 0,
36
+ "LABEL_1": 1
37
+ },
38
+ "layer_norm_eps": 1e-12,
39
+ "length_penalty": 1.0,
40
+ "max_length": 20,
41
+ "max_position_embeddings": 512,
42
+ "min_length": 0,
43
+ "model_type": "bert",
44
+ "no_repeat_ngram_size": 0,
45
+ "num_attention_heads": 12,
46
+ "num_beam_groups": 1,
47
+ "num_beams": 1,
48
+ "num_hidden_layers": 12,
49
+ "num_return_sequences": 1,
50
+ "output_attentions": false,
51
+ "output_hidden_states": false,
52
+ "output_scores": false,
53
+ "pad_token_id": 0,
54
+ "position_embedding_type": "absolute",
55
+ "prefix": null,
56
+ "pruned_heads": {},
57
+ "repetition_penalty": 1.0,
58
+ "return_dict": true,
59
+ "return_dict_in_generate": false,
60
+ "sep_token_id": null,
61
+ "task_specific_params": null,
62
+ "temperature": 1.0,
63
+ "tie_encoder_decoder": false,
64
+ "tie_word_embeddings": true,
65
+ "tokenizer_class": null,
66
+ "top_k": 50,
67
+ "top_p": 1.0,
68
+ "torchscript": false,
69
+ "transformers_version": "4.3.2",
70
+ "type_vocab_size": 2,
71
+ "use_bfloat16": false,
72
+ "use_cache": true,
73
+ "vocab_size": 30522,
74
+ "xla_device": null
75
+ },
76
+ "decoder_start_token_id": 101,
77
+ "early_stopping": true,
78
+ "encoder": {
79
+ "_name_or_path": "bert-base-uncased",
80
+ "add_cross_attention": false,
81
+ "architectures": [
82
+ "BertForMaskedLM"
83
+ ],
84
+ "attention_probs_dropout_prob": 0.1,
85
+ "bad_words_ids": null,
86
+ "bos_token_id": null,
87
+ "chunk_size_feed_forward": 0,
88
+ "decoder_start_token_id": null,
89
+ "diversity_penalty": 0.0,
90
+ "do_sample": false,
91
+ "early_stopping": false,
92
+ "encoder_no_repeat_ngram_size": 0,
93
+ "eos_token_id": null,
94
+ "finetuning_task": null,
95
+ "gradient_checkpointing": false,
96
+ "hidden_act": "gelu",
97
+ "hidden_dropout_prob": 0.1,
98
+ "hidden_size": 768,
99
+ "id2label": {
100
+ "0": "LABEL_0",
101
+ "1": "LABEL_1"
102
+ },
103
+ "initializer_range": 0.02,
104
+ "intermediate_size": 3072,
105
+ "is_decoder": false,
106
+ "is_encoder_decoder": false,
107
+ "label2id": {
108
+ "LABEL_0": 0,
109
+ "LABEL_1": 1
110
+ },
111
+ "layer_norm_eps": 1e-12,
112
+ "length_penalty": 1.0,
113
+ "max_length": 20,
114
+ "max_position_embeddings": 512,
115
+ "min_length": 0,
116
+ "model_type": "bert",
117
+ "no_repeat_ngram_size": 0,
118
+ "num_attention_heads": 12,
119
+ "num_beam_groups": 1,
120
+ "num_beams": 1,
121
+ "num_hidden_layers": 12,
122
+ "num_return_sequences": 1,
123
+ "output_attentions": false,
124
+ "output_hidden_states": false,
125
+ "output_scores": false,
126
+ "pad_token_id": 0,
127
+ "position_embedding_type": "absolute",
128
+ "prefix": null,
129
+ "pruned_heads": {},
130
+ "repetition_penalty": 1.0,
131
+ "return_dict": true,
132
+ "return_dict_in_generate": false,
133
+ "sep_token_id": null,
134
+ "task_specific_params": null,
135
+ "temperature": 1.0,
136
+ "tie_encoder_decoder": false,
137
+ "tie_word_embeddings": true,
138
+ "tokenizer_class": null,
139
+ "top_k": 50,
140
+ "top_p": 1.0,
141
+ "torchscript": false,
142
+ "transformers_version": "4.3.2",
143
+ "type_vocab_size": 2,
144
+ "use_bfloat16": false,
145
+ "use_cache": true,
146
+ "vocab_size": 30522,
147
+ "xla_device": null
148
+ },
149
+ "eos_token_id": 102,
150
+ "is_encoder_decoder": true,
151
+ "max_length": 32,
152
+ "min_length": 3,
153
+ "model_type": "encoder-decoder",
154
+ "no_repeat_ngram_size": 3,
155
+ "num_beams": 4,
156
+ "pad_token_id": 0,
157
+ "vocab_size": 30522
158
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5aed176fb47d1e4c5b43064aa756da4cfc7f12c4561ddd07b8a92f78b82924e2
3
+ size 989691346
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"do_lower_case": true, "do_basic_tokenize": true, "never_split": null, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "tokenize_chinese_chars": true, "strip_accents": null, "model_max_length": 512, "name_or_path": "bert-base-uncased"}
vocab.txt ADDED
The diff for this file is too large to render. See raw diff