Pengcheng He
commited on
Commit
•
dfcd5df
1
Parent(s):
95496f0
DeBERTa V2 XXLarge 1.5B model
Browse files- README.md +39 -0
- config.json +25 -0
- pytorch_model.bin +3 -0
- spm.model +3 -0
- tokenizer_config.json +3 -0
README.md
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
thumbnail: https://huggingface.co/front/thumbnails/microsoft.png
|
3 |
+
license: mit
|
4 |
+
---
|
5 |
+
|
6 |
+
## DeBERTa: Decoding-enhanced BERT with Disentangled Attention
|
7 |
+
|
8 |
+
[DeBERTa](https://arxiv.org/abs/2006.03654) improves the BERT and RoBERTa models using disentangled attention and enhanced mask decoder. With those two improvements, DeBERTa out perform RoBERTa on a majority of NLU tasks with 80GB training data.
|
9 |
+
|
10 |
+
Please check the [official repository](https://github.com/microsoft/DeBERTa) for more details and updates.
|
11 |
+
|
12 |
+
This the DeBERTa V2 xxlarge model with 48 layers, 1536 hidden size. Total parameters 1.5B.
|
13 |
+
|
14 |
+
|
15 |
+
#### Fine-tuning on NLU tasks
|
16 |
+
|
17 |
+
We present the dev results on SQuAD 1.1/2.0 and several GLUE benchmark tasks.
|
18 |
+
|
19 |
+
| Model | SQuAD 1.1 | SQuAD 2.0 | MNLI-m | SST-2 | QNLI | CoLA | RTE | MRPC | QQP |STS-B|
|
20 |
+
|-------------------|-----------|-----------|--------|-------|------|------|------|------|------|-----|
|
21 |
+
| BERT-Large | 90.9/84.1 | 81.8/79.0 | 86.6 | 93.2 | 92.3 | 60.6 | 70.4 | 88.0 | 91.3 |90.0 |
|
22 |
+
| RoBERTa-Large | 94.6/88.9 | 89.4/86.5 | 90.2 | 96.4 | 93.9 | 68.0 | 86.6 | 90.9 | 92.2 |92.4 |
|
23 |
+
| XLNet-Large | 95.1/89.7 | 90.6/87.9 | 90.8 | 97.0 | 94.9 | 69.0 | 85.9 | 90.8 | 92.3 |92.5 |
|
24 |
+
| **DeBERTa-Large** | 95.5/90.1 | 90.7/88.0 | 91.1 | 96.5 | 95.3 | 69.5 | 88.1 | 92.5 | 92.3 |92.5 |
|
25 |
+
|
26 |
+
### Citation
|
27 |
+
|
28 |
+
If you find DeBERTa useful for your work, please cite the following paper:
|
29 |
+
|
30 |
+
``` latex
|
31 |
+
@misc{he2020deberta,
|
32 |
+
title={DeBERTa: Decoding-enhanced BERT with Disentangled Attention},
|
33 |
+
author={Pengcheng He and Xiaodong Liu and Jianfeng Gao and Weizhu Chen},
|
34 |
+
year={2020},
|
35 |
+
eprint={2006.03654},
|
36 |
+
archivePrefix={arXiv},
|
37 |
+
primaryClass={cs.CL}
|
38 |
+
}
|
39 |
+
```
|
config.json
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"attention_probs_dropout_prob": 0.1,
|
3 |
+
"hidden_act": "gelu",
|
4 |
+
"hidden_dropout_prob": 0.1,
|
5 |
+
"hidden_size": 1536,
|
6 |
+
"initializer_range": 0.02,
|
7 |
+
"intermediate_size": 6144,
|
8 |
+
"max_position_embeddings": 512,
|
9 |
+
"relative_attention": true,
|
10 |
+
"position_buckets": 256,
|
11 |
+
"norm_rel_ebd": "layer_norm",
|
12 |
+
"share_att_key": true,
|
13 |
+
"pos_att_type": "p2c|c2p",
|
14 |
+
"layer_norm_eps": 1e-7,
|
15 |
+
"conv_kernel_size": 3,
|
16 |
+
"conv_act": "gelu",
|
17 |
+
"max_relative_positions": -1,
|
18 |
+
"position_biased_input": false,
|
19 |
+
"num_attention_heads": 24,
|
20 |
+
"attention_head_size": 64,
|
21 |
+
"num_hidden_layers": 48,
|
22 |
+
"self_attention_type": "dense_att",
|
23 |
+
"type_vocab_size": 0,
|
24 |
+
"vocab_size": 128100
|
25 |
+
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3e8c17906df6c1dd8be000ed342e92dca41434b0700f1a96e3823aedadafcef6
|
3 |
+
size 3135783639
|
spm.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5598d5e96f339a8d980c15f9afd405a2e5e1be7db41de3ed13b0f03fac1e8c17
|
3 |
+
size 2447305
|
tokenizer_config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"do_lower_case": false
|
3 |
+
}
|