MoritzLaurer's picture
MoritzLaurer HF staff
Update README.md
ade8c81
|
raw
history blame
6.63 kB
metadata
language:
  - multilingual
  - en
  - ar
  - bg
  - de
  - el
  - es
  - fr
  - hi
  - ru
  - sw
  - th
  - tr
  - ur
  - vi
  - zh
license: mit
tags:
  - zero-shot-classification
  - text-classification
  - nli
  - pytorch
metrics:
  - accuracy
datasets:
  - multi_nli
  - xnli
pipeline_tag: zero-shot-classification
widget:
  - text: Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU
    candidate_labels: politics, economy, entertainment, environment

Multilingual XLM-V-base-mnli-xnli

Model description

This multilingual model can perform natural language inference (NLI) on 116 languages and is therefore also suitable for multilingual zero-shot classification. The underlying XLM-V-base model was created by Meta AI and pretrained on the CC100 multilingual dataset. It was then fine-tuned on the XNLI dataset, which contains hypothesis-premise pairs from 15 languages, as well as the English MNLI dataset. XLM-V-base was publish on 23.01.2023 in this paper. Its main innovation is a larger vocabulary: previous multilingual models had a vocabulary of 250 000 tokens, while XLM-V has 1 million tokens. The improved vocabulary allows for better representations of more languages.

mDeBERTa-v3.

How to use the model

Simple zero-shot classification pipeline

from transformers import pipeline
classifier = pipeline("zero-shot-classification", model="MoritzLaurer/xlm-v-base-mnli-xnli")

sequence_to_classify = "Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU"
candidate_labels = ["politics", "economy", "entertainment", "environment"]
output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
print(output)

NLI use-case

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")

model_name = "MoritzLaurer/xlm-v-base-mnli-xnli"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

premise = "Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU"
hypothesis = "Emmanuel Macron is the President of France"

input = tokenizer(premise, hypothesis, truncation=True, return_tensors="pt")
output = model(input["input_ids"].to(device))  # device = "cuda:0" or "cpu"
prediction = torch.softmax(output["logits"][0], -1).tolist()
label_names = ["entailment", "neutral", "contradiction"]
prediction = {name: round(float(pred) * 100, 1) for pred, name in zip(prediction, label_names)}
print(prediction)

Training data

This model was trained on the XNLI development dataset and the MNLI train dataset. The XNLI development set consists of 2490 professionally translated texts from English to 14 other languages (37350 texts in total) (see this paper). Note that the XNLI contains a training set of 15 machine translated versions of the MNLI dataset for 15 languages, but due to quality issues with these machine translations, this model was only trained on the professional translations from the XNLI development set and the original English MNLI training set (392 702 texts). Not using machine translated texts can avoid overfitting the model to the 15 languages; avoids catastrophic forgetting of the other 85 languages mDeBERTa was pre-trained on; and significantly reduces training costs.

Training procedure

xlm-v-base-mnli-xnli was trained using the Hugging Face trainer with the following hyperparameters.

training_args = TrainingArguments(
    num_train_epochs=3,              # total number of training epochs
    learning_rate=2e-05,
    per_device_train_batch_size=32,   # batch size per device during training
    per_device_eval_batch_size=120,    # batch size for evaluation
    warmup_ratio=0.06,                # number of warmup steps for learning rate scheduler
    weight_decay=0.01,               # strength of weight decay
)

Eval results

The model was evaluated on the XNLI test set on 15 languages (5010 texts per language, 75150 in total). Note that multilingual NLI models are capable of classifying NLI texts without receiving NLI training data in the specific language (cross-lingual transfer). This means that the model is also able of doing NLI on the other 101~ languages XLM-V was training on, but performance is most likely lower than for those languages available in XNLI.

Also note that if other multilingual models on the model hub claim performance of around 90% on languages other than English, the authors have most likely made a mistake during testing since non of the latest papers shows a multilingual average performance of more than a few points above 80% on XNLI (see here or here).

average ar bg de el en es fr hi ru sw th tr ur vi zh
0.808 0.802 0.829 0.825 0.826 0.883 0.845 0.834 0.771 0.813 0.748 0.793 0.807 0.740 0.795 0.8116
Datasets mnli_m mnli_mm ar bg de el en es fr hi ru sw th tr ur vi zh
Accuracy 0.852 0.854 0.757 0.808 0.796 0.79 0.856 0.814 0.806 0.751 0.782 0.725 0.757 0.766 0.729 0.784 0.782
Speed (text/sec) 2098.0 2170.0 3501.0 3324.0 3438.0 3174.0 3713.0 3500.0 3129.0 3042.0 3419.0 3468.0 3782.0 3772.0 3099.0 3117.0 4217.0

Limitations and bias

Please consult the original XLM-V paper and literature on different NLI datasets for potential biases.

Citation

If you use this model, please cite: Laurer, Moritz, Wouter van Atteveldt, Andreu Salleras Casas, and Kasper Welbers. 2022. ‘Less Annotating, More Classifying – Addressing the Data Scarcity Issue of Supervised Machine Learning with Deep Transfer Learning and BERT - NLI’. Preprint, June. Open Science Framework. https://osf.io/74b8k.

Ideas for cooperation or questions?

If you have questions or ideas for cooperation, contact me at m{dot}laurer{at}vu{dot}nl or LinkedIn