MoE架构详解:稀疏激活如何改变大模型经济学
MoE:用参数量换计算效率 MoE(Mixture of Experts)的核心思想很简单:不是所有参数都需要参与每次计算。通过路由机制,只激活与当前输入最相关的"专家"子网络,在保持大参数量的同时大幅降低计算量。 基本原理 稠密模型 vs 稀疏模型 传统稠密模型:每个token经过所有参数的计算。 输入token → [Attention] → [MLP(d, d_ff)] → 输出 所有参数参与计算 MoE模型:MLP层被替换为多个专家MLP,路由器选择激活哪些专家。 输入token → [Attention] → [Router] → 选择Top-K专家 ↓ [Expert1] [Expert2] ... [ExpertN] ↓ (仅K个被激活) 加权合并 → 输出 关键参数 专家数(N):总专家数量。通常8-64个 激活数(K):每个token激活的专家数。通常2-8个 稀疏率:1 - K/N。如8/64的配置,稀疏率为87.5% 以DeepSeek-V3为例:256个专家,每token激活8个,总参数671B但推理时只激活37B——等效于37B模型的计算量,但拥有671B参数的知识容量。 路由机制 基础路由 最简单的路由方式: def route(token, experts): # 计算token与每个专家的亲和度 scores = softmax(W_router @ token) # [N] # 选择Top-K个专家 top_k_indices = argsort(scores)[-K:] top_k_scores = scores[top_k_indices] # 归一化 top_k_scores /= top_k_scores.sum() # 加权合并 output = sum(experts[i](token) * top_k_scores[i] for i in top_k_indices) return output 负载均衡问题 基础路由有一个严重问题:路由坍缩。模型倾向于总是选择少数几个"好"专家,其他专家得不到训练。 ...