GRIP: Algorithm-Agnostic Machine Unlearning for Mixture-of-Experts via Geometric Router Constraints
On the map: Mixture-of-experts → Expert routing strategies
also in 1 more
Abstract (arXiv)
Machine unlearning in Mixture-of-Experts (MoE) large language models presents a critical yet under-explored challenge. Current unlearning methods applied to MoE architectures often exploit dynamic routing as an optimization shortcut: rather than genuinely erasing knowledge from expert parameters, they manipulate routers to redirect queries away from the originally assigned experts. This not only causes severe utility degradation but also leaves hazardous knowledge intact. Consequently, adversaries can bypass the router to recover sensitive information directly from dormant experts. In this study, we propose Geometric Routing Invariance Preservation (GRIP), an algorithm-agnostic framework that resolves these failure modes by enforcing hard geometric constraints on router updates. By projecting router gradient updates into the null space of the retain set's routing matrix, GRIP suppresses routing manipulation without freezing the router entirely, thereby directing the unlearning pressure into the expert parameters themselves across all relevant experts. GRIP offers two complementary variants: training-time stochastic projection and a post-training closed-form analytical correction. Extensive experiments on two MoE models across hazardous knowledge removal and copyright unlearning benchmarks demonstrate that GRIP restores routing stability from 0.21 to >0.94, improves retain accuracy by up to 89%, and reduces white-box adversarial knowledge recovery from 11% to just 3% while in line with dense-architecture unlearning under black-box prompt attack, establishing geometric constraints as a principled solution for genuine unlearning in sparse MoE architectures.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 39 of its 48 references that are in the map. First posted on arXiv 23 Jan 2026.
Builds on
39 of 48 references.
Closest ancestors · 3
Same direction, newest first
- 2024Stealing User Prompts from Mixture of Experts30 Oct
- 2022Mixture-of-Experts with Expert Choice Routing18 Feb
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 36
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2024Mixtral of Expertscited by 1,785
- 2023Fine-tuning Aligned Language Models Compromises Safety, Even When Users Do Not Intend To!cited by 1,106
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2019Machine Unlearningcited by 1,020
- 2021Ethical and social risks of harm from Language Modelscited by 957
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
- 2021Fast Model Editing at Scalecited by 491
24 more
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearningcited by 484
- 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Modelscited by 424
- 2023Editing Large Language Models: Problems, Methods, and Opportunitiescited by 367
- 2022A Survey of Machine Unlearningcited by 298
- 2023Towards Unbounded Machine Unlearningcited by 261
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Modelscited by 251
- 2023Pretraining Language Models with Human Preferencescited by 245
- 2023From Sparse to Soft Mixtures of Expertscited by 245
- 2023In-Context Unlearning: Language Models as Few Shot Unlearnerscited by 214
- 2023Unlearn What You Want to Forget: Efficient Unlearning for LLMscited by 211
- 2021Counterfactual Memorization in Neural Language Modelscited by 155
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearningcited by 104
- 2023DEPN: Detecting and Editing Privacy Neurons in Pretrained Language Modelscited by 75
- 2023Scaling Expert Language Models with Unsupervised Domain Discoverycited by 59
- 2025Unlearning Isn't Deletion: Investigating Reversibility of Machine Unlearning in LLMscited by 41
- 2024SEUF: Is Unlearning One Expert Enough for Mixture-of-Experts LLMs?cited by 14
- 2024Underestimated Privacy Risks for Minority Populations in Large Language Model Unlearningcited by 9
- 2016A Sampling Kaczmarz-Motzkin Algorithm for Linear Feasibilitycited by 7
- 2025I Know What You Said: Unveiling Hardware Cache Side-Channels in Local Large Language Model Inferencecited by 6
- 2025Prompt Attacks Reveal Superficial Knowledge Removal in Unlearning Methodscited by 5
- 2025PII Jailbreaking in LLMs via Activation Steering Reveals Personal Information Leakagecited by 5
- 2025Erase or Hide? Suppressing Spurious Unlearning Neurons for Robust Unlearningcited by 5
- 2025CryptoMoE: Privacy-Preserving and Scalable Mixture of Experts Inference via Balanced Expert Routingcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Routing-Aware Expert Calibration for Machine Unlearning in Mixture-of-Experts Language Models
- 2026MoRA: MoE Pruning via Router Bias Learning and Expert Approximation
- 2026Share First, Route What Remains: A Unified Framework for Token-Adaptive MoE Computation
- 2026Routing by Analogy: kNN-Augmented Expert Assignment for Mixture-of-Experts
- 2026RouteHijack: Routing-Aware Attack on Mixture-of-Experts LLMs
- 2026When Load-Balancing Goes Too Far: Expert Pruning in Over-Dispersed Mixture-of-Experts Models
- 2026Hierarchical Mixture-of-Experts with Two-Stage Optimization
- 2026When Are Experts Misrouted? Counterfactual Routing Analysis in Mixture-of-Experts Language Models