Pretrained language models underpin several AI applications, but their high computational cost for training limits accessibility. Initiatives such as BLOOM and StarCoder aim to democratize access to pretrained models for collaborative community development. However, such existing models face challenges: limited multilingual capabilities, continual pretraining causing catastrophic forgetting, whereas pretraining from scratch is computationally expensive, and compliance with AI safety and development laws. This paper presents Aurora-M, a 15B parameter multilingual open-source model trained on English, Finnish, Hindi, Japanese, Vietnamese, and code. Continually pretrained from StarCoderPlus on 435 billion additional tokens, Aurora-M surpasses 2 trillion tokens in total training token count. It is the first open-source multilingual model fine-tuned on human-reviewed safety instructions, thus aligning its development not only with conventional red-teaming considerations, but also with the specific concerns articulated in the Biden-Harris Executive Order on the Safe, Secure, and Trustworthy Development and Use of Artificial Intelligence. Aurora-M is rigorously evaluated across various tasks and languages, demonstrating robustness against catastrophic forgetting and outperforming alternatives in multilingual settings, particularly in safety evaluations. To promote responsible open-source LLM development, Aurora-M and its variants are released at https://huggingface.co/collections/aurora-m/aurora-m-models-65fdfdff62471e09812f5407 .

Aurora-M: The First Open Source Multilingual Language Model Red-teamed according to the U.S. Executive Order / Nakamura, Taishi; Mishra, Mayank; Tedeschi, Simone; Chai, Yekun; T Stillerman, Jason; Friedrich, Felix; Yadav, Prateek; Laud, Tanmay; Minh Chien, Vu; Yue Zhuo, Terry; Misra, Diganta; Bogin, Ben; Vu, Xuan-Son; Karpinska, Marzena; Varma Dantuluri, Arnav; Kusa, Wojciech; Furlanello, Tommaso; Yokota, Rio; Muennighoff, Niklas; Pai, Suhas; Adewumi, Tosin; Laippala, Veronika; Yao, Xiaozhe; Junior, Adalberto; Ariyak, Alpay; Drozd, Aleksandr; Clive, Jordan; Gupta, Kshitij; Chen, Liangyu; Sun, Qi; Tsui, Ken; Persaud, Noah; Fahmy, Nour; Chen, Tianlong; Bansal, Mohit; Monti, Nicolo; Dang, Tai; Luo, Ziyang; Bui, Tien-Tung; Navigli, Roberto; Mehta, Virendra; Blumberg, Matthew; May, Victor; Nguyen, Huu; Pyysalo, Sampo. - (2024).

Aurora-M: The First Open Source Multilingual Language Model Red-teamed according to the U.S. Executive Order

Simone Tedeschi;Roberto Navigli;
2024

Abstract

Pretrained language models underpin several AI applications, but their high computational cost for training limits accessibility. Initiatives such as BLOOM and StarCoder aim to democratize access to pretrained models for collaborative community development. However, such existing models face challenges: limited multilingual capabilities, continual pretraining causing catastrophic forgetting, whereas pretraining from scratch is computationally expensive, and compliance with AI safety and development laws. This paper presents Aurora-M, a 15B parameter multilingual open-source model trained on English, Finnish, Hindi, Japanese, Vietnamese, and code. Continually pretrained from StarCoderPlus on 435 billion additional tokens, Aurora-M surpasses 2 trillion tokens in total training token count. It is the first open-source multilingual model fine-tuned on human-reviewed safety instructions, thus aligning its development not only with conventional red-teaming considerations, but also with the specific concerns articulated in the Biden-Harris Executive Order on the Safe, Secure, and Trustworthy Development and Use of Artificial Intelligence. Aurora-M is rigorously evaluated across various tasks and languages, demonstrating robustness against catastrophic forgetting and outperforming alternatives in multilingual settings, particularly in safety evaluations. To promote responsible open-source LLM development, Aurora-M and its variants are released at https://huggingface.co/collections/aurora-m/aurora-m-models-65fdfdff62471e09812f5407 .
2024
File allegati a questo prodotto
Non ci sono file associati a questo prodotto.

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11573/1724729
 Attenzione

Attenzione! I dati visualizzati non sono stati sottoposti a validazione da parte dell'ateneo

Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus ND
  • ???jsp.display-item.citation.isi??? ND
social impact