Expert-Aware Causal Tracing of Factual Recall in Sparse MoE Language Models

Yuetian Lu, Ali Modarressi, Yihong Liu, Hinrich Schütze
6/2/2026
cs.CLcs.LG

No confident code implementation found for this paper.

Abstract

Causal tracing of factual recall has been studied predominantly in dense transformer language models, where interventions localize information flow to layers or feed-forward modules. Sparse mixture-of-experts (MoE) language models introduce a sharper question: when a factual prediction is mediated by a routed MoE block, which routed expert contributions matter? We formulate expert-aware causal tracing for sparse MoE language models. Using CounterFact facts, we first corrupt the model's factual preference by adding noise to subject-token embeddings, and then test whether clean MoE-block outputs or clean expert-level updates restore the true-vs-foil logit contrast. For Qwen3-30B-A3B-Base, a layer sweep selects and validates layer 44, and expert-level tracing identifies L44E069 as an expert repeatedly selected in the clean run whose held-out patch outperforms other active same-layer expert patches. For Mixtral-8x7B-v0.1, layer-level tracing validates a mid-layer signal, but the signal is not localized to the selected singleton expert; a coalition check instead recovers it with routed multi-expert updates. These results suggest that MoE factual tracing can be made expert-aware, while also showing that expert-level localization is model- and protocol-dependent rather than universal.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{lu2026expertaware,
  title  = {Expert-Aware Causal Tracing of Factual Recall in Sparse MoE Language Models},
  author = {Yuetian Lu and Ali Modarressi and Yihong Liu and Hinrich Schütze},
  year   = {2026},
  eprint = {2606.03780},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.03780v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.03780.svg)](https://codeofpaper.com/paper/2606.03780)

Discussion