1. 项目概述:当扩散模型遇上Transformer
2025年NIPS这篇论文的标题已经透露了它的野心——把当下最火的几个生成模型架构全都揉在一起。Generative Pre-trained Autoregressive Diffusion Transformer(GPADT)这个冗长的名字里,每个单词都代表着过去五年AI生成领域的一个技术里程碑。我读完论文后的第一反应是:这不就是生成式AI的"复仇者联盟"吗?
作为同时玩过Diffusion和Transformer的老玩家,我深知这两个架构的互补性。扩散模型擅长高质量生成但推理速度慢,Transformer长于序列建模却对长程依赖头疼。作者团队把自回归的渐进生成、扩散模型的去噪过程、以及Transformer的注意力机制这三者用预训练范式串联起来,在图像和文本跨模态生成任务上实现了SOTA。最让我惊讶的是,他们在附录B.3展示的消融实验证明,这种混合架构在COCO数据集上的FID分数比纯扩散模型提升了37%,而推理速度反而快了2.8倍。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 核心架构拆解
2.1 三位一体的模型设计
GPADT的核心创新在于它的级联式处理流程。输入文本首先通过一个预训练的T5编码器得到语义嵌入,然后进入三个关键阶段:
-
自回归规划器(Autoregressive Planner):用稀疏Transformer逐token预测内容大纲,类似写文章先列提纲。这里采用了GShard的专家混合(MoE)设计,每个专家负责不同抽象层次的特征。
-
扩散调度器(Diffusion Scheduler):动态调整后续扩散步骤的噪声水平。论文里那个可学习的调度函数特别巧妙:
python复制def schedule(step, total_steps): # 基于当前步骤和内容复杂度自适应调整噪声 beta_t = 1 - (step / total_steps)**0.3 + 0.1 * content_complexity return torch.clamp(beta_t, 0.05, 0.95) -
Transformer解码器:在扩散过程的每个step,用多头注意力融合规划器输出和当前噪声特征
