Czert-B-base-cased / README.md
ondfa's picture
Update README.md
32fb1c3
|
raw
history blame
6.01 kB

CZERT

This repository keeps trained model Czert-B for the paper Czert – Czech BERT-like Model for Language Representation For more information, see the paper

How to Use CZERT?

Sentence Level Tasks

We evaluate our model on two sentence level tasks:

  • Sentiment Classification,
  • Semantic Text Similarity.

Document Level Tasks

We evaluate our model on one document level task

  • Multi-label Document Classification.

Token Level Tasks

We evaluate our model on three token level tasks:

  • Named Entity Recognition,
  • Morphological Tagging,
  • Semantic Role Labelling.

Downstream Tasks Fine-tuning Results

Sentiment Classification

mBERT SlavicBERT ALBERT-r Czert-A Czert-B
FB 71.72β€…Β±β€…0.91 73.87β€…Β±β€…0.50 59.50β€…Β±β€…0.47 72.47β€…Β±β€…0.72 76.55β€…Β±β€…0.14
CSFD 82.80β€…Β±β€…0.14 82.51β€…Β±β€…0.14 75.40β€…Β±β€…0.18 79.58β€…Β±β€…0.46 84.79β€…Β±β€…0.26

Average F1 results for the Sentiment Classification task. For more information, see the paper.

Semantic Text Similarity

mBERT Pavlov Albert-random Czert-A Czert-B
STA-CNA 83.335β€…Β±β€…0.063 83.593β€…Β±β€…0.050 43.184β€…Β±β€…0.125 82.942β€…Β±β€…0.106 84.345β€…Β±β€…0.028
STS-SVOB-img 79.367β€…Β±β€…0.486 79.900β€…Β±β€…0.810 15.739β€…Β±β€…2.992 79.444β€…Β±β€…0.338 83.744β€…Β±β€…0.395
STS-SVOB-hl 78.833β€…Β±β€…0.296 76.996β€…Β±β€…0.305 33.949β€…Β±β€…1.807 75.089β€…Β±β€…0.806 79.827β€…Β±β€…0.469

Comparison of Pearson correlation achieved using pre-trained CZERT-A, CZERT-B, mBERT, Pavlov and randomly initialised Albert on semantic text similarity. For more information see the paper.

Multi-label Document Classification

mBERT SlavicBERT ALBERT-r Czert-A Czert-B
AUROC 97.62β€…Β±β€…0.08 97.80β€…Β±β€…0.06 94.35β€…Β±β€…0.13 97.49β€…Β±β€…0.07 98.00β€…Β±β€…0.04
F1 83.04β€…Β±β€…0.16 84.08β€…Β±β€…0.14 72.44β€…Β±β€…0.22 82.27β€…Β±β€…0.17 85.06β€…Β±β€…0.11

Comparison of F1 and AUROC score achieved using pre-trained CZERT-A, CZERT-B, mBERT, Pavlov and randomly initialised Albert on multi-label document classification. For more information see the paper.

Morphological Tagging

mBERT Pavlov Albert-random Czert-A Czert-B
Universal Dependencies 99.176β€…Β±β€…0.006 99.211β€…Β±β€…0.008 96.590β€…Β±β€…0.096 98.713β€…Β±β€…0.008 99.300β€…Β±β€…0.009

Comparison of F1 score achieved using pre-trained CZERT-A, CZERT-B, mBERT, Pavlov and randomly initialised Albert on morphological tagging task. For more information see the paper.

Semantic Role Labelling

mBERT Pavlov Albert-random Czert-A Czert-B dep-based gold-dep
span 78.547 Β± 0.110 79.333 Β± 0.080 51.365 Β± 0.423 72.254 Β± 0.172 79.112 Β± 0.141 - -
syntax 90.226 Β± 0.224 90.492 Β± 0.040 80.747 Β± 0.131 80.319 Β± 0.054 90.516 Β± 0.047 85.19 89.52

SRL results – dep columns are evaluate with labelled F1 from CoNLL 2009 evaluation script, other columns are evaluated with span F1 score same as it was used for NER evaluation. For more information see the paper.

Named Entity Recognition

mBERT Pavlov Albert-random Czert-A Czert-B
CNEC 86.225β€…Β±β€…0.208 86.565β€…Β±β€…0.198 34.635β€…Β±β€…0.343 72.945β€…Β±β€…0.227 86.274β€…Β±β€…0.116
BSNLP 2019 84.006β€…Β±β€…1.248 86.699β€…Β±β€…0.370 19.773β€…Β±β€…0.938 48.859β€…Β±β€…0.605 86.729 Β± 0.344

Comparison of f1 score achieved using pre-trained CZERT-A, CZERT-B, mBERT, Pavlov and randomly initialised Albert on named entity recognition task. For more information see the paper.

How should I cite CZERT?

For now, please cite the Arxiv paper:

@article{sido2021czert,
      title={Czert -- Czech BERT-like Model for Language Representation}, 
      author={Jakub Sido and OndΕ™ej PraΕΎΓ‘k and Pavel PΕ™ibÑň and Jan PaΕ‘ek and Michal SejΓ‘k and Miloslav KonopΓ­k},
      year={2021},
      eprint={2103.13031},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      journal={arXiv preprint arXiv:2103.13031},
}