探索生成模型新范式,提升图像生成质量与效率
Exploring the Design Space of Transition Matching
- 用小型模块实现高效渐进式生成,替代传统扩散模型的固定路径
- 56个17亿参数模型实验表明,MLP头+高频采样达最优性能
- 揭示哪些设计能提升效果,哪些已接近极限,指导后续优化
过渡匹配(Transition Matching, TM)是一种新兴的生成建模范式,可统一扩散模型、流匹配模型及连续状态自回归模型。与以往方法类似,它通过逐步将噪声样本转化为数据样本,但使用一个内部生成模型来实现过渡步骤,使转换过程更具表达力。为使该范式可训练,TM采用大型骨干网络和小型“头部”模块来高效执行生成步骤。本文系统研究了头部模块在时间连续双向变体中的设计、训练与采样策略,共训练56个17亿参数的文本到图像模型(产生549次独立评估)。我们分析了头部架构、训练方式及一类随机采样器的影响,评估其对生成质量、训练效率和推理效率的影响。结果表明,采用MLP头部、特定时间加权训练并结合高频采样器的方案,在所有指标上表现最佳,达到当前测试基线中的最先进水平;而使用序列缩放的Transformer头部搭配低频采样则在图像美学方面表现优异。最后,实验揭示了最具潜力的设计方向,也指出了难以进一步提升的冗余选项。
原文摘要 · Abstract (English)
Transition Matching (TM) is an emerging paradigm for generative modeling that generalizes diffusion and flow-matching models as well as continuous-state autoregressive models. TM, similar to previous paradigms, gradually transforms noise samples to data samples, however it uses a second ``internal'' generative model to implement the transition steps, making the transitions more expressive compared to diffusion and flow models. To make this paradigm tractable, TM employs a large backbone network and a smaller "head" module to efficiently execute the generative transition step. In this work, we present a large-scale, systematic investigation into the design, training and sampling of the head in TM frameworks, focusing on its time-continuous bidirectional variant. Through comprehensive ablations and experimentation involving training 56 different 1.7B text-to-image models (resulting in 549 unique evaluations) we evaluate the affect of the head module architecture and modeling during training as-well as a useful family of stochastic TM samplers. We analyze the impact on generation quality, training, and inference efficiency. We find that TM with an MLP head, trained with a particular time weighting and sampled with high frequency sampler provides best ranking across all metrics reaching state-of-the-art among all tested baselines, while Transformer head with sequence scaling and low frequency sampling is a runner up excelling at image aesthetics. Lastly, we believe the experiments presented highlight the design aspects that are likely to provide most quality and efficiency gains, while at the same time indicate what design choices are not likely to provide further gains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。