大模型 | LLM预训练和后训练新范式（三）Google 的 Gemma 2

3. Google 的 Gemma 2

论文地址：https://arxiv.org/abs/2408.00118

Google 的 Gemma 模型最近在 Gemma 2: Improving Open Language Models at a Practical Size 中进行了描述。

在讨论预训练和后训练过程之前，我将概述一些关键事实。

3.1 Gemma 2 概述

Gemma 2 模型有三种规模：20 亿、90 亿和 270 亿参数。主要关注点是探索不必增加训练数据集规模的技术，而是开发相对较小且高效的 LLMs。

值得注意的是，Gemma 2 具有 256k 词汇表大小。相比之下，Llama 2 使用的是 32k 词汇表，而 Llama 3 使用的是 128k 词汇表。

此外，Gemma 2 采用滑动窗口注意力机制，类似于 Mistral 的早期模型，可能是为了减少内存成本。有关 Gemma 2 架构的更多详细信息，请参阅我之前文章中的 Gemma 2 部分。

3.2 Gemma 2 预训练

Gemma 研究人员认为，即使是小模型也常常训练不足。然而，他们并不只是简单地增加训练数据集的规模，而是通过其他方法保持质量并取得改进，例如知识蒸馏，类似于苹果的做法。

尽管 27B Gemma 2 模型是从头开始训练的，但较小的模型是通过类似于苹果之前解释的方法进行知识蒸馏训练的。

27B 模型在 13 万亿个标记上进行了训练，9B 模型在 8 万亿个标记上进行了训练，2B 模型在 2 万亿个标记上进行了训练。此外，类似于苹果的做法，Gemma 团队优化了数据混合以提高性能。

Gemma 2 预训练技术总结。

3.3 Gemma 2 后训练

Gemma 模型的后训练过程包括典型的监督微调 (SFT) 和带有人类反馈的强化学习 (RLHF) 步骤。

指令数据使用了仅英文的提示对，这些提示对混合了人工生成和合成生成的内容。具体来说，值得注意的是，响应主要由教师模型生成，并且在 SFT 阶段也应用了知识蒸馏。

在 SFT 之后，他们的 RLHF 方法的一个有趣方面是用于 RLHF 的奖励模型比策略（目标）模型大十倍。

Gemma 使用的 RLHF 算法相当标准，但有一个独特之处：他们通过一种称为 WARP 的方法对策略模型进行平均，这是一种 WARM（权重平均奖励模型）的后继方法。我之前在我的文章 “模型合并、专家混合及走向更小的 LLMs” 中详细讨论了这种方法。

Gemma 2 后训练技术总结。

3.4 结论

Gemma 团队似乎非常重视知识蒸馏，他们在与 Apple 类似的预训练和后训练中都使用了这一技术。有趣的是，他们并没有采用多阶段预训练方法，或者至少他们在论文中没有详细说明这一点。

---

最后分享

AI大模型作为人工智能领域的重要技术突破，正成为推动各行各业创新和转型的关键力量。抓住AI大模型的风口，掌握AI大模型的知识和技能将变得越来越重要。

学习AI大模型是一个系统的过程，需要从基础开始，逐步深入到更高级的技术。

这里给大家精心整理了一份全面的AI大模型学习资源，包括：AI大模型全套学习路线图（从入门到实战）、精品AI大模型学习书籍手册、视频教程、实战学习、面试题等，资料免费分享！

1. 成长路线图&学习规划

要学习一门新的技术，作为新手一定要先学习成长路线图，方向不对，努力白费。

这里，我们为新手和想要进一步提升的专业人士准备了一份详细的学习成长路线图和规划。可以说是最科学最系统的学习成长路线。

2. 大模型经典PDF书籍

书籍和学习文档资料是学习大模型过程中必不可少的，我们精选了一系列深入探讨大模型技术的书籍和学习文档，它们由领域内的顶尖专家撰写，内容全面、深入、详尽，为你学习大模型提供坚实的理论基础。（书籍含电子版PDF）

3. 大模型视频教程

对于很多自学或者没有基础的同学来说，书籍这些纯文字类的学习教材会觉得比较晦涩难以理解，因此，我们提供了丰富的大模型视频教程，以动态、形象的方式展示技术概念，帮助你更快、更轻松地掌握核心知识。

4. 大模型项目实战

学以致用 ，当你的理论知识积累到一定程度，就需要通过项目实战，在实际操作中检验和巩固你所学到的知识，同时为你找工作和职业发展打下坚实的基础。

5. 大模型面试题

面试，不仅是技术的较量，更需要充分的准备。在你已经掌握了大模型技术之后，就需要开始准备面试，我们将提供精心整理的大模型面试题库，涵盖当前面试中可能遇到的各种技术问题，让你在面试中游刃有余。

全套的AI大模型学习资源已经整理打包，有需要的小伙伴可以微信扫描下方CSDN官方认证二维码，免费领取【保证100%免费】

如有侵权，请联系删除。