日本研究团队利用理化学研究所的超级计算机"Fugaku"发布了具有增强日语能力的大型语言模型--Fugaku-LLM。该团队由东京工业大学的横田理夫教授、东北大学的坂口圭介副教授、富士通公司的白畑幸一、理化学研究所的穆罕默德-瓦希卜(MohamedWahib)组长、名古屋大学的西口浩二副教授、CyberAgent公司的佐佐木翔太(ShotaSasaki)和KotobaTechnologies公司的小岛典之(NoriyukiKojima)领导。
为了在Fugaku上训练大型语言模型,研究人员开发了分布式训练方法,包括将深度学习框架Megatron-DeepSpeed移植到Fugaku,以优化变形金刚在Fugaku上的性能。他们加速了Transformers的密集矩阵乘法库,并通过结合三种并行化技术优化了Fugaku的通信性能,还加速了Tofu互联D上的集体通信库。
Fugaku-LLM有130亿个参数,比日本广泛开发的70亿个参数模型更大。Fugaku-LLM增强了日语能力,在日语MT-Bench中平均得分5.5,是使用日本原始数据训练的开放模型中最高的。特别是人文和社会科学任务的基准性能达到了9.18分的惊人高分。
Fugaku-LLM是在CyberAgent收集的日语专有数据、英语数据和其他数据的基础上进行训练的。Fugaku-LLM的源代码可在GitHub上获取,模型可在HuggingFace上获取。只要用户遵守许可证,Fugaku-LLM可用于研究和商业目的。
未来,随着更多研究人员和工程师参与改进模型及其应用,训练效率将得到提高,从而实现下一代创新研究和商业应用,例如科学模拟与生成式人工智能的联动,以及拥有成千上万人工智能的虚拟社区的社会模拟。
研究背景
近年来,大型语言模型(LLM)的发展十分活跃,尤其是在美国。其中,由OpenAI开发的ChatGPT
因此,东京工业大学、东北大学、富士通公司、理化学研究所、名古屋大学、CyberAgent和KotobaTechnologies启动了一项关于开发大型语言模型的联合研究项目。
各机构/公司的作用
未来发展
这项研究的成果将通过GitHub和HuggingFace公开,以便其他研究人员和工程师可以利用这些成果进一步开发大型语言模型。只要用户遵守许可协议,Fugaku-LLM可用于研究和商业目的。从2024年5月10日起,Fugaku-LLM还将通过富士通研究门户网站提供给用户。
未来,随着越来越多的研究人员和工程师参与到模型及其应用的改进中来,训练的效率将得到提高,从而实现下一代创新研究和商业应用,例如科学模拟与生成式人工智能的联系,以及拥有成千上万人工智能的虚拟社区的社会模拟。
本研究得到了Fugaku政策支持提案"利用Fugaku开发大型语言模型的分布式并行训练"(提案号:hp230254)的支持。
【查看完整讨论话题】 | 【用户登录】 | 【用户注册】