arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.09094cs.CV

Dense2MoE:将扩散Transformer重构为MoE以实现高效文本到图像生成

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)(琶洲实验室(黄埔))

Youwei Zheng, Yuxi Ren, Xin Xia, Xuefeng Xiao, Xiaohua Xie

AI总结:

Dense2MoE通过将稠密扩散Transformer重构为专家混合结构,减少激活参数并保持模型容量,实现高效文本到图像生成。

摘要:

扩散Transformer(DiT)在文本到图像生成中展现了卓越的性能;然而,其庞大的参数量导致了大量的推理开销。现有的参数压缩方法主要聚焦于剪枝,但激进的剪枝往往因模型容量减少而导致严重的性能退化。为解决这一局限,我们开创性地将稠密DiT转换为专家混合(MoE)以实现结构化稀疏化,在减少激活参数数量的同时保持模型容量。具体而言,我们将DiT块中的前馈网络(FFN)替换为MoE层,使FFN中的激活参数数量减少62.5%。此外,我们提出了块混合(MoB)机制以选择性激活DiT块,从而进一步增强稀疏性。为确保有效的稠密到MoE转换,我们设计了一个多步蒸馏流程,包括基于泰勒度量的专家初始化、带负载均衡的知识蒸馏以及用于MoB优化的组特征损失。我们将大型扩散Transformer(例如FLUX.1 [dev])转换为MoE结构,在保持原始性能的同时将激活参数减少60%,并在大量实验中超越了基于剪枝的方法。总体而言,Dense2MoE为高效文本到图像生成建立了一种新范式。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑