8.7 KiB
Modelli multilingue per l'inferenza
Ci sono diversi modelli multilingue in 🤗 Transformers, e il loro utilizzo per l'inferenza differisce da quello dei modelli monolingua. Non tutti gli utilizzi dei modelli multilingue sono però diversi. Alcuni modelli, come google-bert/bert-base-multilingual-uncased, possono essere usati come un modello monolingua. Questa guida ti mostrerà come utilizzare modelli multilingue che utilizzano un modo diverso per fare l'inferenza.
XLM
XLM ha dieci diversi checkpoint, di cui solo uno è monolingua. I nove checkpoint rimanenti possono essere suddivisi in due categorie: i checkpoint che utilizzano i language embeddings e quelli che non li utilizzano.
XLM con language embeddings
I seguenti modelli XLM utilizzano gli embeddings linguistici per specificare la lingua utilizzata per l'inferenza:
FacebookAI/xlm-mlm-ende-1024
(Modellazione mascherata del linguaggio (Masked language modeling, in inglese), Inglese-Tedesco)FacebookAI/xlm-mlm-enfr-1024
(Modellazione mascherata del linguaggio, Inglese-Francese)FacebookAI/xlm-mlm-enro-1024
(Modellazione mascherata del linguaggio, Inglese-Rumeno)FacebookAI/xlm-mlm-xnli15-1024
(Modellazione mascherata del linguaggio, lingue XNLI)FacebookAI/xlm-mlm-tlm-xnli15-1024
(Modellazione mascherata del linguaggio + traduzione, lingue XNLI)FacebookAI/xlm-clm-enfr-1024
(Modellazione causale del linguaggio, Inglese-Francese)FacebookAI/xlm-clm-ende-1024
(Modellazione causale del linguaggio, Inglese-Tedesco)
Gli embeddings linguistici sono rappresentati come un tensore delle stesse dimensioni dell' input_ids
passato al modello. I valori in questi tensori dipendono dal linguaggio usato e sono identificati dagli attributi lang2id
e id2lang
del tokenizer.
In questo esempio, carica il checkpoint FacebookAI/xlm-clm-enfr-1024
(Modellazione causale del linguaggio, Inglese-Francese):
>>> import torch
>>> from transformers import XLMTokenizer, XLMWithLMHeadModel
>>> tokenizer = XLMTokenizer.from_pretrained("FacebookAI/xlm-clm-enfr-1024")
>>> model = XLMWithLMHeadModel.from_pretrained("FacebookAI/xlm-clm-enfr-1024")
L'attributo lang2id
del tokenizer mostra il linguaggio del modello e il suo ids:
>>> print(tokenizer.lang2id)
{'en': 0, 'fr': 1}
Poi, crea un esempio di input:
>>> input_ids = torch.tensor([tokenizer.encode("Wikipedia was used to")]) # batch size of 1
Imposta l'id del linguaggio a "en"
e usalo per definire il language embedding. Il language embedding è un tensore riempito con 0
perché questo è il language id per l'inglese. Questo tensore dovrebbe avere la stessa dimensione di input_ids
.
>>> language_id = tokenizer.lang2id["en"] # 0
>>> langs = torch.tensor([language_id] * input_ids.shape[1]) # torch.tensor([0, 0, 0, ..., 0])
>>> # We reshape it to be of size (batch_size, sequence_length)
>>> langs = langs.view(1, -1) # is now of shape [1, sequence_length] (we have a batch size of 1)
Adesso puoi inserire input_ids
e language embedding nel modello:
>>> outputs = model(input_ids, langs=langs)
Lo script run_generation.py può generare testo tramite i language embeddings usando i checkpoints xlm-clm
.
XLM senza language embeddings
I seguenti modelli XLM non richiedono l'utilizzo dei language embeddings per fare inferenza:
FacebookAI/xlm-mlm-17-1280
(Modellazione mascherata del linguaggio, 17 lingue)FacebookAI/xlm-mlm-100-1280
(Modellazione mascherata del linguaggio, 100 lingue)
Questi modelli sono utilizzati per rappresentazioni generiche di frasi, a differenza dei precedenti checkpoints XML.
BERT
Il seguente modello BERT può essere usato per compiti multilingue:
google-bert/bert-base-multilingual-uncased
(Modellazione mascherata del linguaggio + Previsione della prossima frase, 102 lingue)google-bert/bert-base-multilingual-cased
(Modellazione mascherata del linguaggio + Previsione della prossima frase, 104 lingue)
Questi modelli non richiedono language embeddings per fare inferenza. Riescono ad identificare il linguaggio dal contesto e inferire di conseguenza.
XLM-RoBERTa
Il seguente modello XLM-RoBERTa può essere usato per compiti multilingue:
FacebookAI/xlm-roberta-base
(Modellazione mascherata del linguaggio, 100 lingue)FacebookAI/xlm-roberta-large
(Modellazione mascherata del linguaggio, 100 lingue)
XLM-RoBERTa è stato addestrato su 2.5TB di dati CommonCrawl appena creati e puliti in 100 lingue. Offre notevoli vantaggi rispetto ai modelli multilingue rilasciati in precedenza, come mBERT o XLM, in compiti come la classificazione, l'etichettatura delle sequenze e la risposta alle domande.
M2M100
Il seguente modello M2M100 può essere usato per compiti multilingue:
facebook/m2m100_418M
(Traduzione)facebook/m2m100_1.2B
(Traduzione)
In questo esempio, carica il checkpoint facebook/m2m100_418M
per tradurre dal cinese all'inglese. Puoi impostare la lingua di partenza nel tokenizer:
>>> from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
>>> en_text = "Do not meddle in the affairs of wizards, for they are subtle and quick to anger."
>>> chinese_text = "不要插手巫師的事務, 因為他們是微妙的, 很快就會發怒."
>>> tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M", src_lang="zh")
>>> model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
Applica il tokenizer al testo:
>>> encoded_zh = tokenizer(chinese_text, return_tensors="pt")
M2M100 forza l'id della lingua obiettivo come primo token generato per tradurre nella lingua obiettivo. Imposta il parametro forced_bos_token_id
a en
nel metodo generate
per tradurre in inglese:
>>> generated_tokens = model.generate(**encoded_zh, forced_bos_token_id=tokenizer.get_lang_id("en"))
>>> tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
'Do not interfere with the matters of the witches, because they are delicate and will soon be angry.'
MBart
Il seguente modello MBart può essere usato per compiti multilingue:
facebook/mbart-large-50-one-to-many-mmt
(Traduzione automatica multilingue uno-a-molti, 50 lingue)facebook/mbart-large-50-many-to-many-mmt
(Traduzione automatica multilingue molti-a-molti, 50 lingue)facebook/mbart-large-50-many-to-one-mmt
(Traduzione automatica multilingue molti-a-uno, 50 lingue)facebook/mbart-large-50
(Traduzione multilingue, 50 lingue)facebook/mbart-large-cc25
In questo esempio, carica il checkpoint facebook/mbart-large-50-many-to-many-mmt
per tradurre dal finlandese all'inglese. Puoi impostare la lingua di partenza nel tokenizer:
>>> from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
>>> en_text = "Do not meddle in the affairs of wizards, for they are subtle and quick to anger."
>>> fi_text = "Älä sekaannu velhojen asioihin, sillä ne ovat hienovaraisia ja nopeasti vihaisia."
>>> tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt", src_lang="fi_FI")
>>> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
Applica il tokenizer sul testo:
>>> encoded_en = tokenizer(en_text, return_tensors="pt")
MBart forza l'id della lingua obiettivo come primo token generato per tradurre nella lingua obiettivo. Imposta il parametro forced_bos_token_id
a en
nel metodo generate
per tradurre in inglese:
>>> generated_tokens = model.generate(**encoded_en, forced_bos_token_id=tokenizer.lang_code_to_id("en_XX"))
>>> tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
"Don't interfere with the wizard's affairs, because they are subtle, will soon get angry."
Se stai usando il checkpoint facebook/mbart-large-50-many-to-one-mmt
, non hai bisogno di forzare l'id della lingua obiettivo come primo token generato altrimenti l'uso è lo stesso.