生成模型与扩散 / CVPR2026
一个模型,多种预算:面向扩散 Transformer 的弹性潜变量接口
One Model, Many Budgets: Elastic Latent Interfaces for Diffusion Transformers
把 DiT 的计算量与输入分辨率解耦:用可学习、可变长的潜变量序列作为中间接口,配合轻量 Read/Write 交叉注意力在空间词元与潜变量之间搬运信息,并通过随机丢弃尾部潜变量训练出重要性有序的表示,推理时按算力预算截断潜变量数量即可实现连续的延迟-质量权衡。做高分辨率生成或需要在固定算力下部署扩散模型的工作可直接把该模块插进现有 DiT/U-ViT/MM-DiT 主干。
扩散-Transformer交叉注意力潜变量接口