English
Related papers

Related papers: Swimba: Switch Mamba Model Scales State Space Mode…

200 papers

Mamba-based models have recently demonstrated significant potential in hyperspectral image (HSI) classification, primarily due to their ability to perform contextual modeling with linear computational complexity. However, existing…

Computer Vision and Pattern Recognition · Computer Science 2025-10-07 Yichu Xu , Di Wang , Hongzan Jiao , Lefei Zhang , Liangpei Zhang

Foundation models, now powering most of the exciting applications in deep learning, are almost universally based on the Transformer architecture and its core attention module. Many subquadratic-time architectures such as linear attention,…

Machine Learning · Computer Science 2024-06-03 Albert Gu , Tri Dao

The goal of style transfer is, given a content image and a style source, generating a new image preserving the content but with the artistic representation of the style source. Most of the state-of-the-art architectures use transformers or…

Computer Vision and Pattern Recognition · Computer Science 2025-04-15 Filippo Botti , Alex Ergasti , Leonardo Rossi , Tomaso Fontanini , Claudio Ferrari , Massimo Bertozzi , Andrea Prati

Mixture of expert (MoE) models are a promising approach to increasing model capacity without increasing inference cost, and are core components of many state-of-the-art language models. However, current MoE models typically use only few…

Machine Learning · Computer Science 2025-07-31 Ryo Bertolissi , Jonas Hübotter , Ido Hakimi , Andreas Krause

The essence of multi-modal fusion lies in exploiting the complementary information inherent in diverse modalities. However, prevalent fusion methods rely on traditional neural architectures and are inadequately equipped to capture the…

Artificial Intelligence · Computer Science 2025-06-19 Wenbing Li , Hang Zhou , Junqing Yu , Zikai Song , Wei Yang

Selective state space models (SSMs) have rapidly become a compelling backbone for large language models, especially for long-context workloads. Yet in deployment, their inference performance is often bounded by the memory capacity,…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-25 Anurag Dutt , Nimit Shah , Hazem Masarani , Anshul Gandhi

Scale has opened new frontiers in natural language processing -- but at a high cost. In response, Mixture-of-Experts (MoE) and Switch Transformers have been proposed as an energy efficient path to even larger and more capable language…

Computation and Language · Computer Science 2022-05-03 Barret Zoph , Irwan Bello , Sameer Kumar , Nan Du , Yanping Huang , Jeff Dean , Noam Shazeer , William Fedus

Large pre-trained models have achieved outstanding results in sequence modeling. The Transformer block and its attention mechanism have been the main drivers of the success of these models. Recently, alternative architectures, such as…

Machine Learning · Computer Science 2025-01-29 J. Pablo Muñoz , Jinjie Yuan , Nilesh Jain

The sparsely activated mixture-of-experts (MoE) transformer has become a common architecture for large language models (LLMs) due to its sparsity, which requires fewer computational demands while easily scaling the model size. In MoE…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-14 Wenxiang Lin , Xinglin Pan , Lin Zhang , Shaohuai Shi , Xuan Wang , Xiaowen Chu

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

Mixture-of-Experts (MoE) is an emerging technique for scaling large models with sparse activation. MoE models are typically trained in a distributed manner with an expert parallelism scheme, where experts in each MoE layer are distributed…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-11-26 Fahao Chen , Peng Li , Zicong Hong , Zhou Su , Song Guo

Spiking Transformers have shown strong potential for long-range visual modeling through spike-driven self-attention. However, their quadratic token interactions remain fundamentally misaligned with the sparse and event-driven nature of…

Computer Vision and Pattern Recognition · Computer Science 2026-04-29 Dewei Bai , Hongxiang Peng , Yunyun Zeng , Ziyu Zhang , Hong Qu , Yi Zhang

Mixture-of-Experts (MoE) architectures enhance the efficiency of large language models by activating only a subset of experts per token. However, standard MoE employs a fixed Top-K routing strategy, leading to redundant computation and…

Artificial Intelligence · Computer Science 2026-05-15 Juntong Wu , Jialiang Cheng , Qishen Yin , Yue Dai , Yuliang Yan , Fuyu Lv , Ou Dan , Li Yuan

Mixture-of-Experts (MoE) has become the dominant architecture for scaling large language models: frontier models routinely decouple total parameters from per-token computation through sparse expert routing. Scaling laws show that under…

Machine Learning · Computer Science 2026-05-12 Chaitanya Dwivedi , Binxuan Huang , Himanshu Gupta , Pratik Jayarao , Neeraj Varshney , Bing Yin

The Mixtures-of-Experts (MoE) model is a widespread distributed and integrated learning method for large language models (LLM), which is favored due to its ability to sparsify and expand models efficiently. However, the performance of MoE…

Machine Learning · Computer Science 2024-05-24 Jing Li , Zhijie Sun , Xuan He , Li Zeng , Yi Lin , Entong Li , Binfan Zheng , Rongqian Zhao , Xin Chen

As deep learning models expand, the pre-training-fine-tuning paradigm has become the standard approach for handling various downstream tasks. However, shared parameters can lead to diminished performance when dealing with complex datasets…

Machine Learning · Computer Science 2025-05-13 Junzhou Xu , Boyu Diao

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) has become a dominant architecture for scaling large language models (LLMs). However, the execution characteristics of MoE inference are changing rapidly and increasingly mismatch the assumptions underlying existing…

Hardware Architecture · Computer Science 2026-05-13 Jungwoo Kim , Rubens Lacouture , Genghan Zhang , Gina Sohn , Qizheng Zhang , Swapnil Gandhi , Christos Kozyrakis , Kunle Olukotun

We focus on multi-domain Neural Machine Translation, with the goal of developing efficient models which can handle data from various domains seen during training and are robust to domains unseen during training. We hypothesize that Sparse…

Computation and Language · Computer Science 2024-07-02 Nadezhda Chirkova , Vassilina Nikoulina , Jean-Luc Meunier , Alexandre Bérard

Mixture-of-Experts (MoE) has emerged as a practical approach to scale up parameters for the Transformer model to achieve better generalization while maintaining a sub-linear increase in computation overhead. Current MoE models are mainly…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-03 Shuqing Luo , Jie Peng , Pingzhi Li , Hanrui Wang , Tianlong Chen