-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
201e3e5
commit cbb9db4
Showing
23 changed files
with
375 additions
and
120 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,86 @@ | ||
| ## Models | ||
|
|
||
| We are making a few models fine-tuned on the datasets in this benchmark available on the Hugging Face Hub. | ||
|
|
||
| | Datasets | mDeBERTa v3 | BERT Large | BERT Base | | ||
| |:----------------------------:|:--------------------------------------------------------------------------------------------------------------:|:-------------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------------------------------------------------------------:| | ||
| | **ASSIN 2 - STS** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin2-similarity) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin2-similarity) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin2-similarity) | | ||
| | **ASSIN 2 - RTE** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin2-entailment) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin2-entailment) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin2-entailment) | | ||
| | **ASSIN - STS** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin-similarity) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin-similarity) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin-similarity) | | ||
| | **ASSIN - RTE** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin-entailment) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin-entailment) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin-entailment) | | ||
| | **HateBR** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-hatebr) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-hatebr) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-hatebr) | | ||
| | **FaQUaD-NLI** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-faquad-nli) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-faquad-nli) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-faquad-nli) | | ||
| | **PorSimplesSent** | [Link](https://huggingface.co/ruanchaves/mdeberta-v3-base-porsimplessent) | [Link](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-porsimplessent) | [Link](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-porsimplessent) | | ||
|
|
||
| ### Summary of the Fine-Tuning procedure | ||
|
|
||
| * **Step 1**. Hyperparameter optimization is performed using quasi-random search based on Google's [Deep Learning Playbook](https://github.com/google-research/tuning_playbook) instructions. The best learning rate, weight decay, and adam beta1 parameters for each Transformer model on each task are identified. | ||
|
|
||
| * **Step 2**. The best hyperparameters from step 1 are used to fine-tune each model 40 times with different random seeds for up to one epoch ( [Dodge et al. (2020)](https://arxiv.org/abs/2002.06305) ). The 10 best models after the first epoch are selected for the next step. | ||
|
|
||
| * **Step 3**. The top 10 models from step 2 are fine-tuned for 20 epochs, generating predictions for the test set ( [Mosbach et al. (2021)](https://arxiv.org/abs/2006.04884) ). We select the model that is closest to the average of predictions (for regression tasks) or the mode of predictions (for classification tasks). This final model is then uploaded to the Hugging Face Hub and displayed in the tables below. | ||
|
|
||
| ### Links and Results | ||
|
|
||
| Our fine-tuning procedure has achieved results that are either slightly superior or at the same level as the previous state-of-the-art (if any). | ||
| Below is a summary of the results achieved on each dataset. | ||
|
|
||
| #### [ASSIN 2](huggingface.co/datasets/assin2) - STS ( Semantic Textual Similarity ) | ||
|
|
||
| | Model | Pearson | MSE | | ||
| |----------------------------------------------------------|---------|------| | ||
| | **ruanchaves/bert-large-portuguese-cased-assin2-similarity** | **0.86** | **0.48** | | ||
| | Previous SOTA ( for Pearson ) - [Souza et al. (2020)](https://link.springer.com/chapter/10.1007/978-3-030-61377-8_28/tables/2) | 0.852 | 0.50 | | ||
| | SOTA ( for MSE ) - [Stilingue](https://link.springer.com/chapter/10.1007/978-3-030-41505-1_39/tables/2) | 0.817 | **0.47** | | ||
| | [ruanchaves/mdeberta-v3-base-assin2-similarity](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin2-similarity) | 0.847 | 0.62 | | ||
| | [ruanchaves/bert-base-portuguese-cased-assin2-similarity](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin2-similarity) | 0.843 | 0.54 | | ||
|
|
||
| #### [ASSIN 2](huggingface.co/datasets/assin2) - RTE ( Recognizing Textual Entailment ) | ||
|
|
||
| | Model | Accuracy | F1 | | ||
| |----------------------------------------------------------|---------|------| | ||
| | **[ruanchaves/bert-large-portuguese-cased-assin2-entailment](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin2-entailment)** | **0.90** | **0.90** | | ||
| | [ruanchaves/mdeberta-v3-base-assin2-entailment](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin2-entailment) | **0.90** | **0.90** | | ||
| | Previous SOTA | **0.90** | **0.90** | | ||
| | [ruanchaves/bert-base-portuguese-cased-assin2-entailment](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin2-entailment) | 0.88 | 0.88 | | ||
|
|
||
| #### [ASSIN](https://huggingface.co/datasets/assin) - STS ( Semantic Textual Similarity ) | ||
|
|
||
| | Model | Pearson | MSE | | ||
| |----------------------------------------------------------|---------|------| | ||
| | [ruanchaves/bert-large-portuguese-cased-assin-similarity](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin-similarity) | 0.859 | 0.3 | | ||
| | [ruanchaves/mdeberta-v3-base-assin-similarity](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin-similarity) | 0.855 | 0.39 | | ||
| | [ruanchaves/bert-base-portuguese-cased-assin-similarity](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin-similarity) | 0.847 | 0.33 | | ||
|
|
||
| #### [ASSIN](https://huggingface.co/datasets/assin) - RTE ( Recognizing Textual Entailment ) | ||
|
|
||
| | Model | Accuracy | F1 | | ||
| |----------------------------------------------------------|---------|------| | ||
| | [ruanchaves/mdeberta-v3-base-assin-entailment](https://huggingface.co/ruanchaves/mdeberta-v3-base-assin-entailment) | 0.927 | 0.862 | | ||
| | [ruanchaves/bert-large-portuguese-cased-assin-entailment](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-assin-entailment) | 0.92 | 0.828 | | ||
| | [ruanchaves/bert-base-portuguese-cased-assin-entailment](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-assin-entailment) | 0.92 | 0.827 | | ||
|
|
||
| #### [HateBR](https://huggingface.co/datasets/ruanchaves/hatebr) ( Offensive Language Detection ) | ||
|
|
||
|
|
||
| | Model | Accuracy | F1 | | ||
| |----------------------------------------------------------|---------|------| | ||
| | [ruanchaves/bert-large-portuguese-cased-hatebr](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-hatebr) | 0.928 | 0.928 | | ||
| | [ruanchaves/mdeberta-v3-base-hatebr](https://huggingface.co/ruanchaves/mdeberta-v3-base-hatebr) | 0.916 | 0.916 | | ||
| | [ruanchaves/bert-base-portuguese-cased-hatebr](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-hatebr) | 0.914 | 0.914 | | ||
|
|
||
| #### [FaQUaD-NLI](https://huggingface.co/datasets/ruanchaves/faquad-nli) ( Question Answering ) | ||
|
|
||
| | Model | Accuracy | F1 | | ||
| |----------------------------------------------------------|---------|------| | ||
| | [ruanchaves/bert-large-portuguese-cased-faquad-nli](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-faquad-nli) | 0.929 | 0.93 | | ||
| | [ruanchaves/mdeberta-v3-base-faquad-nli](https://huggingface.co/ruanchaves/mdeberta-v3-base-faquad-nli) | 0.926 | 0.926 | | ||
| | [ruanchaves/bert-base-portuguese-cased-faquad-nli](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-faquad-nli) | 0.92 | 0.883 | | ||
|
|
||
| #### [PorSimplesSent](https://huggingface.co/datasets/ruanchaves/porsimplessent) ( Text Simplification ) | ||
|
|
||
| | Model | Accuracy | F1 | | ||
| |----------------------------------------------------------|---------|------| | ||
| | [ruanchaves/mdeberta-v3-base-porsimplessent](https://huggingface.co/ruanchaves/mdeberta-v3-base-porsimplessent) | 0.96 | 0.956 | | ||
| | [ruanchaves/bert-base-portuguese-cased-porsimplessent](https://huggingface.co/ruanchaves/bert-base-portuguese-cased-porsimplessent) | 0.942 | 0.937 | | ||
| | [ruanchaves/bert-large-portuguese-cased-porsimplessent](https://huggingface.co/ruanchaves/bert-large-portuguese-cased-porsimplessent) | 0.921 | 0.913 | |
Oops, something went wrong.