AReA:面向高效高分辨率Diffusion Transformer的自适应感受野注意力机制

Hugging Face Daily Papers 论文

摘要

提出AReA:一种免训练的注意力机制,其感受野随去噪过程从粗到细自适应演化,用于实现高效高分辨率Diffusion Transformer图像生成,在多个分辨率和模型上超越现有免训练基线且速度更快。

在超出Diffusion Transformer(DiT)训练分辨率的条件下生成图像仍然面临挑战:token数量随分辨率快速增长,全注意力计算代价高昂,且输出常出现结构失真和细节退化。我们发现统一的全注意力是这两个问题的共同根源。在每一步都对所有token进行注意力计算会导致二次方的计算开销,同时使模型接触到超出训练范围的更多token和更大的相对距离,从而使模型偏离训练分布。我们观察到,注意力的有效感受野在去噪过程中会逐渐收窄——在构建结构的早期步骤中呈全局性,在细化细节的后期步骤中呈局部性。我们提出AReA,一种免训练的注意力机制,其感受野遵循去噪进程从粗到细逐步演化。在早期步骤中,AReA对下采样后的key和value施加跨尺度的粗粒度注意力,以捕捉全局上下文;在后期步骤中,则在逐步缩小的窗口内施加局部注意力。无需引入额外参数或额外训练,实验表明AReA在不同目标分辨率和不同模型上均持续优于最先进的免训练基线方法,同时速度更快、效率更高。
查看原文
查看缓存全文

缓存时间: 2026/10/14 06:52

论文页面 - AReA:面向高效高分辨率扩散 Transformer 的自适应感受野注意力

来源:https://huggingface.co/papers/2610.17227

摘要

在扩散 Transformer(DiT)的训练分辨率之外生成图像仍然极具挑战,因为随着分辨率的提高,token 数量迅速增长,全注意力计算变得昂贵,输出也往往出现结构失真和细节退化。我们发现均匀全注意力是这两个问题的共同根源。在每一步对所有 token 进行注意力计算会导致二次方级的计算开销,并使模型接触到超出训练范围的更多 token 和相对距离,从而将模型推入分布外状态。我们观察到,注意力的有效感受野在去噪过程中会逐渐收窄:在构建结构的早期步骤中是全局性的,而在精修细节的后期步骤中则变为局部性的。我们提出 AReA,一种无需训练的注意力机制,其感受野随着去噪过程从粗到细逐步演化。在早期步骤中,AReA 在下采样的键和值上执行跨尺度粗粒度注意力以捕获全局上下文;在后期步骤中,则在逐步缩小的窗口内执行局部注意力。无需额外参数或训练,实验表明 AReA 在不同目标分辨率和多种模型上均持续优于最先进的无需训练的基线方法,同时速度更快、效率更高。

查看 arXiv 页面 | 查看 PDF | 项目页面 | GitHub | 加入收藏

在你的 agent 中获取此论文:

hf papers read 2610.17227

还没有最新版 CLI?curl -LsSf https://hf.co/cli/install.sh | bash

引用此论文的模型 0

暂无关联此论文的模型

在模型的 README.md 中引用 arxiv.org/abs/2610.17227,即可从此页面链接该模型。

引用此论文的数据集 0

暂无关联此论文的数据集

在数据集的 README.md 中引用 arxiv.org/abs/2610.17227,即可从此页面链接该数据集。

引用此论文的 Spaces 1

收录此论文的合集 0

暂无收录此论文的合集

将此论文添加到合集即可从此页面链接它。

相似文章

DiffusionART:高效、稳定且可扩展的扩散模型强化方法

Hugging Face Daily Papers

本文提出DiffusionART,一种通过聚合奖励方向进行扩散模型强化学习微调的新方法,消除了一致性力并用信任区域掩码实现免EMA调参的稳定训练,在HunyuanVideo-13B上以约12倍更少的GPU小时数取得显著更高的训练奖励。

循环扩散 Transformer:在每个去噪步骤中重复运行相同的模块,并且可以在采样时调整循环深度 [R]

Reddit r/MachineLearning

介绍 Looped Diffusion Transformer 论文:通过在每个去噪步骤内重复运行共享的 transformer 块,并结合深度监督和自调制注意力,使 260M 模型以 4.9 倍更少的推理计算超越 6.5 倍大的模型,且循环深度可在采样时调节;作者还用 MIT 开源权重在 M5 Pro 上实测了不同深度的效果。