Dense2MoE: Restructuring Diffusion Transformer to MoE for Efficient Text-to-Image Generation
Sun Yat-sen University(中山大学);ByteDance Intelligent Creation(字节跳动智能创作);ByteDance Seed Vision(字节跳动种子视觉);Guangdong Province Key Laboratory of Information Security Technology(广东省信息安全技术重点实验室);Pazhou Lab (Huangpu)(琶洲实验室(黄埔))
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, but aggressive pruning often leads to severe performance degradation due to reduced model capacity. To address this limitation, we pioneer the transformation of a dense DiT into a Mixture of Experts (MoE) for structured sparsification, reducing the number of activated parameters while preserving model capacity. Specifically, we replace the Feed-Forward Networks (FFNs) in DiT Blocks with MoE layers, reducing the number of activated parameters in the FFNs by 62.5\%. Furthermore, we propose the Mixture of Blocks (MoB) to selectively activate DiT blocks, thereby further enhancing sparsity. To ensure an effective dense-to-MoE conversion, we design a multi-step distillation pipeline, incorporating Taylor metric-based expert initialization, knowledge distillation with load balancing, and group feature loss for MoB optimization. We transform large diffusion transformers (e.g., FLUX.1 [dev]) into an MoE structure, reducing activated parameters by 60\% while maintaining original performance and surpassing pruning-based approaches in extensive experiments. Overall, Dense2MoE establishes a new paradigm for efficient text-to-image generation.
Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock
University of Texas at Austin(德克萨斯大学奥斯汀分校);Microsoft(微软)
Mixture of Experts (MoE) is able to scale up vision transformers effectively. However, it requires prohibiting computation resources to train a large MoE transformer. In this paper, we propose Residual Mixture of Experts (RMoE), an efficient training pipeline for MoE vision transformers on downstream tasks, such as segmentation and detection. RMoE achieves comparable results with the upper-bound MoE training, while only introducing minor additional training cost than the lower-bound non-MoE training pipelines. The efficiency is supported by our key observation: the weights of an MoE transforme…
MoECa: Aligning Feature Reuse with Expert Decomposition in Diffusion Transformers
School of Computer Science, Peking University(北京大学计算机科学学院);School of Software Engineering, University of Electronic Science and Technology of China(电子科技大学软件工程学院)
In this paper, we introduce Cross-modal Alignment with mixture experts Neural Network (CameNN) recommendation model for intral-city retail industry, which aims to provide fresh foods and groceries retailing within 5 hours delivery service arising for the outbreak of Coronavirus disease (COVID-19) pandemic around the world. We propose CameNN, which is a multi-task model with three tasks including Image to Text Alignment (ITA) task, Text to Image Alignment (TIA) task and CVR prediction task. We use pre-trained BERT to generate the text embedding and pre-trained InceptionV4 to generate image patc…
GazeFormer-MoE: Context-Aware Gaze Estimation via CLIP and MoE Transformer
AIPM, School of Artificial Intelligence, Guilin University of Electronic Technology, China(AIPM,人工智能学院,桂林电子科技大学,中国);Laboratory for Imagery, Vision and Artificial Intelligence, École de Technologie Supérieure, Canada(影像、视觉与人工智能实验室,École de Technologie Supérieure,加拿大)
Routing Matters in MoE: Scaling Diffusion Transformers with Explicit Routing Guidance
Fudan University(复旦大学);Tongyi Lab, Alibaba Group(阿里云实验室,阿里巴巴集团);Zhejiang University(浙江大学);The University of Hong Kong(香港大学);MMLab;Institute of Science and Technology for Brain-inspired Intelligence, MOE Frontiers Center for Brain Science, Key Laboratory of Computational Neuroscience and Brain-Inspired Intelligence, and State Key Laboratory of Brain Function and Disorders, Fudan University(脑启发智能科学技术研究院、MOE前沿脑科学中心、计算神经科学与脑启发智能重点实验室、脑功能与疾病国家重点实验室,复旦大学)
MoE-FFD: Mixture of Experts for Generalized and Parameter-Efficient Face Forgery Detection
Rapid-Rich Object Search (ROSE) Lab, School of Electrical and Electronic Engineering, Nanyang Technology University(快速丰富目标搜索(ROSE)实验室,电气与电子工程学院,南洋理工大学);School of Computing and Artificial Intelligence, Jiangxi University of Finance and Economics(计算机与人工智能学院,江西财经大学);Department of Electrical and Engineering, City University of Hong Kong(电气与工程系,香港城市大学);Department of Computer Science and Computing, Zhejiang University City College(计算机科学与计算系,浙江工业大学城市学院);Department of Computer Science, City University of Hong Kong(计算机科学系,香港城市大学)
Experts Weights Averaging: A New General Training Scheme for Vision Transformers
School of Information Science and Technology, Fudan University(复旦大学信息科学与技术学院);Shanghai AI Lab(上海人工智能实验室);Jiangxi University of Finance and Economics(江西财经大学)
ExFusion: Efficient Transformer Training via Multi-Experts Fusion
School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University(上海交通大学电子信息与电气工程学院);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);Huazhong University of Science and Technology(华中科技大学);Soochow University(苏州大学);Chinese University of Hong Kong(香港中文大学);School of Biomedical Engineering, Shanghai Jiao Tong University(上海交通大学生物医学工程学院)
GMGaze: MoE-Based Context-Aware Gaze Estimation with CLIP and Multiscale Transformer
School of Artificial Intelligence, Guilin University of Electronic Technology, Guilin, China(桂林电子科技大学人工智能学院);The Laboratory for Imagery, Vision and Artificial Intelligence, École de Technologie Supérieure, Montreal, Canada(图像、视觉与人工智能实验室,蒙特利尔大学高级技术学院);Information Technology Department, College of Computer and Information Sciences, King Saud University, Riyadh, Saudi Arabia(计算机与信息科学学院信息科技系,沙特国王大学);Department of Cybersecurity, College of Computer Science and Engineering, Taibah University, Medina, Saudi Arabia(计算机科学与工程学院网络安全系,塔伊卜大学);Department of Networked Engineering, College of Computer Science and Engineering, Jeddah University, Jeddah, Saudi Arabia(计算机科学与工程学院网络工程系,吉达大学)
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 parameter. We present a Vision MoE (V-MoE), a sparse version of the Vision Transformer, that is scalable and competitive with the largest dense networks. When applied to image recognition, V-MoE matches the performance of state-of-the-art networks, while requiring as little as half of the compute at inference time. Further, we propose an extension to the routi…
Mixture-of-Experts (MoE), a conditional computation architecture, achieved promising performance by scaling local module (i.e. feed-forward network) of transformer. However, scaling the cross-token module (i.e. self-attention) is challenging due to the unstable training. This work proposes Sparse-MLP, an all-MLP model which applies sparsely-activated MLPs to cross-token modeling. Specifically, in each Sparse block of our all-MLP model, we apply two stages of MoE layers: one with MLP experts mixing information within channels along image patch dimension, the other with MLP experts mixing inform…
Life-IQA: Boosting Blind Image Quality Assessment through GCN-enhanced Layer Interaction and MoE-based Feature Decoupling
College of Integrated Circuits, Shanghai Jiao Tong University(集成电路学院,上海交通大学);College of Integrated Circuits, Beijing University of Chemical Technology(集成电路学院,北京化工大学);ICCI and the Institute of Image Communication and Information Processing, Shanghai Jiao Tong University(ICCI和图像通信与信息处理研究所,上海交通大学)