mlsanityData to 14 Oct 2026

RAPTOR: Role-Aware Private Training for Mixture-of-Experts

arXiv 2609.05770 ↗First posted 4 Sep 2026cs.LG cs.CLcited by 0

On the map: Mixture-of-experts → Expert routing strategies

Abstract (arXiv)

Differentially private (DP) fine-tuning methods treat sparse Mixture-of-Experts (MoE) models as a single dense block, ignoring that shared layers see all data while experts only see routed records. We identify and formally characterize three resulting failure modes: global clipping suppresses expert gradients, batch-level normalization dilutes sparse expert updates, and fixed privacy noise degrades signal-to-noise ratio on low-load experts. We introduce RAPTOR - a Role-Aware Private Training framework, which alternates shared and expert optimization and targets each failure directly, using expert-specific clipping and noise together with a public expected-owner denominator and a count-independent update schedule that avoids conditioning on private, realized expert counts. We prove the resulting mechanism satisfies (varepsilon,δ)-DP: because each record is assigned to exactly one owner expert, per-expert mechanisms within a layer compose in parallel, so updating all E experts costs no more, in privacy terms, than updating one, with shared and expert streams composing sequentially across training. We further derive a bias-variance decomposition of the public-denominator estimator showing its bias grows predictably with routing imbalance, yielding a privacy-free rule for selecting which layer to protect from routing entropy measured on a small public corpus. Experiments on Switch Transformer and OLMoE fine-tuning across GLUE tasks, and on DeepSeek-VL2-Tiny, show consistent gains over standard DP baselines across several privacy levels (varepsilon), with the largest margins typically at the tightest budgets. Code and models are publicly available: https://github.com/leduckhai/RAPTOR

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 28 of its 40 references that are in the map. First posted on arXiv 4 Sep 2026.

Builds on

28 of 40 references.

Closest ancestors · 4

Same direction, newest first

  1. 2024Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Experts28 Aug
  2. 2024Differentially Private Training of Mixture of Experts Models11 Feb
  3. 2022Mixture-of-Experts with Expert Choice Routing18 Feb
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan

Other references · 24

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
  3. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  4. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  5. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  6. 2016Deep Learning with Differential Privacycited by 2,728
  7. 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
  8. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
  9. 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
  10. 2021Membership Inference Attacks From First Principlescited by 878
  11. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
  12. 2019Adversarial NLI: A New Benchmark for Natural Language Understandingcited by 807
12 more
  1. 2021Scaling Vision with Sparse Mixture of Expertscited by 747
  2. 2024DeepSeek-VL2: Mixture-of-Experts Vision-Language Models for Advanced Multimodal Understandingcited by 572
  3. 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
  4. 2021Differentially Private Fine-tuning of Language Modelscited by 400
  5. 2021Opacus: User-Friendly Differential Privacy Library in PyTorchcited by 383
  6. 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 320
  7. 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
  8. 2021Hash Layers For Large Sparse Modelscited by 237
  9. 2019Differentially Private Learning with Adaptive Clippingcited by 220
  10. 2021Numerical Composition of Differential Privacycited by 180
  11. 2016Max-Information, Differential Privacy, and Post-Selection Hypothesis Testingcited by 27
  12. 2024DiSK: Differentially Private Optimizer with Simplified Kalman Filter for Noise Reductioncited by 11

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026MetaMoE: Diversity-Aware Proxy Selection for Privacy-Preserving Mixture-of-Experts Unification
  2. 2025PC-MoE: Memory-Efficient and Privacy-Preserving Collaborative Training for Mixture-of-Experts LLMs
  3. 2024Beyond Parameter Count: Implicit Bias in Soft Mixture of Experts
  4. 2026Policy-Masked Private Experts: Auditable and Reversible Capability Access Control in Sparse MoE Models
  5. 2026Preserving Long-Tailed Expert Information in Mixture-of-Experts Tuning
  6. 2026φ-Balancing for Mixture-of-Experts Training
  7. 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
  8. 2026Towards a Statistical Understanding of Mixture-of-Experts