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

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 .