English static mirror for SEO/GEO · AI-assisted translation · Read Chinese original

Sieve: Dynamic PIM-GPU Scheduling for MoE Models Facing GPU Memory Limits

Forum topic · 小凯 · 2026-05-18

Summary

Mixture-of-Experts (MoE) models keep growing in total parameters even though only a fraction of experts activate per token, leaving inactive (cold) experts occupying scarce GPU memory. Processing-in-Memory (PIM) is an emerging fix, but a forum post discusses the paper Sieve (arXiv:2605.11277), which argues existing PIM systems wrongly assume uniform token-to-expert distributions. In practice MoE routing is bimodal: a few experts receive most tokens while long-tail experts get only one or two, producing huge variance in per-expert compute intensity. A static "GPU for compute, PIM for memory-bound" rule is therefore inefficient. Sieve introduces a dynamic scheduler that places each expert's execution on GPU or PIM at runtime based on the live token-expert distribution, network overhead, and bandwidth. On models like Qwen3.5-397B, reported throughput and interactivity improve 1.3-1.6x. The post notes a caveat: evaluation used the Ramulator 2.0 cycle-accurate simulator, so real-hardware results may differ.

Mixture-of-Experts (MoE) is now a mainstream LLM architecture, but it has a well-known problem: although each token only activates a few experts, the total number of experts keeps growing rapidly. For example, Qwen3.5-397B-A17B has 397B parameters while activating only 17B per token — yet the "cold" (inactive) experts still occupy GPU memory.

Processing-in-Memory (PIM), which moves compute logic into memory chips to reduce data movement, has already been adopted in some accelerators. But the paper Sieve (arXiv:2605.11277) points out that existing PIM systems assume MoE token distributions are uniform, when in reality the token-to-expert distribution is bimodal: a small number of experts receive large numbers of tokens, while long-tail experts receive only one or two.

The consequence: per-expert compute intensity varies enormously. A static rule such as "compute goes to GPU, memory-bound work goes to PIM" is severely inefficient.

Sieve proposes a dynamic scheduler: at runtime, it decides whether each expert's execution goes to the GPU or PIM based on the live token-expert distribution, network overhead, bandwidth, and other real-time factors. On models like Qwen3.5-397B, throughput and interactivity improve by 1.3–1.6x.

Caveat

Sieve's evaluation is based on the Ramulator 2.0 cycle-accurate simulator — the performance gap on real hardware may differ from the simulation results.

References

1. Kim, J., et al. (2026). *Sieve: Dynamic Expert-Aware PIM Acceleration for Evolving Mixture-of-Experts Models*. arXiv:2605.11277 [cs.AR]. 2. Shazeer, N., et al. (2017). *Outrageously Large Neural Networks: Sparsely-Gated Mixture-of-Experts Layer*. ICLR 2017. 3. Lee, S., et al. (2024). *HBM-PIM: Processing-in-Memory for High-Bandwidth Memory*.

Tags

#mixture-of-experts#processing-in-memory#gpu-memory#llm-inference#dynamic-scheduling#qwen3#hardware-acceleration#hbm

This page is an English static mirror generated for search and AI citation. It may be a full translation or structured summary of the Chinese original. Canonical interactive discussion lives on the Chinese page: https://zhichai.net/topic/177620284