New method proposed to interpret Mixture-of-Experts reward models via contribution contrast
CoCo aims to provide faithful, response-level explanations of MoE reward models by analyzing chosen-rejected response pairs, outperforming router-based and score-based alternatives in evaluations.
1 source · cross-referenced
- A new method called CoCo is introduced to interpret Mixture-of-Experts (MoE) reward models at the response level.
- CoCo uses chosen-rejected response pairs to characterize experts' roles, capturing both routing and preference behavior.
- Evaluations show CoCo produces more coherent, faithful, and specialized interpretations than router-based, score-based, and sparse autoencoder-based alternatives.
- The approach maintains competitive reward modeling accuracy while improving interpretability.
Researchers propose Contribution-Contrast (CoCo), a method to interpret Mixture-of-Experts (MoE) reward models at the response level. Unlike prior approaches that rely on routing weights to infer expert behavior, CoCo analyzes chosen-rejected response pairs with the largest contribution contrasts to jointly capture routing and preference behavior.
The authors argue that routing weights alone reveal only which prompts an expert receives, not how it judges responses, providing an incomplete picture of expert behavior. CoCo addresses this gap by directly examining how experts contribute to preference judgments between responses.
In evaluations, CoCo outperformed router-based, score-based, and sparse autoencoder-based interpretation methods in coherence, faithfulness, and specialization, as measured by both automatic and human assessments. The method also maintained competitive reward modeling accuracy relative to these alternatives.
The study is presented as the first systematic investigation of interpretation methods for MoE reward models, highlighting the need for faithful response-level explanations in reward modeling.
- Aug 10, 2026 · arXiv cs.AI
Researchers propose EntropyMoE, a sparse Mixture-of-Experts architecture for tokenizer-free LLMs
Trust79 - Aug 8, 2026 · Apple — Machine Learning Research
Apple proposes DLR-Lock to restrict fine-tuning of open-weight language models
Trust79 - Aug 8, 2026 · Apple — Machine Learning Research
Apple study compares diffusion and autoregressive language models on performance and scaling trade-offs
Trust79