Skip to content
Preprint

RARE: Decoupling Representation Steering from Expert Routing in Mixture-of-Experts Language Models

Aug 2026 · 0 citations · 27 references
Computer Science

TL;DR

It is found that preserving clean routing substantially recovers steering performance and that routing is more sensitive to semantic content than to behavioral changes under controlled content, which supports routing consistency as an important architectural consideration for adapting representation engineering to MoE models.

Abstract

Representation engineering offers a lightweight means of controlling language-model behavior by modifying intermediate hidden states, but its direct application to Mixture-of-Experts (MoE) models introduces a structural mismatch. We first verify this failure mode through a series of empirical studies and find that preserving clean routing substantially recovers steering performance and that routing is more sensitive to semantic content than to behavioral changes under controlled content. Motivated by these findings, we introduce RARE, a router-agnostic representation engineering framework for MoE language models. RARE projects arbitrary behavioral perturbations onto the null space of the router matrix, thereby removing router-visible components, and further corrects routing drift propagated to selected downstream layers. To decide the best perturbation estimator in this framework, we evaluate five estimators on six heterogeneous open-weight MoE models across three steering scenarios: harmfulness, truthfulness, and factual editing. On harmfulness steering, RARE reaches an average attack success rate of 53.3% while retaining 67.8% MMLU accuracy, yielding a stronger aggregate effectiveness--utility trade-off than baselines. It further improves average TruthfulQA MC1 accuracy from 41.0% to 58.6% and CounterFact efficacy from 16.8% to 96.3%. These results support routing consistency as an important architectural consideration for adapting representation engineering to MoE models.

View source

Similar papers

Multi-level context Modeling for consistent expert selection in Mixture-of-Experts

This work revisit expert selection from a representation perspective and identifies context incompleteness as a key bottleneck limiting effective expert specialization, proposing Multi-level Context Fusion MOE (MCF-MOE), a framework that constructs context-aware representations by integrating complementary signals from...

Shuhan Huang, Naifan Zhang, Yuan-Bo Tang et al. · 0 citations
#artificial intelligence Preprint Sep 2026

SEAL: Reinforcing Global Safety in Mixture-of-Experts through Shared Expert ALignment

Mixture-of-Experts (MoE) is a scaling architecture for large language models that activates only a small subset of expert modules per token, enabling massive parameter growth with nearly constant computation. Recent Hybrid MoE architecture adds \textit{shared experts} to capture consistently useful representations, fur...

Qingyu Meng, Yiwei Zha, Jia-Huan Pei et al. · 0 citations
Jul 2026

It Takes a MAESTRO To Prune Bad Experts

MAESTRO (Markov-chain Approximated Expert Sparsification via Transition-based ROuting), a structured pruning framework designed for MoE architectures that models autoregressive expert activation trajectories as Ergodic Markov chains whose stationary distributions encode cross-layer dependencies, yielding a globally awa...

Palaash Goel, Ayush Maheshwari, Tanmoy Chakraborty · 1 citation
Jul 2026

Grounding latent algorithm routing in transformer reasoning

These results provide controlled evidence that dense transformers trained on ROUTEBENCH can develop route-like internal variables, but they do not establish universal routing in pretrained language models or unrestricted natural-language reasoning.

Xiangbo Zhang, Xiaoxu Ma · 1 citation
Jul 2026

CausalGate: Causal Importance Distillation for Transformer Module Pruning

This work introduces CausalGate, an intervention-guided framework for compute-efficient transformer inference, which consistently outperforms prominent dynamic routing and layer-skipping baselines, translating theoretical compute savings into concrete hardware latency reductions with zero operational overhead.

K. Nair, Smriti Regmi, Rodrigue Rizk · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.