模型:
dbmdz/bert-tiny-historic-multilingual-cased
我们的历史语言模型库包含以下语言的支持 - 包括训练数据来源:
Language | Training data | Size |
---|---|---|
German | 1234321 | 13-28GB (filtered) |
French | 1234321 | 11-31GB (filtered) |
English | 1236321 | 24GB (year filtered) |
Finnish | 1234321 | 1.2GB |
Swedish | 1234321 | 1.1GB |
目前,模型库中提供以下模型:
Model identifier | Model Hub link |
---|---|
dbmdz/bert-base-historic-multilingual-cased | 1239321 |
dbmdz/bert-base-historic-english-cased | 12310321 |
dbmdz/bert-base-finnish-europeana-cased | 12311321 |
dbmdz/bert-base-swedish-europeana-cased | 12312321 |
我们还发布了多语言模型的较小模型:
Model identifier | Model Hub link |
---|---|
dbmdz/bert-tiny-historic-multilingual-cased | 12313321 |
dbmdz/bert-mini-historic-multilingual-cased | 12314321 |
dbmdz/bert-small-historic-multilingual-cased | 12315321 |
dbmdz/bert-medium-historic-multilingual-cased | 1239321 |
注意:我们先前发布了使用更多噪声数据训练的历史德语和法语的语言模型-有关详细信息,请参见 this repo 。
Model identifier | Model Hub link |
---|---|
dbmdz/bert-base-german-europeana-cased | 12318321 |
dbmdz/bert-base-french-europeana-cased | 12319321 |
我们提供使用不同的OCR置信度阈值的统计信息,以缩小语料库的大小并使用较少噪声的数据:
OCR confidence | Size |
---|---|
0.60 | 28GB |
0.65 | 18GB |
0.70 | 13GB |
我们使用OCR置信度为0.6的最终语料库(28GB)。下图显示了每年的标记分布情况:
与德语类似,我们使用不同的OCR置信度阈值:
OCR confidence | Size |
---|---|
0.60 | 31GB |
0.65 | 27GB |
0.70 | 27GB |
0.75 | 23GB |
0.80 | 11GB |
我们使用OCR置信度为0.7的最终语料库(27GB)。下图显示了每年的标记分布情况:
元数据来自 here 。经过年份筛选的统计数据:
Years | Size |
---|---|
ALL | 24GB |
>= 1800 && < 1900 | 24GB |
我们使用经过年份筛选的变体。下图显示了每年的标记分布情况:
OCR confidence | Size |
---|---|
0.60 | 1.2GB |
下图显示了每年的标记分布情况:
OCR confidence | Size |
---|---|
0.60 | 1.1GB |
下图显示了每年的标记分布情况:
下图显示了完整训练语料库的每年标记分布情况:
对于第一次尝试,我们使用每个预训练语料库的前10GB。我们将芬兰语和瑞典语提升到约10GB。以下是用于生成32k和64k子词词汇的确切大小的表格:
Language | Size |
---|---|
German | 10GB |
French | 10GB |
English | 10GB |
Finnish | 9.5GB |
Swedish | 9.7GB |
然后我们计算以下NER语料库中子词生育率和[UNK]的比例:
Language | NER corpora |
---|---|
German | CLEF-HIPE, NewsEye |
French | CLEF-HIPE, NewsEye |
English | CLEF-HIPE |
Finnish | NewsEye |
Swedish | NewsEye |
32k词汇表各语言的子词生育率和未知比例的详细信息如下:
Language | Subword fertility | Unknown portion |
---|---|---|
German | 1.43 | 0.0004 |
French | 1.25 | 0.0001 |
English | 1.25 | 0.0 |
Finnish | 1.69 | 0.0007 |
Swedish | 1.43 | 0.0 |
64k词汇表各语言的子词生育率和未知比例的详细信息如下:
Language | Subword fertility | Unknown portion |
---|---|---|
German | 1.31 | 0.0004 |
French | 1.16 | 0.0001 |
English | 1.17 | 0.0 |
Finnish | 1.54 | 0.0007 |
Swedish | 1.32 | 0.0 |
我们将瑞典语和芬兰语提升到约27GB。所有预训练语料库的最终统计信息如下:
Language | Size |
---|---|
German | 28GB |
French | 27GB |
English | 24GB |
Finnish | 27GB |
Swedish | 27GB |
总大小为130GB。
受 "Well-Read Students Learn Better: On the Importance of Pre-training Compact Models" 论文的启发,我们训练了较小的模型(不同的层数和隐藏大小),并报告了参数数量和预训练成本:
Model (Layer / Hidden size) | Parameters | Pre-Training time |
---|---|---|
hmBERT Tiny ( 2/128) | 4.58M | 4.3 sec / 1,000 steps |
hmBERT Mini ( 4/256) | 11.55M | 10.5 sec / 1,000 steps |
hmBERT Small ( 4/512) | 29.52M | 20.7 sec / 1,000 steps |
hmBERT Medium ( 8/512) | 42.13M | 35.0 sec / 1,000 steps |
hmBERT Base (12/768) | 110.62M | 80.0 sec / 1,000 steps |
然后我们对多语言 NewsEye 数据集进行下游评估:
我们使用32k词汇表使用官方BERT实现在v3-32 TPU上进行训练,使用以下参数:
python3 run_pretraining.py --input_file gs://histolectra/historic-multilingual-tfrecords/*.tfrecord \ --output_dir gs://histolectra/bert-base-historic-multilingual-cased \ --bert_config_file ./config.json \ --max_seq_length=512 \ --max_predictions_per_seq=75 \ --do_train=True \ --train_batch_size=128 \ --num_train_steps=3000000 \ --learning_rate=1e-4 \ --save_checkpoints_steps=100000 \ --keep_checkpoint_max=20 \ --use_tpu=True \ --tpu_name=electra-2 \ --num_tpu_cores=32
下图显示了预训练损失曲线:
我们使用与基础模型训练相同的参数。
下图显示了迷你模型的预训练损失曲线:
下图显示了小型模型的预训练损失曲线:
下图显示了中型模型的预训练损失曲线:
下图显示了中等模型的预训练损失曲线:
英语BERT模型-使用英国图书馆语料库的文本-在v3-8 TPU上使用Hugging Face JAX/FLAX实现进行了10个epochs(约1M步)的训练,使用以下命令:
python3 run_mlm_flax.py --model_type bert \ --config_name /mnt/datasets/bert-base-historic-english-cased/ \ --tokenizer_name /mnt/datasets/bert-base-historic-english-cased/ \ --train_file /mnt/datasets/bl-corpus/bl_1800-1900_extracted.txt \ --validation_file /mnt/datasets/bl-corpus/english_validation.txt \ --max_seq_length 512 \ --per_device_train_batch_size 16 \ --learning_rate 1e-4 \ --num_train_epochs 10 \ --preprocessing_num_workers 96 \ --output_dir /mnt/datasets/bert-base-historic-english-cased-512-noadafactor-10e \ --save_steps 2500 \ --eval_steps 2500 \ --warmup_steps 10000 \ --line_by_line \ --pad_to_max_length
下图显示了预训练损失曲线:
BERT模型-使用芬兰 Europeana 中的文本-在v3-8 TPU上使用Hugging Face JAX/FLAX实现进行了40个epochs(约1M步)的训练,使用以下命令:
python3 run_mlm_flax.py --model_type bert \ --config_name /mnt/datasets/bert-base-finnish-europeana-cased/ \ --tokenizer_name /mnt/datasets/bert-base-finnish-europeana-cased/ \ --train_file /mnt/datasets/hlms/extracted_content_Finnish_0.6.txt \ --validation_file /mnt/datasets/hlms/finnish_validation.txt \ --max_seq_length 512 \ --per_device_train_batch_size 16 \ --learning_rate 1e-4 \ --num_train_epochs 40 \ --preprocessing_num_workers 96 \ --output_dir /mnt/datasets/bert-base-finnish-europeana-cased-512-dupe1-noadafactor-40e \ --save_steps 2500 \ --eval_steps 2500 \ --warmup_steps 10000 \ --line_by_line \ --pad_to_max_length
下图显示了预训练损失曲线:
BERT模型-使用瑞典 Europeana 中的文本-在v3-8 TPU上使用Hugging Face JAX/FLAX实现进行了40个epochs(约660K步)的训练,使用以下命令:
python3 run_mlm_flax.py --model_type bert \ --config_name /mnt/datasets/bert-base-swedish-europeana-cased/ \ --tokenizer_name /mnt/datasets/bert-base-swedish-europeana-cased/ \ --train_file /mnt/datasets/hlms/extracted_content_Swedish_0.6.txt \ --validation_file /mnt/datasets/hlms/swedish_validation.txt \ --max_seq_length 512 \ --per_device_train_batch_size 16 \ --learning_rate 1e-4 \ --num_train_epochs 40 \ --preprocessing_num_workers 96 \ --output_dir /mnt/datasets/bert-base-swedish-europeana-cased-512-dupe1-noadafactor-40e \ --save_steps 2500 \ --eval_steps 2500 \ --warmup_steps 10000 \ --line_by_line \ --pad_to_max_length
下图显示了预训练损失曲线:
通过Google的TPU Research Cloud(TRC)计划(前身为TensorFlow Research Cloud(TFRC))提供了云TPU的支持。非常感谢提供TRC的访问权限❤️
感谢 Hugging Face 团队的慷慨支持,可以从他们的S3存储中下载带大小写的模型?