Janus: Disaggregating Attention and Experts for Scalable MoE Inference
By: Zhexiang Zhang , Ye Wang , Xiangyu Wang and more
Large Mixture-of-Experts (MoE) model inference is challenging due to high resource demands and dynamic workloads. Existing solutions often deploy the entire model as a single monolithic unit, which applies a unified resource configuration to both attention and expert modules despite their different requirements, leading to limited scalability and resource inefficiency. In this paper, we propose Janus, a scalable MoE inference system that disaggregates attention and experts on separate GPU sub-clusters, enabling each module to be managed and scaled independently. Janus incorporates three key designs for efficient, disaggregated MoE inference. First, it proposes an adaptive two-phase communication scheme that exploits intra- and inter-node bandwidth hierarchies for low-latency data exchange. Second, motivated by the memory-bound nature of MoE modules, Janus introduces a lightweight scheduler and implements it as a GPU kernel to balance the number of activated experts across GPUs at minimal overhead, thereby reducing inference latency. Third, Janus performs fine-grained resource management to dynamically adjust expert placement and independently scale attention and MoE resources to improve overall efficiency. Evaluation shows Janus achieves up to 3.9 higher perGPU throughput than state-of-the-art systems while meeting per-token latency requirements.
Similar Papers
MegaScale-Infer: Serving Mixture-of-Experts at Scale with Disaggregated Expert Parallelism
Distributed, Parallel, and Cluster Computing
Makes AI models run faster and cheaper.
HeterMoE: Efficient Training of Mixture-of-Experts Models on Heterogeneous GPUs
Distributed, Parallel, and Cluster Computing
Trains smart computer brains faster on mixed computers.
ExpertFlow: Adaptive Expert Scheduling and Memory Coordination for Efficient MoE Inference
Distributed, Parallel, and Cluster Computing
Makes AI models run faster on less memory.