CoolFace
Modelpublic

ikim-uk-essen/geberta-xlarge

sourceHugging Facemitupdated 2y agoView on Hugging Face
1likes31downloads
Model Card

GeBERTa

<!-- Provide a quick summary of what the model is/does. --> GeBERTa is a set of German DeBERTa models developed in a joint effort between the University of Florida, NVIDIA, and IKIM. The models range in size from 122M to 750M parameters.

Model details

The models follow the architecture of DeBERTa-v2 and make use of sentence piece tokenizers. The base and large models use a 50k token vocabulary, while the large model uses a 128k token vocabulary. All models were trained with a batch size of 2k for a maximum of 1 million steps and have a maximum sequence length of 512 tokens.

Dataset

The pre-training dataset consists of documents from different domains:

DomainDatasetData Size#Docs#Tokens
FormalWikipedia9GB2,665,3571.9B
FormalNews28GB12,305,3266.1B
FormalGC490GB31,669,77219.4B
InformalReddit 2019-2023 (GER)5.8GB15,036,5921.3B
InformalHoliday Reviews2GB4,876,405428M
LegalOpenLegalData: German cases and laws5.4GB308,2281B
MedicalSmaller public datasets253MB179,77650M
MedicalCC medical texts3.6GB2,000,000682M
MedicalMedicine Dissertations1.4GB14,496295M
MedicalPubmed abstracts (translated)8.5GB21,044,3821.7B
MedicalMIMIC III (translated)2.6GB24,221,834695M
MedicalPMC-Patients-ReCDS (translated)2.1GB1,743,344414M
LiteratureGerman Fiction1.1GB3,219243M
LiteratureEnglish books (translated)7.1GB11,0381.6B
-Total167GB116,079,76935.8B

Benchmark

In a comprehensive benchmark, we evaluated existing German models and our own. The benchmark included a variety of task types, such as question answering, classification, and named entity recognition (NER). In addition, we introduced a new task focused on hate speech detection using two existing datasets. When the datasets provided training, development, and test sets, we used them accordingly.

We randomly split the data into 80% for training, 10% for validation, and 10% for test in cases where such sets were not available. The following table presents the F1 scores:

Model[GE14](https://huggingface.co/datasets/germeval_14)[GQuAD](https://huggingface.co/datasets/deepset/germanquad)[GE18](https://huggingface.co/datasets/philschmid/germeval18)TS[GGP](https://github.com/JULIELab/GGPOnc)GRAS<sup>1</sup>[JS](https://github.com/JULIELab/jsyncc)[DROC](https://gitlab2.informatik.uni-wuerzburg.de/kallimachos/DROC-Release)Avg
GBERT<sub>large</sub>88.48±0.2381.51±0.8454.37±1.6573.60±0.6179.17±0.1469.28±0.8076.32±4.4290.29±0.1576.63±0.63
GELECTRA<sub>large</sub>88.39±0.1380.51±0.4155.41±1.5473.84±0.8679.09±0.0970.16±0.9273.73±2.3589.83±0.2776.37±0.69
GeBERTa<sub>large</sub>88.84±0.1882.52±0.5953.76±1.8675.32±0.5378.35±0.0870.02±1.3482.16±2.3690.39±0.2477.67±0.69
GeBERTa<sub>xlarge</sub>89.04±0.2685.05±0.6355.80±1.4276.25±0.70476.71±0.0867.92±1.0082.42±4.7090.63±0.2177.98±0.62

Publication

bibtex
@inproceedings{dada2023impact,
  title={On the Impact of Cross-Domain Data on German Language Models},
  author={Dada, Amin and Chen, Aokun and Peng, Cheng and Smith, Kaleb E and Idrissi-Yaghir, Ahmad and Seibold, Constantin Marc and Li, Jianning and Heiliger, Lars and Friedrich, Christoph M and Truhn, Daniel and others},
  booktitle={The 2023 Conference on Empirical Methods in Natural Language Processing},
  year={2023}
}

Arxiv to link paper on Hugging Face: https://arxiv.org/abs/2310.07321

Contact

<amin.dada@uk-essen.de>