跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

路径约束的专家混合模型

文章摘要

本文提出从“专家路径”的视角理解稀疏专家混合(MoE)模型,指出尽管存在N^L条可能路径,实际中token只聚集在少数与语言功能对齐的路径上,大部分路径未被探索,造成统计低效。为此作者引入PathMoE,通过跨连续层共享路由参数来约束有效路径空间,实验表明在0.9B和16B参数模型上,PathMoE在困惑度和下游任务上均优于独立路由,且无需辅助损失。

路径约束的专家混合模型
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

稀疏混合专家(MoE)架构在每一层都会独立地为每个词元选择一部分专家进行处理。苹果与谷歌的研究人员提出了一种新的分析视角:把MoE的计算过程看作“专家路径”,也就是一个词元在全部层中所做出的专家选择序列。这一视角揭示出一个有趣的现象:当模型有N个专家、L层时,理论上存在N^L条可能的路径,但在实际训练中,词元往往只聚集在少数与语言功能相对应的路径上,而绝大多数路径从未被探索,这构成了一种统计效率上的浪费。

正是基于这一观察,作者提出了路径约束型架构,用以压缩有效路径空间,从而放大这种自然形成的集中趋势。作为该思路的一个具体实现,他们推出了PathMoE:在连续若干层组成的层块之间共享路由器参数。这样做的效果是,模型在层与层之间的专家选择更具一致性,路径结构更加清晰,路由结果对扰动的抵抗力也更强。

论文的分析确认,PathMoE确实强化了涌现出的路径结构:路径簇更加集中,跨层一致性更好,并且在路由受到扰动时更为稳健。研究团队在0.9B和16B参数规模的模型上进行了实验,结果显示PathMoE在困惑度和下游任务上均优于独立路由的基线,同时不再需要依赖辅助损失函数来维持负载均衡。

作者指出,这些结果说明“专家路径”可以作为MoE架构设计的一个有益维度,与现有的独立路由机制研究形成互补而非替代关系。该论文由Zijin Gu、Tatiana Likhomanenko、Vimal Thilak、Jason Ramapuram和Navdeep Jaitly共同完成,部分工作是在Apple期间开展的,论文于2026年7月发表。

这一工作延续了该团队此前在路由共享方面的探索。例如2025年7月发表的Omni-Router研究了语音识别中的路由决策共享问题,发现多数层的路由器所做的专家选择与其他层关联不强,因而尝试加强跨层专家之间的协作。更早的Duo-LLM工作则关注大语言模型中的自适应计算问题。PathMoE可以视为这一系列思路在语言模型MoE架构上的进一步延伸,将跨层路由的一致性问题提升为架构设计的核心考量。

展开要点与分析

文章情报

工程师进阶

要点

  • MoE中token的专家选择序列(专家路径)高度集中,但绝大多数路径未被利用,存在统计低效。
  • PathMoE通过跨层块共享路由器参数来约束路径空间,放大自然形成的路径结构。
  • 在0.9B和16B参数模型上,PathMoE提升了困惑度和下游任务表现,并消除了对辅助损失的需求。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。