English
Related papers

Related papers: Multi-Task Dense Prediction Fine-Tuning with Mixtu…

200 papers

The Mixture of Experts (MoE) is a widely known neural architecture where an ensemble of specialized sub-models optimizes overall performance with a constant computational cost. However, conventional MoEs pose challenges at scale due to the…

Computation and Language · Computer Science 2023-09-12 Ted Zadouri , Ahmet Üstün , Arash Ahmadian , Beyza Ermiş , Acyr Locatelli , Sara Hooker

Sparse Mixture of Experts (MoE) large language models (LLMs) are gradually becoming the mainstream approach for ultra-large-scale models. Existing optimization efforts for MoE models have focused primarily on coarse-grained MoE…

Computation and Language · Computer Science 2025-05-07 Haoqi Yang , Luohe Shi , Qiwei Li , Zuchao Li , Ping Wang , Bo Du , Mengjia Shen , Hai Zhao

In recent years, Mixture-of-Experts (MoE) has emerged as an effective approach for enhancing the capacity of deep neural network (DNN) with sub-linear computational costs. However, storing all experts on GPUs incurs significant memory…

Machine Learning · Computer Science 2025-03-11 Suraiya Tairin , Shohaib Mahmud , Haiying Shen , Anand Iyer

The interpretability of Mixture-of-Experts (MoE) models, especially those with heterogeneous designs, remains underexplored. Existing attribution methods for dense models fail to capture dynamic routing-expert interactions in sparse MoE…

Computation and Language · Computer Science 2025-06-12 Junzhuo Li , Bo Wang , Xiuze Zhou , Peijie Jiang , Jia Liu , Xuming Hu

Large Language Models (LLMs) have gained immense success in revolutionizing various applications, including content generation, search and recommendation, and AI-assisted operation. To reduce high training costs, Mixture-of-Experts (MoE)…

Machine Learning · Computer Science 2025-10-07 Hanfei Yu , Xingqi Cui , Hong Zhang , Hao Wang , Hao Wang

Mixture-of-Experts (MoE) is a promising way to scale up the learning capacity of large language models. It increases the number of parameters while keeping FLOPs nearly constant during inference through sparse activation. Yet, it still…

Machine Learning · Computer Science 2025-02-26 Pingzhi Li , Xiaolong Jin , Zhen Tan , Yu Cheng , Tianlong Chen

Large language models (LLMs) increasingly adopt Mixture-of-Experts (MoE) architectures to scale model capacity while reducing computation. Fine-tuning these MoE-based LLMs often requires access to distributed and privacy-sensitive data,…

Machine Learning · Computer Science 2026-03-24 Zihan Fang , Qianru Wang , Haonan An , Zheng Lin , Yiqin Deng , Xianhao Chen , Yuguang Fang

The Mixture-of-Experts (MoE) technique has proven to be a promising solution to efficiently scale the model size, which has been widely applied in recent LLM advancements. However, the substantial memory overhead of MoE models has made…

Machine Learning · Computer Science 2025-10-17 Ruijie Miao , Yilun Yao , Zihan Wang , Zhiming Wang , Bairen Yi , LingJun Liu , Yikai Zhao , Tong Yang

Merging various task-specific Transformer-based models trained on different tasks into a single unified model can execute all the tasks concurrently. Previous methods, exemplified by task arithmetic, have been proven to be both effective…

Machine Learning · Computer Science 2024-06-10 Anke Tang , Li Shen , Yong Luo , Nan Yin , Lefei Zhang , Dacheng Tao

Large language models (LLMs) encounter significant adaptation challenges in diverse multitask finetuning. Mixture-of-experts (MoE) provides a promising solution with a dynamic architecture, enabling effective task decoupling. However,…

Machine Learning · Computer Science 2025-05-28 Rongyu Zhang , Yijiang Liu , Huanrui Yang , Shenli Zheng , Dan Wang , Yuan Du , Li Du , Shanghang Zhang

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) selects a few feed-forward networks (FFNs) per token, achieving an effective trade-off between computational cost and performance. In conventional MoE, each expert is treated as entirely independent, and experts…

Machine Learning · Computer Science 2026-01-27 Shota Takashiro , Takeshi Kojima , Shohei Taniguchi , Yusuke Iwasawa , Yutaka Matsuo

Optimizing various wireless user tasks poses a significant challenge for networking systems because of the expanding range of user requirements. Despite advancements in Deep Reinforcement Learning (DRL), the need for customized optimization…

Networking and Internet Architecture · Computer Science 2024-02-16 Hongyang Du , Guangyuan Liu , Yijing Lin , Dusit Niyato , Jiawen Kang , Zehui Xiong , Dong In Kim

The emergence of large-scale Mixture of Experts (MoE) models represents a significant advancement in artificial intelligence, offering enhanced model capacity and computational efficiency through conditional computation. However, deploying…

Machine Learning · Computer Science 2025-01-23 Jiacheng Liu , Peng Tang , Wenfeng Wang , Yuhang Ren , Xiaofeng Hou , Pheng-Ann Heng , Minyi Guo , Chao Li

As machine learning models in critical fields increasingly grapple with multimodal data, they face the dual challenges of handling a wide array of modalities, often incomplete due to missing elements, and the temporal irregularity and…

Machine Learning · Computer Science 2025-04-10 Xing Han , Huy Nguyen , Carl Harris , Nhat Ho , Suchi Saria

Mixture-of-Experts (MoE) models have emerged as a dominant paradigm for efficient LLM scaling, yet adapting them to non-English downstream tasks remains challenging. Existing fine-tuning approaches treat MoE models as monolithic learners,…

Computation and Language · Computer Science 2026-05-28 Guanzhi Deng , Kuan Wu , Haibo Wang , Shing Yin Wong , Sichun Luo , Linqi Song

Despite recent advancements in offline multi-task reinforcement learning (MTRL) have harnessed the powerful capabilities of the Transformer architecture, most approaches focus on a limited number of tasks, with scaling to extremely massive…

Machine Learning · Computer Science 2025-06-02 Yilun Kong , Guozheng Ma , Qi Zhao , Haoyu Wang , Li Shen , Xueqian Wang , Dacheng Tao

The sparsely gated mixture of experts (MoE) architecture sends different inputs to different subnetworks, i.e., experts, through trainable routers. MoE reduces the training computation significantly for large models, but its deployment can…

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

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,…