RAPTOR: Role-Aware Private Training for Mixture-of-Experts
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
- 2024Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Experts28 Aug
- 2024Differentially Private Training of Mixture of Experts Models11 Feb
- 2022Mixture-of-Experts with Expert Choice Routing18 Feb
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 24
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2016Deep Learning with Differential Privacycited by 2,728
- 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
- 2021Membership Inference Attacks From First Principlescited by 878
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
- 2019Adversarial NLI: A New Benchmark for Natural Language Understandingcited by 807
12 more
- 2021Scaling Vision with Sparse Mixture of Expertscited by 747
- 2024DeepSeek-VL2: Mixture-of-Experts Vision-Language Models for Advanced Multimodal Understandingcited by 572
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2021Differentially Private Fine-tuning of Language Modelscited by 400
- 2021Opacus: User-Friendly Differential Privacy Library in PyTorchcited by 383
- 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 320
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2021Hash Layers For Large Sparse Modelscited by 237
- 2019Differentially Private Learning with Adaptive Clippingcited by 220
- 2021Numerical Composition of Differential Privacycited by 180
- 2016Max-Information, Differential Privacy, and Post-Selection Hypothesis Testingcited by 27
- 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
- 2026MetaMoE: Diversity-Aware Proxy Selection for Privacy-Preserving Mixture-of-Experts Unification
- 2025PC-MoE: Memory-Efficient and Privacy-Preserving Collaborative Training for Mixture-of-Experts LLMs
- 2024Beyond Parameter Count: Implicit Bias in Soft Mixture of Experts
- 2026Policy-Masked Private Experts: Auditable and Reversible Capability Access Control in Sparse MoE Models
- 2026Preserving Long-Tailed Expert Information in Mixture-of-Experts Tuning
- 2026φ-Balancing for Mixture-of-Experts Training
- 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
- 2026Towards a Statistical Understanding of Mixture-of-Experts