专家混合模型

💡 原文中文,约1400字,阅读约需4分钟。
📝

内容提要

该文综述混合专家(MoE)模型研究:Mistral 7B采用GQA与滑动窗口注意力提升推理效率,其Instruct版性能超越Llama 2 13B Chat;构建中文Mixtral-8x7B模型Aurora,验证指令微调的有效性;提出融合MoE层的多语言Conformer以降低WER;并涉及MoE多语言生成训练、专家修剪、参数卸载(使Mixtral-8x7B可在消费级硬件运行)、Mistral 7B医学翻译微调、三维混合并行训练、Mixture of Tokens及自适应门控MoE等。

🔎

延伸解读

MoE 效率优化的两条路径

文章展示了提升 MoE 效率的两种思路:一是参数卸载,利用 MoE 固有属性让 Mixtral-8x7B 在消费级硬件和免费 Colab 上运行;二是专家修剪,移除多语言翻译中不相关的习惯用语并检测语种专家,使模型能在单 GPU 上运行。两者都试图降低部署门槛,但前者侧重内存调度,后者侧重模型精简。

多语言场景下的 MoE 实证

多语言 Conformer 融合 MoE 层后,在 12 种语言上相对 WER 改进 11.9%,且不使用语言信息时与使用 Ground Truth 的适配器模型效果相近,多语言浅层融合再带来约 3% 的相对改进。这说明 MoE 能在不显式依赖语言标签的情况下提升多语言识别效率,但改进幅度仍属渐进。

中文指令微调的验证意义

Aurora 基于 Mixtral-8x7B 构建,通过对三个中文指令数据集进行系统研究、预处理和整合,验证了指令微调对中文对话能力的有效性。文章称该工作在稀疏专家组合模型的指令微调方面具有开创性意义,但结论主要来自该模型与数据集的实验,读者不宜直接外推到所有 MoE 模型。

训练与路由的替代设计

文章提到三维混合并行结合张量、专家和数据并行,用于内存与通信优化,提升 MoE 训练效率和精度;Mixture of Tokens 则在专家之前混合不同示例的标记,避免 MoE 的困难并保留其优点;自适应门控 MoE 用可变数量专家处理令牌,在减少训练时间的同时保持推理质量。这些设计分别针对训练系统、路由方式和稀疏度控制。

❓

Q&A

Mistral 7B 模型在推理效率上做了哪些优化?

Mistral 7B 通过使用分组查询注意力(GQA)和滑动窗口注意力(SWA)来提高推理效率。

Aurora 模型是什么?它在中文对话能力上有什么验证结果?

Aurora 是一个基于 Mixtral-8x7B 的稀疏专家组合模型,通过整合三个中文指令数据集进行指令微调,验证了其在中文对话能力上的有效性,并在稀疏专家组合模型的指令微调方面具有开创性意义。

如何让 Mixtral-8x7B 模型在消费级硬件上运行?

通过参数卸载算法和利用 MoE LLMs 的固有属性,提出一种新策略,使消费者硬件和免费 Google Colab 实例上能够运行 Mixtral-8x7B 型号的 MoE 语言模型。

在多语言 Conformer 模型中融合 MoE 层有什么效果?

融合 MoE 层提高了推理效率,在 12 种语言上评估,相对于基线获得 11.9% 的相对 WER 改进;在不使用语言信息的前提下,与使用 Ground Truth 信息的适配器模型相比,实现了类似的 WER 和相似数量的参数激活,并通过多语言浅层融合实现了约 3% 的相对 WER 改进。

Mixture of Tokens 方法相比传统 MoE 有什么优势?

Mixture of Tokens 是一种全可微分的模型,在避免 Mixture of Experts 模型困难的同时保留了其优点,通过在专家之前混合来自不同示例的标记,使模型能够从所有标记-专家组合中学习,并且与大型语言模型的训练和推理完全兼容。

自适应门控混合专家模型如何提高训练效率?

自适应门控混合专家模型通过采用可变数量的专家处理令牌,实现了稀疏度的保持和训练效率的提高,实验验证了这种方法在减少训练时间的同时保持推理质量。

🏷️

标签

➡️

继续阅读