机器学习初学者 22小时前
【深度学习】图解Transformer和MoE的差别
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

专家混合(MoE)架构是一种改进Transformer模型的策略,通过引入多个“专家”来提升模型性能,例如DeepSeek V3和R1。MoE在解码器块中使用专家,而非Transformer的前馈网络,从而加速推理过程。路由器负责选择合适的专家,但训练过程中面临专家选择不均等的问题。文章探讨了MoE的挑战及解决方案,包括添加噪声、限制专家处理的token数量,以确保所有专家都能得到充分训练。Mixtral 8x7B是基于MoE的知名LLM。

💡 MoE与Transformer的区别在于解码器结构:Transformer使用前馈网络,而MoE使用更小的“专家”网络。推理时,仅激活部分专家,从而提升速度。

⚙️ 路由器是MoE的核心组件:它像一个多分类分类器,为专家生成softmax分数,并选择前K个专家。路由器与网络共同训练,学习如何选择合适的专家。

⚠️ 训练中面临的挑战:专家选择不均等,导致部分专家训练不足。通过在路由器前馈输出中添加噪声,以及限制每个专家处理的token数量来解决。

🚀 MoE的优势:虽然MoE拥有更多参数,但每次仅激活部分参数,使其推理速度更快。Mixtral 8x7B是基于MoE的知名LLM,证明了MoE的潜力。

2025-06-27 12:03 浙江

专家混合(MoE)是一种流行的架构,比如最近火爆天的 DeepSeek V3 和 R1 就是这类模型。它利用不同的“专家”来改进 Transformer 模型。

下面的示意图展示了它们与 Transformer 的不同之处。

图片Transformer 和 MoE 在 decoder 块中有所不同:

Transformer 使用前馈网络。

MoE 使用 experts,它们是前馈网络,但与 Transformer 中的网络相比更小。

在推理过程中,将选择专家的子集。这使得 MoE 中的推理速度更快。

图片由于网络包含多个解码器层:

文本会在不同的层中经过不同的专家。

每个 token 选择的专家也各不相同。

但是,模型如何决定哪些专家是理想的呢?

这由路由器(Router)来完成。接下来我们来讨论它。

图片路由器就像一个多分类分类器,它对专家生成 softmax 分数。根据这些分数,我们选择前 K 个专家。

路由器与网络一起训练,并学习如何选择最合适的专家。

但这并不简单。让我们来看看其中的挑战!

图片挑战 1)注意训练初期的这一模式:

模型选择“专家 2”

该专家变得稍微更好

可能会再次被选中

该专家学到更多知识

又被选中

学到更多知识

如此循环!

许多专家因此训练不足!

图片我们通过两个步骤来解决这个问题:

在路由器的前馈输出中添加噪声,使其他专家的 logits 更高。

将除前 K 个之外的所有 logits 设为负无穷大,这样在 softmax 之后,这些分数就变为零。

这样,其他专家也有机会参与训练。

图片挑战 2)某些专家可能会比其他专家处理更多的 token,导致部分专家训练不足。

我们通过限制每个专家可处理的 token 数量来避免这种情况。

如果某个专家达到上限,输入的 token 就会被传递给下一个最合适的专家。

MoE 具有更多的参数需要加载,但由于每次仅选择部分专家,因此只有一部分参数被激活。

这使得推理速度更快。@MistralAI 的 Mixtral 8x7B 就是一个基于 MoE 的知名大型语言模型(LLM)。

下面是对比 Transformer 和 MoE 的示意图!

图片(完)

阅读原文

跳转微信打开

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

MoE Transformer 专家混合 Mixtral 8x7B
相关文章