arXiv:2501.07124cs.LG2025-01被引 15

开源训练650亿参数大模型,全程透明可复现

LLM360 K2: Building a 65B 360-Open-Source Large Language Model from Scratch

  • 从零构建65B参数模型,全流程公开
  • 性能超LLaMA-65B,用更少算力和数据
  • 适合想复现或学习大模型训练的研究者

本文详细介绍了LLM360 K2-65B模型的训练过程,将360度开源理念扩展至最大规模的模型。尽管开源大模型持续进步,但顶级大模型如何训练仍不透明。因成本高昂,相关实现细节常被企业保密。本项目通过完整公开训练资源,填补这一空白。报告重点介绍首个模型K2 DIAMOND——一个650亿参数的大语言模型,其性能超越LLaMA-65B,接近LLaMA2-70B,且所需FLOPs和训练数据更少。文中详述了实现步骤,并对模型训练全过程进行纵向分析。此外还介绍了在研项目TXT360,为后续模型奠定基础。本项目践行透明、可复现、可访问的开源原则,助力资源密集型AI研究发展。

原文摘要 · Abstract (English)

We detail the training of the LLM360 K2-65B model, scaling up our 360-degree OPEN SOURCE approach to the largest and most powerful models under project LLM360. While open-source LLMs continue to advance, the answer to "How are the largest LLMs trained?" remains unclear within the community. The implementation details for such high-capacity models are often protected due to business considerations associated with their high cost. This lack of transparency prevents LLM researchers from leveraging valuable insights from prior experience, e.g., "What are the best practices for addressing loss spikes?" The LLM360 K2 project addresses this gap by providing full transparency and access to resources accumulated during the training of LLMs at the largest scale. This report highlights key elements of the K2 project, including our first model, K2 DIAMOND, a 65 billion-parameter LLM that surpasses LLaMA-65B and rivals LLaMA2-70B, while requiring fewer FLOPs and tokens. We detail the implementation steps and present a longitudinal analysis of K2 DIAMOND's capabilities throughout its training process. We also outline ongoing projects such as TXT360, setting the stage for future models in the series. By offering previously unavailable resources, the K2 project also resonates with the 360-degree OPEN SOURCE principles of transparency, reproducibility, and accessibility, which we believe are vital in the era of resource-intensive AI research.

大模型训练开源模型65B参数可复现

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。