MoT架构让多模态大模型训练更高效,节省超半数算力。
Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models
- 按模态拆分模型参数,实现跨模态稀疏计算
- 7B模型仅用55.8%算力达基线性能,语音扩展后仅需37.2%算力
- 适合追求高性价比的多模态模型研发团队
大语言模型发展至可统一处理文本、图像和语音的多模态系统,但其训练需远超纯文本模型的数据与算力。为应对扩展挑战,我们提出稀疏多模态Transformer架构Mixture-of-Transformers(MoT),显著降低预训练算力成本。MoT将非嵌入参数按模态解耦,包括前馈网络、注意力矩阵和层归一化,实现模态专属处理,并保持对全输入序列的全局自注意力。我们在多个设置和模型规模下评估MoT。在Chameleon 7B设置中(文本与图像生成),MoT仅使用55.8%的FLOPs即达到稠密基线性能;扩展至语音后,仅需37.2%的FLOPs即可达到相近语音性能。在Transfusion设置中,7B MoT模型以三分之一算力达到稠密基线的图像性能,760M MoT模型在关键图像生成指标上超越1.4B稠密基线。系统分析显示,MoT在AWS p4de.24xlarge实例(NVIDIA A100 GPU)上,图像质量达成时间仅为基线的47.2%,文本质量为75.6%。
原文摘要 · Abstract (English)
The development of large language models (LLMs) has expanded to multi-modal systems capable of processing text, images, and speech within a unified framework. Training these models demands significantly larger datasets and computational resources compared to text-only LLMs. To address the scaling challenges, we introduce Mixture-of-Transformers (MoT), a sparse multi-modal transformer architecture that significantly reduces pretraining computational costs. MoT decouples non-embedding parameters of the model by modality -- including feed-forward networks, attention matrices, and layer normalization -- enabling modality-specific processing with global self-attention over the full input sequence. We evaluate MoT across multiple settings and model scales. In the Chameleon 7B setting (autoregressive text-and-image generation), MoT matches the dense baseline's performance using only 55.8\% of the FLOPs. When extended to include speech, MoT reaches speech performance comparable to the dense baseline with only 37.2\% of the FLOPs. In the Transfusion setting, where text and image are trained with different objectives, a 7B MoT model matches the image modality performance of the dense baseline with one third of the FLOPs, and a 760M MoT model outperforms a 1.4B dense baseline across key image generation metrics. System profiling further highlights MoT's practical benefits, achieving dense baseline image quality in 47.2\% of the wall-clock time and text quality in 75.6\% of the wall-clock time (measured on AWS p4de.24xlarge instances with NVIDIA A100 GPUs).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。