Interesting model 3.6x smaller and 4x faster based on m2m 1.2B

Argos Translate
1
2 Likes
2

Link to paper: https://arxiv.org/pdf/2210.11621.pdf

image
image849Ɨ267 45.3 KB

Side question: are there benchmarks of the BLEU scores for current argos models?

1 Like
3

I donā€™t think so but I was thinking about it. it would be necessary to benchmark all argos models to be able to compare with other models.

4

I have tried to export the model to ctranslate2 but I canā€™t manage with the tokenizer.

5

I also didnā€™t manage to use it with argos, it looks like you have to use their script to tokenize

6

I have tried:

# wget https://huggingface.co/alirezamsh/small100/raw/main/tokenization_small100.py
# ct2-transformers-converter --model alirezamsh/small100 --output_dir model
from tokenization_small100 import SMALL100Tokenizer
import ctranslate2


translator = ctranslate2.Translator("model")
tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
tokenizer.src_lang = "en"

source = tokenizer.convert_ids_to_tokens(tokenizer.encode("Hello world, it's a small test"))
target_prefix = [tokenizer.lang_code_to_token["fr"]]
results = translator.translate_batch([source], target_prefix=[target_prefix])
target = results[0].hypotheses[0][1:]

print(tokenizer.decode(tokenizer.convert_tokens_to_ids(target)))

but the model return ā€œku Š½Š°, itā€™s a small testā€ xD

1 Like
7

Here are some links for BLEU scores:

In general I think it can be difficult to compare scores unless your both using the same test andl language. We could try comparing Argos Translate Sacrebleu scores to other models for languages Sacrebleu supports.

2 Likes
8

yes, it can also be interesting to compare an old to a new model.

1 Like
9

it may be a problem with ctranslate2 or a bad use of the model.
but if already with ctranslate2 it doesnā€™t work it wonā€™t work with argos

I didnā€™t look any further but I couldnā€™t get it to work either just by replacing the m2m model, there must be some other adjustment to make.
it would be good to succeed in making it work because if it offers the same results as m2m but being 3x smaller and 4x faster, itā€™s super interesting.

1 Like
10
2 Likes
11

Interesting, good to see it works. I donā€™t think this would be a good candidate for including in Argos Translate. Iā€™m planning to stick with the SentencePiece tokenizer and donā€™t want to include a bunch of custom tokenizers. I think SentencePiece is the most popular tokenization library so there will probably be plenty of models coming out that support it. Iā€™m also looking into training my own custom Argos Translate multilingual models too where compatibility wouldnā€™t be an issue.

2 Likes