pytorch tf electra pretraining yue transformers license:apache-2.0
.gitattributes README.md config.json pytorch_model.bin tf_model.h5 vocab.txt
---
language: yue
license: apache-2.0
metrics:
- DRCD
- openrice-senti
- lihkg-cat
- wordshk-sem
---

# ELECTRA Hongkongese Base

## Model description

ELECTRA trained exclusively with data from Hong Kong. A signaficant amount of Hongkongese/Cantonese/Yue is included in the training data.

## Intended uses & limitations

This model is an alternative to Chinese models. It may offer better performance for tasks catering to the langauge usage of Hong Kongers. Yue Wikipedia is used which is much smaller than Chinese Wikipedia; this model will lack the breath of knowledge compared to other Chinese models.

#### How to use

This is the base model trained from the official repo. Further finetuning will be needed for use on downstream tasks. Other model sizes are also available.

#### Limitations and bias

The training data consists of mostly news articles and blogs. There is probably a bias towards formal language usage. 

## Training data

The following is the list of data sources. Total characters is about 507M.

| Data                                              |   % |
| ------------------------------------------------- | --: |
| News Articles / Blogs                             | 58% |
| Yue Wikipedia / EVCHK                             | 18% |
| Restaurant Reviews                                | 12% |
| Forum Threads                                     | 12% |
| Online Fiction                                    |  1% |

The following is the distribution of different languages within the corpus.

| Language                                          |   % |
| ------------------------------------------------- | --: |
| Standard Chinese                                  | 62% |
| Hongkongese                                       | 30% |
| English                                           |  8% |

## Training procedure

Model was trained on a single TPUv3 from the official repo with the default parameters.

| Parameter                                        | Value |
| ------------------------------------------------ | ----: |
| Batch Size                                       | 256   |
| Max Sequence Size                                | 512   |
| Vocab Size                                       | 30000 |

 *Research supported with Cloud TPUs from Google's TensorFlow Research Cloud (TFRC)*

## Eval results

Average evaluation task results over 10 runs. Comparison using the original repo model and code. Chinese models are available from [Joint Laboratory of HIT and iFLYTEK Research (HFL)](https://huggingface.co/hfl)

| Model       | DRCD (EM/F1) | openrice-senti | lihkg-cat | wordshk-sem |
|:-----------:|:------------:|:--------------:|:---------:|:-----------:|
| Chinese     | 86.6 / 91.7  | 79.1           | 67.4      | 88.1        |
| Hongkongese | 83.0 / 89.6  | 81.5           | 70.0      | 90.1        |
