Abstract:While modern internet services, such as chatbots, search engines, and online advertising, demand the use of large-scale deep neural networks (DNNs), distributed training and inference over heterogeneous computing systems are desired to facilitate these DNN models. Mixture-of-Experts (MoE) is one the most common strategies to lower the cost of training subject to the overall size of models/data through gating and parallelism in a divide-and-conquer fashion. While DeepSpeed has made efforts in carrying out large-scale MoE training over heterogeneous infrastructures, the efficiency of training and inference could be further improved from several system aspects, including load balancing, communication/computation efficiency, and memory footprint limits. In this work, we present a novel MoESys that boosts efficiency in both large-scale training and inference. Specifically, in the training procedure, the proposed MoESys adopts an Elastic MoE training strategy with 2D prefetch and Fusion communication over Hierarchical storage, so as to enjoy efficient parallelisms. For scalable inference in a single node, especially when the model size is larger than GPU memory, MoESys builds the CPU-GPU memory jointly into a ring of sections to load the model, and executes the computation tasks across the memory sections in a round-robin manner for efficient inference. We carried out extensive experiments to evaluate MoESys, where MoESys successfully trains a Unified Feature Optimization (UFO) model with a Sparsely-Gated Mixture-of-Experts model of 12B parameters in 8 days on 48 A100 GPU cards. The comparison against the state-of-the-art shows that MoESys outperformed DeepSpeed with 33% higher throughput (tokens per second) in training and 13% higher throughput in inference in general. Particularly, under unbalanced MoE Tasks, e.g., UFO, MoESys achieved 64% higher throughput with 18% lower memory footprints.

Interpretable Mixture of Experts for Decomposition Network on Server Performance Metrics Forecasting

KAN4TSF: Are KAN and KAN-based models Effective for Time Series Forecasting?

Kolmogorov-Arnold Networks for Time Series: Bridging Predictive Power and Interpretability

Towards MoE Deployment: Mitigating Inefficiencies in Mixture-of-Expert (moe) Inference

A Gated Residual Kolmogorov-Arnold Networks for Mixtures of Experts

A Novel Interpretable Short-Term Load Forecasting Method Based on Kolmogorov-Arnold Networks

CA-MoE: Channel-Adapted MoE for Incremental Weather Forecasting

A multi-energy load forecasting method based on complementary ensemble empirical model decomposition and composite evaluation factor reconstruction

Prediction Is All MoE Needs: Expert Load Distribution Goes from Fluctuating to Stabilizing

KAE-Informer: A Knowledge Auto-Embedding Informer for Forecasting Long-Term Workloads of Microservices

MoE-Infinity: Offloading-Efficient MoE Model Serving

MoESys: A Distributed and Efficient Mixture-of-Experts Training and Inference System for Internet Services

MoDE: A Mixture-of-Experts Model with Mutual Distillation among the Experts

M-EDEM: A MNN-based Empirical Decomposition Ensemble Method for Improved Time Series Forecasting

Predictive Modeling of Flexible EHD Pumps using Kolmogorov-Arnold Networks

MoNDE: Mixture of Near-Data Experts for Large-Scale Sparse Models

AdapMoE: Adaptive Sensitivity-based Expert Gating and Management for Efficient MoE Inference

Mixture of Online and Offline Experts for Non-stationary Time Series

EWMoE: An effective model for global weather forecasting with mixture-of-experts

Hybrid Model with Secondary Decomposition, Randomforest Algorithm, Clustering Analysis and Long Short Memory Network Principal Computing for Short-Term Wind Power Forecasting on Multiple Scales

Efficient prediction of potential energy surface and physical properties with Kolmogorov-Arnold Networks