arXiv:2605.19378cs.CV2026-05被引 1

诊断视觉扩散模型中专家路由失败机制并提出改进路线图

Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock

论文配图:Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock
图 1 · 摘自论文原文
  • 通过分析路由决策时序,发现专家路由存在五类失效模式
  • 揭示选择性死锁是专家成熟前的理性等待策略,非单纯故障
  • 提出从稠密到稀疏专家架构的三定律与精度陷阱解决方案

本文系统诊断了视频扩散变换器中基于Token-Choice的稀疏专家混合(MoE)训练失败模式。从约50亿参数的预训练稠密模型出发,遵循三条转换法则:路由专家完全复制原前馈网络权重,共享专家初始化为零以验证,再设为极小非零噪声用于实际训练,仅门控网络随机初始化。实验揭示五类失效模式:(1) 线性路由导致全局软饱和,专家完全同质化;(2) MLP路由引发选择性死锁,约三分之一层退化为单专家模式,增加辅助损失无法缓解;(3) 交叉注意力路由具初步自恢复能力,但仍有约九层顽固死锁;(4) 死锁层呈U型分布,集中于浅层视觉处理与深层语义融合层;(5) bfloat16混合精度导致微小权重更新被硬件截断为零。基于跨5000步、6500万令牌的路由决策时间序列,提出功能冗余假说:死锁是门控-共享-路由专家三元系统中共享专家成熟前的理性等待策略,获系统生物学功能冗余理论支持。工程上总结稠密转MoE三定律,提供完整bfloat16精度陷阱应对方案,校准当前Token-Choice范式的能力建界,并勾勒从视觉统一到世界模型的三步演化路线。

原文摘要 · Abstract (English)

This paper systematically diagnoses the training failure modes of Token-Choice sparse Mixture-of-Experts (MoE) on video Diffusion Transformers. Starting from a pretrained dense model of about 5 billion parameters, we convert it into an MoE architecture following three laws: routed experts exactly clone the original FFN weights, shared experts are initialized to zero for verification and then to extremely small non-zero noise for actual training, while only the gating networks start from random initialization. Experiments reveal a hierarchy of five failure modes: (1) linear routers suffer global soft saturation with complete expert homogenization; (2) MLP routers introduce selective deadlock, where roughly one-third of layers degenerate into a single-expert mode that cannot be prevented by increasing the auxiliary loss; (3) cross-attention routers exhibit preliminary self-recovery, yet about nine layers remain stubbornly deadlocked; (4) deadlocked layers display a U-shaped distribution, concentrated in shallow visual processing layers and deep semantic integration layers; (5) bfloat16 mixed precision causes tiny weight updates to be truncated to zero by hardware. Based on routing decision time series over 65 million tokens across 5,000 training steps, we propose the Functional Redundancy Hypothesis: deadlock is a rational waiting strategy before the shared expert matures within the gate-shared expert-routed expert triadic system. This hypothesis is supported by the theory of functional redundancy in systems biology. On the engineering side, we summarize the Three Laws of dense-to-MoE conversion and provide a complete solution for the bfloat16 precision trap. We calibrate the current capability boundary of the Token-Choice paradigm and outline a three-step evolutionary roadmap from visual unification to a world model.

MoE路由扩散模型专家系统训练失效

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。