首个三模态扩散模型,统一文本、图像、音频生成。
The Design Space of Tri-Modal Masked Diffusion Models
- 从头预训练三模态扩散模型,支持文本、图像、音频联合生成。
- 在6.4万亿标记上训练30亿参数模型,跨模态任务表现强劲。
- 提出新重参数化方法,解耦物理与逻辑批大小,降低调参成本。
离散扩散模型已成为自回归语言模型的有力替代方案,近期研究已将基础单模态模型初始化并微调用于双模态生成。本文首次提出从头预训练的三模态掩码扩散模型,涵盖文本、图像-文本和音频-文本数据。我们系统分析了多模态缩放定律、模态混合比例、噪声调度和批量大小效应,并提供优化的推理采样默认设置。批量大小分析催生了一种基于随机微分方程(SDE)的新重参数化方法,消除了近期工作中对最优批量大小调优的需求。该方法将物理批量大小(由计算资源决定,如GPU饱和度、浮点运算效率、实际时间)与逻辑批量大小(用于平衡随机优化中的梯度方差)解耦。最后,我们在6.4万亿标记上预训练了一个初步的30亿参数三模态模型,展示了统一设计的能力,在文本生成、文生图和文生语音任务中均取得优异效果。本工作是迄今最大规模的多模态离散扩散模型系统性研究,为多模态扩展行为提供了深刻见解。
原文摘要 · Abstract (English)
Discrete diffusion models have emerged as strong alternatives to autoregressive language models, with recent work initializing and fine-tuning a base unimodal model for bimodal generation. Diverging from previous approaches, we introduce the first tri-modal masked diffusion model pretrained from scratch on text, image-text, and audio-text data. We systematically analyze multimodal scaling laws, modality mixing ratios, noise schedules, and batch-size effects, and we provide optimized inference sampling defaults. Our batch-size analysis yields a novel stochastic differential equation (SDE)-based reparameterization that eliminates the need for tuning the optimal batch size as reported in recent work. This reparameterization decouples the physical batch size, often chosen based on compute constraints (GPU saturation, FLOP efficiency, wall-clock time), from the logical batch size, chosen to balance gradient variance during stochastic optimization. Finally, we pretrain a preliminary 3B-parameter tri-modal model on 6.4T tokens, demonstrating the capabilities of a unified design and achieving strong results in text generation, text-to-image tasks, and text-to-speech tasks. Our work represents the largest-scale systematic open study of multimodal discrete diffusion models conducted to date, providing insights into scaling behaviors across multiple modalities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。