Sparse Mixture-of-Experts models face a routing bottleneck when offloading inactive experts to host memory, as transfers can only begin after top-K routing completes. SpecPrefetch introduces a lightweight shared adapter to asynchronously predict next-layer expert candidates, decoupling prediction from execution. This approach allows expert data to be fetched in parallel with routing, mitigating the serialization delay inherent in current offloading strategies.
- Decouples expert loading from routing to eliminate serialization bottlenecks.
- Uses a parameter-efficient shared adapter for asynchronous prefetching.
- Enables faster inference on memory-constrained accelerators.
- Reduces latency by overlapping data transfer with computation.