English
Related papers

Related papers: Edge-MoE: Memory-Efficient Multi-Task Vision Trans…

200 papers

Multi-task learning (MTL) leverages a shared model to accomplish multiple tasks and facilitate knowledge transfer. Recent research on task arithmetic-based MTL demonstrates that merging the parameters of independently fine-tuned models can…

Machine Learning · Computer Science 2024-10-30 Li Shen , Anke Tang , Enneng Yang , Guibing Guo , Yong Luo , Lefei Zhang , Xiaochun Cao , Bo Du , Dacheng Tao

Learning efficient and expressive visual representation has long been the pursuit of computer vision research. While Vision Transformers (ViTs) gradually replace traditional Convolutional Neural Networks (CNNs) as more scalable vision…

Computer Vision and Pattern Recognition · Computer Science 2026-03-23 Quan Kong , Yanru Xiao , Yuhao Shen , Cong Wang

In this paper, we explore a strategy that uses Mixture-of-Experts (MoE) to streamline, rather than augment, vision transformers. Each expert in an MoE layer is a SwiGLU feedforward network, where V and W2 are shared across the layer. No…

Computer Vision and Pattern Recognition · Computer Science 2024-07-26 Jen Hong Tan

The application of mixture-of-experts (MoE) is gaining popularity due to its ability to improve model's performance. In an MoE structure, the gate layer plays a significant role in distinguishing and routing input features to different…

Machine Learning · Computer Science 2024-02-05 Zhitian Xie , Yinger Zhang , Chenyi Zhuang , Qitao Shi , Zhining Liu , Jinjie Gu , Guannan Zhang

Vision Transformer (ViT) has achieved remarkable performance in computer vision. However, positional encoding in ViT makes it substantially difficult to learn the intrinsic equivariance in data. Initial attempts have been made on designing…

Computer Vision and Pattern Recognition · Computer Science 2023-07-10 Renjun Xu , Kaifan Yang , Ke Liu , Fengxiang He

In large multimodal models (LMMs), the perception of non-language modalities (e.g., visual representations) is usually not on par with the large language models (LLMs)' powerful reasoning capabilities, deterring LMMs' performance on…

Machine Learning · Computer Science 2025-03-04 Zhongyang Li , Ziyue Li , Tianyi Zhou

Vision Transformers (ViTs) have shown remarkable performance and scalability across various computer vision tasks. To apply single-scale ViTs to image segmentation, existing methods adopt a convolutional adapter to generate multi-scale…

Computer Vision and Pattern Recognition · Computer Science 2025-03-26 Tommie Kerssies , Niccolò Cavagnero , Alexander Hermans , Narges Norouzi , Giuseppe Averta , Bastian Leibe , Gijs Dubbelman , Daan de Geus

Mixture of Experts (MoE) LLMs have recently gained attention for their ability to enhance performance by selectively engaging specialized subnetworks or "experts" for each input. However, deploying MoEs on memory-constrained devices remains…

Vision Transformers (ViTs) have achieved significant success in computer vision. However, their intensive computations and massive memory footprint challenge ViTs' deployment on embedded devices, calling for efficient ViTs. Among them,…

Hardware Architecture · Computer Science 2024-04-01 Haikuo Shao , Huihong Shi , Wendong Mao , Zhongfeng Wang

The deployment of mixture-of-experts (MoE) large language models (LLMs) presents significant challenges due to their high memory demands. These challenges become even more pronounced in multi-tenant environments, where shared resources must…

Machine Learning · Computer Science 2025-05-13 HamidReza Imani , Jiaxin Peng , Peiman Mohseni , Abdolah Amirany , Tarek El-Ghazawi

Continual learning (CL) has garnered significant attention because of its ability to adapt to new tasks that arrive over time. Catastrophic forgetting (of old tasks) has been identified as a major issue in CL, as the model adapts to new…

Machine Learning · Computer Science 2025-02-20 Hongbo Li , Sen Lin , Lingjie Duan , Yingbin Liang , Ness B. Shroff

Vision Transformers (ViTs) have triggered the most recent and significant breakthroughs in computer vision. Their efficient designs are mostly guided by the indirect metric of computational complexity, i.e., FLOPs, which however has a clear…

Computer Vision and Pattern Recognition · Computer Science 2023-04-20 Zizheng Pan , Jianfei Cai , Bohan Zhuang

Optimization in multi-task learning (MTL) is more challenging than single-task learning (STL), as the gradient from different tasks can be contradictory. When tasks are related, it can be beneficial to share some parameters among them…

Computer Vision and Pattern Recognition · Computer Science 2022-12-16 Zitian Chen , Yikang Shen , Mingyu Ding , Zhenfang Chen , Hengshuang Zhao , Erik Learned-Miller , Chuang Gan

Vision Transformers (ViTs) achieve state-of-the-art segmentation accuracy but require large training datasets because each layer has unique parameters that must be learned independently. We present RD-ViT, a Recurrent-Depth Vision…

Computer Vision and Pattern Recognition · Computer Science 2026-05-06 Renjie He

The Vision Transformer (ViT) excels in global modeling but faces deployment challenges on resource-constrained devices due to the quadratic computational complexity of its attention mechanism. To address this, we propose the Semantic-Aware…

Computer Vision and Pattern Recognition · Computer Science 2025-03-04 Youbing Hu , Yun Cheng , Anqi Lu , Dawei Wei , Zhijun Li

The Mixture-of-Experts (MoE) architecture has become increasingly popular as a method to scale up large language models (LLMs). To save costs, heterogeneity-aware training solutions have been proposed to utilize GPU clusters made up of both…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-08 Yongji Wu , Xueshen Liu , Shuowei Jin , Ceyu Xu , Feng Qian , Z. Morley Mao , Matthew Lentz , Danyang Zhuo , Ion Stoica

Artificial intelligence (AI) has achieved astonishing successes in many domains, especially with the recent breakthroughs in the development of foundational large models. These large models, leveraging their extensive training data, provide…

Machine Learning · Computer Science 2026-01-27 Siyuan Mu , Sen Lin

The mixture of experts (MoE) model is a sparse variant of large language models (LLMs), designed to hold a better balance between intelligent capability and computational overhead. Despite its benefits, MoE is still too expensive to deploy…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-04-23 Haodong Wang , Qihua Zhou , Zicong Hong , Song Guo

The deployment of large-scale Mixture-of-Experts (MoE) models on edge devices presents significant challenges due to memory constraints. While MoE architectures enable efficient utilization of computational resources by activating only a…

Machine Learning · Computer Science 2025-08-26 Nishant Gavhane , Arush Mehrotra , Rohit Chawla , Peter Proenca

Adapting large-scale foundation models in multi-task scenarios often suffers from task conflict and oblivion. To mitigate such issues, we propose a novel ''model MoE-ization'' strategy that leads to a conflict- and oblivion-resistant…

Machine Learning · Computer Science 2025-10-28 Shen Yuan , Yin Zheng , Taifeng Wang , Binbin Liu , Hongteng Xu