English
Related papers

Related papers: FlexMoE: Scaling Large-scale Sparse Pre-trained Mo…

200 papers

Sparsely-gated Mixture of Experts networks (MoEs) have demonstrated excellent scalability in Natural Language Processing. In Computer Vision, however, almost all performant networks are "dense", that is, every input is processed by every…

Computer Vision and Pattern Recognition · Computer Science 2021-06-14 Carlos Riquelme , Joan Puigcerver , Basil Mustafa , Maxim Neumann , Rodolphe Jenatton , André Susano Pinto , Daniel Keysers , Neil Houlsby

As deep learning models become increasingly large, they pose significant challenges in heterogeneous devices environments. The size of deep learning models makes it difficult to deploy them on low-power or resource-constrained devices,…

Machine Learning · Computer Science 2023-11-27 Mert Unsal , Ali Maatouk , Antonio De Domenico , Nicola Piovesan , Fadhel Ayed

Training large-scale Mixture-of-Experts (MoE) models typically requires high-memory, high-bandwidth GPUs (e.g., A100), and their high cost has become a major barrier to large-model training. In contrast, affordable hardware is low-cost but…

Machine Learning · Computer Science 2026-01-13 Xin Ye , Daning Cheng , Boyang Zhang , Yunquan Zhang

Transformer models can face practical limitations due to their high computational requirements. At the same time, such models exhibit significant activation sparsity, which can be leveraged to reduce the inference cost by converting parts…

Machine Learning · Computer Science 2024-11-13 Filip Szatkowski , Bartosz Wójcik , Mikołaj Piórczyński , Simone Scardapane

The sparse Mixture of Experts(MoE) architecture has evolved as a powerful approach for scaling deep learning models to more parameters with comparable computation cost. As an important branch of large language model(LLM), MoE model only…

Machine Learning · Computer Science 2026-02-10 Dong Pan , Bingtao Li , Yongsheng Zheng , Jiren Ma , Victor Fei

Mixture-of-Experts (MoE) based Large Language Models (LLMs) have achieved superior performance, yet the massive memory overhead caused by storing multiple expert networks severely hinders their practical deployment. Singular Value…

Machine Learning · Computer Science 2026-02-13 Zhendong Mi , Yixiao Chen , Pu Zhao , Xiaodong Yu , Hao Wang , Yanzhi Wang , Shaoyi Huang

We propose Tensor-Trained Low-Rank Adaptation Mixture of Experts (TT-LoRA MoE), a novel computational framework integrating Parameter-Efficient Fine-Tuning (PEFT) with sparse MoE routing to address scalability challenges in large model…

Machine Learning · Computer Science 2026-01-27 Pradip Kunwar , Minh N. Vu , Maanak Gupta , Mahmoud Abdelsalam , Manish Bhattarai

The integration of Federated Learning (FL) and Mixture-of-Experts (MoE) presents a compelling pathway for training more powerful, large-scale artificial intelligence models (LAMs) on decentralized data while preserving privacy. However,…

Machine Learning · Computer Science 2025-07-09 Xiaobing Chen , Boyang Zhang , Xiangwei Zhou , Mingxuan Sun , Shuai Zhang , Songyang Zhang , Geoffrey Ye Li

Sparse mixture of expert architectures (MoEs) scale model capacity without significant increases in training or inference costs. Despite their success, MoEs suffer from a number of issues: training instability, token dropping, inability to…

Machine Learning · Computer Science 2024-05-28 Joan Puigcerver , Carlos Riquelme , Basil Mustafa , Neil Houlsby

Mixture-of-Experts (MoE) models enable scalable performance by activating large parameter sets sparsely, minimizing computational overhead. To mitigate the prohibitive cost of training MoEs from scratch, recent work employs upcycling,…

Machine Learning · Computer Science 2025-11-13 Qi Wang , Hanyang Peng , Yue Yu

Efficiency, specialization, and adaptability to new data distributions are qualities that are hard to combine in current Large Language Models. The Mixture of Experts (MoE) architecture has been the focus of significant research because its…

Computation and Language · Computer Science 2024-08-29 Nikolas Gritsch , Qizhen Zhang , Acyr Locatelli , Sara Hooker , Ahmet Üstün

Diffusion Transformer (DiT) has demonstrated remarkable performance in text-to-image generation; however, its large parameter size results in substantial inference overhead. Existing parameter compression methods primarily focus on pruning,…

Computer Vision and Pattern Recognition · Computer Science 2025-10-13 Youwei Zheng , Yuxi Ren , Xin Xia , Xuefeng Xiao , Xiaohua Xie

Sparsely activated Mixture-of-Experts (MoE) models are widely adopted to scale up model capacity without increasing the computation budget. However, vanilla TopK routers are trained in a discontinuous, non-differentiable way, limiting their…

Machine Learning · Computer Science 2025-02-28 Ziteng Wang , Jun Zhu , Jianfei Chen

Sparse Mixtures of Experts (MoEs) are typically trained to operate at a fixed sparsity level, e.g. $k$ in a top-$k$ gating function. This global sparsity level determines an operating point on the accuracy/latency curve; currently, meeting…

Despite the computational efficiency of MoE models, the excessive memory footprint and I/O overhead inherent in multi-expert architectures pose formidable challenges for real-time inference on resource-constrained edge platforms. While…

Machine Learning · Computer Science 2026-03-20 Yuegui Huang , Zhiyuan Fang , Weiqi Luo , Ruoyu Wu , Wuhui Chen , Zibin Zheng

Mixture of Experts (MoE) models have emerged as a primary solution for reducing the computational cost of Large Language Models. In this work, we analyze their scaling properties, incorporating an expanded range of variables. Specifically,…

Large multimodal Mixture-of-Experts (MoEs) effectively scale the model size to boost performance while maintaining fixed active parameters. However, previous works primarily utilized full-precision experts during sparse up-cycling. Despite…

Computer Vision and Pattern Recognition · Computer Science 2026-01-08 Hongyu Wang , Jiayu Xu , Ruiping Wang , Yan Feng , Yitao Zhai , Peng Pei , Xunliang Cai , Xilin Chen

Mixture of Experts (MoE) models with conditional execution of sparsely activated layers have enabled training models with a much larger number of parameters. As a result, these models have achieved significantly better quality on various…

Computation and Language · Computer Science 2022-11-21 Young Jin Kim , Rawn Henry , Raffy Fahim , Hany Hassan Awadalla

Mixture-of-Experts (MoE) activates only a subset of experts during inference, allowing the model to maintain low inference FLOPs and latency even as the parameter count scales up. However, since MoE dynamically selects the experts, all the…

Machine Learning · Computer Science 2025-05-27 Shibo Jie , Yehui Tang , Kai Han , Yitong Li , Duyu Tang , Zhi-Hong Deng , Yunhe Wang

The Mixture-of-Experts (MoE) technique can scale up the model size of Transformers with an affordable computational overhead. We point out that existing learning-to-route MoE methods suffer from the routing fluctuation issue, i.e., the…

Machine Learning · Computer Science 2022-04-19 Damai Dai , Li Dong , Shuming Ma , Bo Zheng , Zhifang Sui , Baobao Chang , Furu Wei