ARTFEED — Contemporary Art Intelligence

Router Sensitivity Under Lightweight Fine-Tuning Identifies Prunable Experts in MoE Models

ai-technology · 2026-08-11

A recent paper on arXiv (2608.07890) explores the potential of lightweight fine-tuning to pinpoint prunable experts within Mixture-of-Experts (MoE) models. These models, which are expansive language frameworks, activate only a fraction of their parameters for each token while necessitating the storage of all experts. The researchers introduce a technique that involves a brief fine-tuning phase using a parameter-efficient adapter, ranking experts based on the changes in the L2 router, and pruning the least altered experts in a single step. On Mixtral-8x7B-Instruct (44.83% MMLU-Pro), router-only LoRA, which utilizes just 0.002% of parameters, surpasses all-module LoRA with half the experts removed (27.54% versus 24.42%). As adaptation extends to attention and expert weights, signal quality diminishes. Accuracy consistently increases with LoRA rank, peaking at 28.76%. IA3, which keeps router weights static, achieves results comparable to direct router adaptation, while unconstrained additive adaptation negatively impacts the signal. The findings indicate that router sensitivity during lightweight fine-tuning can effectively reveal prunable experts, presenting a viable method to lessen storage demands in MoE models.

Key facts

  • Paper arXiv:2608.07890
  • MoE models decouple total parameters from per-token compute
  • Pruning experts with smallest router-norm changes during fine-tuning preserves accuracy
  • Lightweight adaptation can recover this signal
  • Router-only LoRA trains 0.002% of parameters
  • Outperforms all-module LoRA at matched rank with half experts removed (27.54% vs. 24.42%)
  • Accuracy improves monotonically with LoRA rank, reaching 28.76%
  • IA3 matches direct router adaptation

Entities

Institutions

  • arXiv

Sources