Search papers, labs, and topics across Lattice.
ReMoE is introduced as a router fine-tuning framework to improve expert reuse in memory-constrained MoE LLM inference by biasing the router towards recently selected experts. This approach enhances temporal routing stability, aligning expert selection with cache locality and reducing the need to fetch experts from slower external storage. Experiments on DeepSeek and Qwen models demonstrate that ReMoE improves expert reuse by 26%, leading to significant throughput improvements (8.4% with vLLM) and decode speedups (1.77-1.99x with llama.cpp) without compromising downstream task performance.
Reduce your MoE LLM's memory bottleneck by up to 49% with a simple router fine-tuning trick that encourages experts to be reused.
Fine-grained Mixture-of-Experts (MoE) models sparsely activate only a subset of experts per token, reducing activated computation while maintaining high model capacity. However, in memory-constrained inference scenarios, only a small set of experts can be cached. Experts not in the cache must be fetched from slow external storage (e.g., UFS), leading to frequent evictions and substantial I/O overhead. We propose ReMoE, a router fine-tuning framework designed to boost token-wise expert reuse. ReMoE biases the router toward recently selected experts, producing temporally stable routing that better matches cache locality constraints. By increasing short-horizon expert reuse, ReMoE reduces expert fetches from storage without adding inference-time computation. Experiments on DeepSeek and Qwen models show that ReMoE improves expert reuse by 26% while maintaining downstream task performance. Real-system evaluations further confirm these benefits, improving output throughput by 8.4% under vLLM GPU-CPU expert offloading and reducing TPOT by 43.6-49.8% under llama.cpp on Jetson Orin NX, corresponding to a 1.77-1.99$\times$ decode speedup across diverse workloads. Checkpoints and usage instructions are available at https://github.com/BUAA-OSCAR/ReMoE.
