We release the first neural machine translation system for translation between Russian, Azerbaijani and the endangered Lezgian languages, as well as monolingual and parallel datasets collected and aligned for training and evaluating the system.
1from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
2
3model = AutoModelForSeq2SeqLM.from_pretrained("AlidarAsvarov/nllb-200-600M-lez-rus-azj")
4tokenizer = NllbTokenizer.from_pretrained("AlidarAsvarov/nllb-200-600M-lez-rus-azj", )
5
6model.eval()
7
8tokenizer.src_lang = 'rus_Cyrl'
9inputs = tokenizer(
10 'Футбольная ассоциации Англии рассчитывала провести финал кубка на перестроенном стадионе "Уэмбли"',
11 return_tensors='pt',
12 padding=True,
13 truncation=True,
14 max_length=40
15)
16
17tokenizer.tgt_lang = 'lez_Cyrl'
18tokenizer.batch_decode(
19 model.generate(
20 **inputs,
21 forced_bos_token_id=tokenizer.convert_tokens_to_ids(tokenizer.tgt_lang),
22 max_new_tokens=int(32 + 3 * inputs.input_ids.shape[1]),
23 num_beams=5
24 ),
25 skip_special_tokens=True
26)
1from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
2
3model = AutoModelForSeq2SeqLM.from_pretrained("AlidarAsvarov/nllb-200-600M-lez-rus-azj")
4tokenizer = NllbTokenizer.from_pretrained("AlidarAsvarov/nllb-200-600M-lez-rus-azj", )
5
6model.eval()
7
8tokenizer.src_lang = 'azj_Latn'
9inputs = tokenizer(
10 '1741-ci ilin payızında Nadir s ̧ah Mu ̈s ̧ku ̈r lәzgilәrinә hu ̈cum edib onlara divan tutur.',
11 return_tensors='pt',
12 padding=True,
13 truncation=True,
14 max_length=102
15)
16
17tokenizer.tgt_lang = 'lez_Cyrl'
18tokenizer.batch_decode(
19 model.generate(
20 **inputs,
21 forced_bos_token_id=tokenizer.convert_tokens_to_ids(tokenizer.tgt_lang),
22 max_new_tokens=int(32 + 3 * inputs.input_ids.shape[1]),
23 num_beams=5
24 ),
25 skip_special_tokens=True
26)
1@misc{asvarov2024neuralmachinetranslationlezgian,
2 title={Neural machine translation system for Lezgian, Russian and Azerbaijani languages},
3 author={Alidar Asvarov and Andrey Grabovoy},
4 year={2024},
5 eprint={2410.05472},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL},
8 url={https://arxiv.org/abs/2410.05472},
9}