Git Product home page Git Product logo

berta's Introduction

BERTa: RoBERTa-based Catalan language model

Model description

BERTa is a transformer-based masked language model for the Catalan language.

It is based on the RoBERTA base model

and has been trained on a medium-size corpus collected from publicly available corpora and crawlers.

The pretrained model is available on the HuggingFace model hub under the name: https://huggingface.co/bsc/roberta-base-ca-cased

Training corpora and preprocessing

The training corpus consists of several corpora gathered from web crawling and public corpora.

The publicly available corpora are:

  1. the Catalan part of the DOGC corpus, a set of documents from the Official Gazette of the Catalan Government

  2. the Catalan Open Subtitles, a collection of translated movie subtitles

  3. the non-shuffled version of the Catalan part of the OSCAR corpus \\cite{suarez2019asynchronous},

    a collection of monolingual corpora, filtered from Common Crawl

  4. The CaWac corpus, a web corpus of Catalan built from the .cat top-level-domain in late 2013

    the non-deduplicated version

  5. the Catalan Wikipedia articles downloaded on 18-08-2020.

The crawled corpora are:

  1. The Catalan General Crawling, obtained by crawling the 500 most popular .cat and .ad domains

  2. the Catalan Government Crawling, obtained by crawling the .gencat domain and subdomains, belonging to the Catalan Government

  3. the ACN corpus with 220k news items from March 2015 until October 2020, crawled from the Catalan News Agency

To obtain a high-quality training corpus, each corpus have preprocessed with a pipeline of operations, including among the others,

sentence splitting, language detection, filtering of bad-formed sentences and deduplication of repetitive contents.

During the process, we keep document boundaries are kept.

Finally, the corpora are concatenated and further global deduplication among the corpora is applied.

The final training corpus consists of about 1,8B tokens.

Tokenization and pretraining

The training corpus has been tokenized using a byte version of Byte-Pair Encoding (BPE)

used in the original RoBERTA model with a vocabulary size of 52,000 tokens.

The BERTa pretraining consists of a masked language model training that follows the approach employed for the RoBERTa base model

with the same hyperparameters as in the original work.

The training lasted a total of 48 hours with 16 NVIDIA V100 GPUs of 16GB DDRAM.

Downstream tasks

CLUB: Catalan Language Understanding Benchmark

The CLUB benchmark consists of 5 tasks, that are Part-of-Speech Tagging (POS), Named Entity Recognition (NER), Text Classification (TC), Semantic Textual Similarity (STS) and Question Answering (QA). For more information, refer to the HuggingFace datasets cards and Zenodo links below :

  1. AnCora (POS):

  2. AnCora-ner (NER):

  3. TeCla (TC):

  4. STS-ca (STS):

  5. ViquiQuAD (QA):

  6. XQuAD (QA):

    • Splits info:
      • test: 1,190 examples

Fine-tuning and evaluation

The fine-tuning scripts for the downstream tasks are based on the HuggingFace Transformers library. To fine-tune and evaluate on CLUB, run the following commands:

The Catalan Language Understanding Benchmark (CLUB)

Fine-tune and evaluate on CLUB

To fine-tune and evaluate BERTa on the CLUB datasets, run the following commands:

bash setup_venv.sh
bash finetune_berta_club.sh

The commands above will run fine-tuning and evaluation on CLUB and the results will be shown in the results.json file. and the logs in the finetune_berta_club.log file.

Results

The official results obtained are:

Task NER (F1) POS (F1) STS (Pearson) TC (accuracy) QA (ViquiQuAD) (F1/EM) QA (XQuAD) (F1/EM)
BERTa 88.13 98.97 79.73 74.16 86.97/72.29 68.89/48.87
mBERT 86.38 98.82 76.34 70.56 86.97/72.22 67.15/46.51
XLM-RoBERTa 87.66 98.89 75.40 71.68 85.50/70.47 67.10/46.42
WikiBERT-ca 77.66 97.60 77.18 73.22 85.45/70.75 65.21/36.60

How to cite

If you use any of these resources in your work, please cite our latest paper:

@misc{armengolestape2021multilingual,
      title={Are Multilingual Models the Best Choice for Moderately Under-resourced Languages? A Comprehensive Assessment for Catalan}, 
      author={Jordi Armengol{-}Estap{\'{e}} and Casimiro Pio Carrino and Carlos Rodriguez-Penagos and Ona de Gibert Bonet and Carme Armentano{-}Oller and Aitor Gonzalez{-}Agirre and Maite Melero and Marta Villegas},
      year={2021},
      eprint={2107.07903},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

berta's People

Contributors

ccasimiro88 avatar jordiae avatar onadegibert avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.