arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2506.06276cs.CVcs.AIcs.LG

STARFlow:面向高分辨率图像合成的可扩展潜空间归一化流

STARFlow: Scaling Latent Normalizing Flows for High-resolution Image Synthesis

Apple(苹果公司)

Jiatao Gu, Tianrong Chen, David Berthelot, Huangjie Zheng, Yuyang Wang, Ruixiang Zhang, Laurent Dinh, Miguel Angel Bautista, Josh Susskind, Shuangfei Zhai

AI总结:

本文提出可扩展归一化流模型 STARFlow,以 TARFlow 为核心,结合深浅架构、潜空间建模与新型引导算法,在高分辨率类别和文本条件图像生成中接近扩散模型。

摘要:

我们提出了 STARFlow,这是一种基于归一化流的可扩展生成模型,在高分辨率图像合成中取得了强劲性能。STARFlow 的核心是 Transformer Autoregressive Flow(TARFlow,Transformer自回归流),它将归一化流的表达能力与自回归 Transformer 的结构化建模能力相结合。我们首先确立了 TARFlow 在建模连续分布方面的理论普适性。在此基础上,我们引入若干关键的架构与算法创新以显著增强可扩展性:(1)深浅设计,其中一个深层 Transformer 块捕获模型的大部分表示能力,并辅以少数计算高效且收益显著的浅层 Transformer 块;(2)在预训练自编码器的潜空间中进行建模,这被证明比直接像素级建模更有效;(3)一种显著提升样本质量的新型引导算法。关键在于,我们的模型仍是端到端归一化流,能够在连续空间中进行精确最大似然训练而无需离散化。STARFlow 在类别条件和文本条件图像生成任务中均取得了有竞争力的性能,在样本质量上接近最先进的扩散模型。据我们所知,本工作是归一化流在此规模和分辨率下有效运行的首次成功展示。

英文摘要:

We present STARFlow, a scalable generative model based on normalizing flows that achieves strong performance in high-resolution image synthesis. The core of STARFlow is Transformer Autoregressive Flow (TARFlow), which combines the expressive power of normalizing flows with the structured modeling capabilities of Autoregressive Transformers. We first establish the theoretical universality of TARFlow for modeling continuous distributions. Building on this foundation, we introduce several key architectural and algorithmic innovations to significantly enhance scalability: (1) a deep-shallow design, wherein a deep Transformer block captures most of the model representational capacity, complemented by a few shallow Transformer blocks that are computationally efficient yet substantially beneficial; (2) modeling in the latent space of pretrained autoencoders, which proves more effective than direct pixel-level modeling; and (3) a novel guidance algorithm that significantly boosts sample quality. Crucially, our model remains an end-to-end normalizing flow, enabling exact maximum likelihood training in continuous spaces without discretization. STARFlow achieves competitive performance in both class-conditional and text-conditional image generation tasks, approaching state-of-the-art diffusion models in sample quality. To our knowledge, this work is the first successful demonstration of normalizing flows operating effectively at this scale and resolution.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑