Conditional Capacity and Routing in Mixture-of-Experts Particle Transformers

arXiv:2610.02701v1 Announce Type: new Abstract: Mixture-of-Experts (MoE) models can increase parameter capacity without proportionally increasing active computation, but it is unclear how this trade-off behaves in particle-physics transformers. We study dense and MoE Particle Transformers on 188-class JetClass-II, varying expert count, routing capacity, top-K, and auxiliary loss. We find that, when token dropping is avoided, top-1 MoE models improve over the dense baseline at nearly unchanged nominal forward compute, while further increasing the number of stored experts produces little additional accuracy gain. Activating multiple experts per token yields additional predictive improvements at higher computational cost. Routing analyses show that expert assignments become more strongly associated with particle identity and kinematics in some configurations, but this structure does not increase monotonically with classification performance. These results highlight the need to distinguish stored parameter capacity, active computation, routing capacity, and routing organization when evaluating sparse expert models for jet classification. Code and experiment configurations are available at https://github.com/kpendiyala/MPT.

Related

• MoRA: MoE Pruning via Router Bias Learning and Expert Approximation
• Uncertainty Is Not Enough: Value-of-Information Routing for Mixtures of LoRA Experts
• Beyond the Previous Layer: Residual Predictive Structure in Sparse MoE Routing
• Beyond Geometric Complementarity: Coherent Overlap in Sparse Mixture-of-Experts Routing

Source: arXiv cs.LG | 2026-10-05