File size: 3,656 Bytes
b87ec2a
 
 
9b0719a
b87ec2a
 
cda64c8
b87ec2a
6e295ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c9e0fe7
6e295ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b87ec2a
 
 
9481193
 
 
 
 
 
 
 
 
 
 
 
b87ec2a
 
 
 
2262c8c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
---
license: llama2
datasets:
- MaLA-LM/mala-monolingual-split
base_model:
- meta-llama/Llama-2-7b-hf
library_name: transformers
---

# EMMA-500: Enhancing Massively Multilingual Adaptation of Large Language Models

## Model Description

**EMMA-500** is a state-of-the-art multilingual language model designed to improve language representation, especially in low-resource languages, through continual pre-training on the **Llama 2 7B** architecture. Leveraging the **MaLA Corpus**, which spans over 500 languages and 74 billion tokens, EMMA-500 excels in multilingual tasks like commonsense reasoning, machine translation, open-ended generation, and text classification.

**EMMA-500** outperforms other Llama 2-based models in diverse multilingual settings while maintaining robustness in specialized tasks.

---

## Model Details

- **Architecture**: Built on Llama 2 7B with enhanced language adaptation through continual pre-training.
- **Languages**: Supports **546 languages** with substantial training data (over 100k tokens each).
- **Data Mix**: A diverse mix of text from domains like code, books, instruction data, and more.
- **Key Tasks**: Commonsense reasoning, machine translation, text classification, natural language inference, code generation, and open-ended generation.

### Data Access
- [MaLA Corpus](https://huggingface.co/collections/MaLA-LM/mala-corpus-66e05127641a51de34d39529)
- [PolyWrite Benchmark](https://huggingface.co/datasets/MaLA-LM/PolyWrite)

---

## Usage

You can use **EMMA-500** for multilingual text generation. Below is an example to generate text using the model:

```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "MaLA-LM/emma-500-llama2-7b"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

input_text = "Once upon a time"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)

print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```

---

## Model Performance

**EMMA-500** was evaluated across multiple benchmarks and tasks, demonstrating:

- **Lowest negative log-likelihood** in intrinsic evaluations.
- Significant improvements in **commonsense reasoning**, **machine translation**, and **open-ended generation**.
- **Outperformed** all Llama 2-based models in **text classification** and **natural language inference**.
- Enhanced performance in **code generation** and **machine reading comprehension (MRC)**.

Challenges remain in low-resource languages, where the model tends to have higher **Self-BLEU** scores, indicating reduced output diversity.

---


## Citation

```
@article{ji2024emma500enhancingmassivelymultilingual,
      title={{EMMA}-500: Enhancing Massively Multilingual Adaptation of Large Language Models}, 
      author={Shaoxiong Ji and Zihao Li and Indraneil Paul and Jaakko Paavola and Peiqin Lin and Pinzhen Chen and Dayyán O'Brien and Hengyu Luo and Hinrich Schütze and Jörg Tiedemann and Barry Haddow},
      year={2024},
      journal={arXiv preprint 2409.17892},
      url={https://arxiv.org/abs/2409.17892}, 
}
```

## Acknowledgements

We extend our thanks to the language communities and contributors who helped source, clean, and validate the diverse data used in the MaLA Corpus. Their efforts are invaluable in supporting linguistic diversity in AI research.

This work is done by researchers at [Helsinki-NLP](https://huggingface.co/Helsinki-NLP) in collaboration with partners from TU Darmstadt, the University of Edinburgh, and LMU Munich. It is funded by [HPLT](https://hplt-project.org) and [UTTER](https://he-utter.eu).