XLM
Released from Facebook
together with the paper Cross-lingual Language Model Pretraining by Guillaume Lample and Alexis Conneau and fine-tuned on XQuAD for multilingual (11 different languages
) Q&A downstream task.
Details of the language model('xlm-mlm-100-1280')
| Languages | --------- | | 100 |
It includes the following languages:
Details of the downstream task (multilingual Q&A) - Dataset
Deepmind XQuAD
Languages covered:
- Arabic:
ar
- German:
de
- Greek:
el
- English:
en
- Spanish:
es
- Hindi:
hi
- Russian:
ru
- Thai:
th
- Turkish:
tr
- Vietnamese:
vi
- Chinese:
zh
As the dataset is based on SQuAD v1.1, there are no unanswerable questions in the data. We chose this setting so that models can focus on cross-lingual transfer.
We show the average number of tokens per paragraph, question, and answer for each language in the table below. The statistics were obtained using Jieba for Chinese and the Moses tokenizer for the other languages.
en | es | de | el | ru | tr | ar | vi | th | zh | hi | |
---|---|---|---|---|---|---|---|---|---|---|---|
Paragraph | 142.4 | 160.7 | 139.5 | 149.6 | 133.9 | 126.5 | 128.2 | 191.2 | 158.7 | 147.6 | 232.4 |
Question | 11.5 | 13.4 | 11.0 | 11.7 | 10.0 | 9.8 | 10.7 | 14.8 | 11.5 | 10.5 | 18.7 |
Answer | 3.1 | 3.6 | 3.0 | 3.3 | 3.1 | 3.1 | 3.1 | 4.5 | 4.1 | 3.5 | 5.6 |
Citation:
@article{Artetxe:etal:2019,
author = {Mikel Artetxe and Sebastian Ruder and Dani Yogatama},
title = {On the cross-lingual transferability of monolingual representations},
journal = {CoRR},
volume = {abs/1910.11856},
year = {2019},
archivePrefix = {arXiv},
eprint = {1910.11856}
}
As XQuAD is just an evaluation dataset, I used Data augmentation techniques (scraping, neural machine translation, etc) to obtain more samples and split the dataset in order to have a train and test set. The test set was created in a way that contains the same number of samples for each language. Finally, I got:
Dataset | # samples |
---|---|
XQUAD train | 50 K |
XQUAD test | 8 K |
Model training
The model was trained on a Tesla P100 GPU and 25GB of RAM. The script for fine tuning can be found here
Model in action
Fast usage with pipelines:
from transformers import pipeline
qa_pipeline = pipeline(
"question-answering",
model="mrm8488/xlm-multi-finetuned-xquadv1",
tokenizer="mrm8488/xlm-multi-finetuned-xquadv1"
)
# English
qa_pipeline({
'context': "Manuel Romero has been working hardly in the repository hugginface/transformers lately",
'question': "Who has been working hard for hugginface/transformers lately?"
})
#Output: {'answer': 'Manuel', 'end': 6, 'score': 8.531880747878265e-05, 'start': 0}
# Russian
qa_pipeline({
'context': "Мануэль Ромеро в последнее время почти не работал в репозитории hugginface / transformers",
'question': "Кто в последнее время усердно работал над обнимашками / трансформерами?"
})
#Output: {'answer': 'работал в репозитории hugginface /','end': 76, 'score': 0.00012340750456964894, 'start': 42}
Try it on a Colab (Do not forget to change the model and tokenizer path in the Colab if necessary):
Created by Manuel Romero/@mrm8488
Made with ♥ in Spain
- Downloads last month
- 10