English
Related papers

Related papers: DSelect-k: Differentiable Selection in the Mixture…

200 papers

Mixture of Experts (MoE), an ensemble of specialized models equipped with a router that dynamically distributes each input to appropriate experts, has achieved successful results in the field of machine learning. However, theoretical…

Machine Learning · Computer Science 2025-08-19 Ryotaro Kawata , Kohsei Matsutani , Yuri Kinoshita , Naoki Nishikawa , Taiji Suzuki

Sparse Mixture-of-Experts (MoE) architectures effectively scale model capacity by activating only a subset of experts for each input token. However, the standard Top-k routing strategy imposes a uniform sparsity pattern that ignores the…

Artificial Intelligence · Computer Science 2025-12-17 Can Jin , Hongwu Peng , Mingcan Xiang , Qixin Zhang , Xiangchi Yuan , Amit Hasan , Ohiremen Dibua , Yifan Gong , Yan Kang , Dimitris N. Metaxas

Pre-training has proven effective in addressing data scarcity and performance limitations in solving PDE problems with neural operators. However, challenges remain due to the heterogeneity of PDE datasets in equation types, which leads to…

Machine Learning · Computer Science 2025-11-03 Hong Wang , Haiyang Xin , Jie Wang , Xuanze Yang , Fei Zha , Huanshuo Dong , Yan Jiang

Predicting multiple heterogeneous biological and medical targets is a challenge for traditional deep learning models. In contrast to single-task learning, in which a separate model is trained for each target, multi-task learning (MTL)…

Machine Learning · Computer Science 2022-05-31 Raquel Aoki , Frederick Tung , Gabriel L. Oliveira

Mixture-of-Experts (MoE) models improve transformer efficiency but lack a unified theoretical explanation, especially when both feed-forward and attention layers are allowed to specialize. To this end, we study the Mixture-of-Transformers…

Machine Learning · Computer Science 2025-11-03 Hongbo Li , Qinhang Wu , Sen Lin , Yingbin Liang , Ness B. Shroff

Modeling feature interactions is essential for accurate click-through rate (CTR) prediction in advertising systems. Recent studies have adopted the Mixture-of-Experts (MoE) approach to improve performance by ensembling multiple feature…

Information Retrieval · Computer Science 2025-09-16 Jiancheng Wang , Mingjia Yin , Hao Wang , Enhong Chen

Sparsely gated Mixture-of-Expert (MoE) has demonstrated its effectiveness in scaling up deep neural networks to an extreme scale. Despite that numerous efforts have been made to improve the performance of MoE from the model design or system…

Machine Learning · Computer Science 2023-02-21 Chang Chen , Min Li , Zhihua Wu , Dianhai Yu , Chao Yang

Recent progress in deep learning has been driven by increasingly large-scale models, but the resulting computational cost has become a critical bottleneck. Sparse Mixture of Experts (MoE) offers an effective solution by activating only a…

Computer Vision and Pattern Recognition · Computer Science 2026-04-24 Masahiro Kada , Ryota Yoshihashi , Satoshi Ikehata , Rei Kawakami , Ikuro Sato

Mixture-of-Experts (MoE) models mostly use a router to assign tokens to specific expert modules, activating only partial parameters and often outperforming dense models. We argue that the separation between the router's decision-making and…

Computation and Language · Computer Science 2025-06-02 Ang Lv , Ruobing Xie , Yining Qian , Songhao Wu , Xingwu Sun , Zhanhui Kang , Di Wang , Rui Yan

Real-world model deployment across multiple domains requires multimodal models to operate under two complementary regimes: (1) multi-task pretraining, tasks are co-available at design time where related tasks could borrow representational…

Machine Learning · Computer Science 2026-05-12 Xing Han , Shravan Chaudhari , Tanvi Ranade , Rama Chellappa , Suchi Saria

The Sparse Mixture of Experts (SMoE) has been widely employed to enhance the efficiency of training and inference for Transformer-based foundational models, yielding promising results.However, the performance of SMoE heavily depends on the…

Machine Learning · Computer Science 2025-03-11 Yongxin Guo , Zhenglin Cheng , Xiaoying Tang , Zhaopeng Tu , Tao Lin

The classification of stellar light curves has become a key task in modern time-domain astronomy, fueled by the rapid growth of data from large-scale surveys such as Kepler and TESS. Although deep learning models have achieved high accuracy…

Instrumentation and Methods for Astrophysics · Physics 2025-07-21 Cunshi Wang , Yu Bai , Xinrui Song , Jiacheng Xu , Henggeng Han , Yuyang Li , Xinjie Hu , Huiqin Yang , Jifeng Liu

Mixture-of-Experts (MoE) architectures decompose prediction tasks into specialized expert sub-networks selected by a gating mechanism. This letter adopts a communication-theoretic view of MoE gating, modeling the gate as a stochastic…

Machine Learning · Statistics 2026-03-27 Ali Khalesi , Mohammad Reza Deylam Salehi

Sparsely activated Mixture-of-Experts (MoE) models effectively increase the number of parameters while maintaining consistent computational costs per token. However, vanilla MoE models often suffer from limited diversity and specialization…

Machine Learning · Computer Science 2025-07-11 Lei Kang , Jia Li , Mi Tian , Hua Huang

In this work, we first explore whether the parameters activated by the MoE layer remain highly sparse at inference. We perform a sparsification study on several representative MoE models. For each expert, we rank parameters by the magnitude…

Computation and Language · Computer Science 2025-10-08 Runxi Cheng , Yuchen Guan , Yucheng Ding , Qingguo Hu , Yongxian Wei , Chun Yuan , Yelong Shen , Weizhu Chen , Yeyun Gong

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

Estimating accurate and well-calibrated predictive uncertainty is important for enhancing the reliability of computer vision models, especially in safety-critical applications like traffic scene perception. While ensemble methods are…

Computer Vision and Pattern Recognition · Computer Science 2025-09-08 Svetlana Pavlitska , Beyza Keskin , Alwin Faßbender , Christian Hubschneider , J. Marius Zöllner

Mixture of Experts (MoE) models constitute a widely utilized class of ensemble learning approaches in statistics and machine learning, known for their flexibility and computational efficiency. They have become integral components in…

Machine Learning · Statistics 2025-05-26 Tuan Thai , TrungTin Nguyen , Dat Do , Nhat Ho , Christopher Drovandi

Mixture-of-Experts (MoE) models have shown remarkable capability in instruction tuning, especially when the number of tasks scales. However, previous methods simply merge all training tasks (e.g. creative writing, coding, and mathematics)…

Computation and Language · Computer Science 2024-06-18 Tong Zhu , Daize Dong , Xiaoye Qu , Jiacheng Ruan , Wenliang Chen , Yu Cheng

Sparse Mixture-of-Experts (MoE) is a neural architecture design that can be utilized to add learnable parameters to Large Language Models (LLMs) without increasing inference cost. Instruction tuning is a technique for training LLMs to…