AI Community Models

Model Technical Details

PERFORMANCE METRICS

Achieves a Bleu Score of 13.02% on test set of 5,000 parallel sentences.

13% Bleu Score – A method for assessing the quality of text that has been machine-translated from one language to another. The closer the machine translation is to expert human translation, the better the score.

OVERVIEW

This model uses the Google Transformer architecture, which is currently the basis for many state-of-the-art translation models. The essence of the Transformer model is the encode-decoder architecture with Attention. Multiple encoders are stacked on top of each other with each one consisting of a self-attention layer, which tries to take into account the full sentence when translating instead of just the word that it is looking at, and a feed-forward neural network. Word embeddings are fed through these encoding layers and are then passed into the decoding layers. In the decoding layer, the self-attention layer only pays attention to earlier positions as opposed to the encode, which allows both directions. Otherwise they work the same way except that the decoding layer attempts to decode the input into the output language. All units have an add and a normalize layer as well. Additionally, in the standard transformer model, there are eight attention heads which are used. The results of these are concatenated into the feed-forward network and then reduced to the correct size. Positional encoding is also used to account for the order of words in the input sequence. This model was trained for 200,000 steps on 4 gpus. The implementation was created using the open source opennmt framework available at opennmt.net.

TRAINING

This model is trained on the Global Voices, Gnome, Infopankki, JW300, KDE4, Opensubtitles, QED, Tanzil, Ted, Tep, Ubuntu and Wikipedia corpora. These total 8,511,738 lines of parallel text. These texts can be found at opus.nlpl.eu.

VALIDATION

This model was validated on 5,000 parallel sentences and achieves a Bleu Score of 13.02%.

INPUT SPECIFICATION

The input(s) to this model must adhere to the following specifications:

OUTPUT DETAILS

This model will output the following:

The “results.json” file will contain the translated text in the following format: {“text”: “text translated from Persian”}