Routing Drift Alone Does Not Diagnose Failure in Merged MoE LLMs

📅 2026-09-26
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
This study addresses whether routing drift in Mixture-of-Experts (MoE) model merging is equivalent to routing failure and when intervention is necessary. We propose redefining routing failure through the lens of task loss recoverability. Leveraging a cross-model routing analysis toolkit, we employ counterfactual interventions and token-level attribution to demonstrate that most expert reassignments stem from input shifts rather than functional degradation. Based on these findings, we introduce a Selective Router Repair (SRR) strategy. This work challenges the prevailing misconception that drift inherently implies failure, showing that naively restoring source routing does not reliably improve performance. By releasing open-source tools and code, we provide theoretical foundations for precise post-merging diagnosis and repair in MoE architectures.
📝 Abstract
Model merging efficiently combines specialized large language models (LLMs) without joint retraining, but can substantially alter expert routing in Mixture-of-Experts (MoE) models. Such \emph{routing drift} is often interpreted as routing failure, raising a fundamental question that remains unclear: \emph{does routing drift after MoE merging actually indicate routing failure, and what evidence should justify repair?} We investigate these questions across DeepSeekMoE, OLMoE, and Qwen3-MoE proposing a routing analysis toolkit for controlled counterfactual interventions and token-level analysis. By crossing source and merged router inputs and parameters, we attribute most expert reassignments to input shifts rather than parameter changes at the same layer. However, source-relative routing differences poorly predict next-token likelihood gains from source-route restoration, and different expert selections can produce directionally similar mixture outputs. We therefore operationalize routing failure as \textit{task loss recoverable under a specified routing intervention, with non-routing parameters fixed.} These tests detect recoverable loss under deliberate router corruption, whereas source-route restoration does not establish reliable task benefits in the evaluated merged models. Motivated by these, we propose \emph{Selective Router Repair (SRR)} as a case study, and find that source-specialist token-likelihood advantages do not reliably identify beneficial local corrections. Together, these findings show that \textbf{routing drift alone is insufficient evidence of routing failure}: source-informed corrections must be judged by their task-level intervention effects. The analysis toolkit and SRR code are released.
Problem

Research questions and friction points this paper is trying to address.

Model Merging
Mixture-of-Experts
Routing Drift
Routing Failure
Large Language Models
Innovation

Methods, ideas, or system contributions that make the work stand out.

Mixture-of-Experts
Model Merging
Routing Drift
Counterfactual Intervention
Selective Router Repair
🔎 Similar Papers
No similar papers found.