arXiv:2604.07026cs.CV2026-04

解决扩散模型生成时重要词被忽略的问题

Not all tokens contribute equally to diffusion learning

论文配图:Not all tokens contribute equally to diffusion learning
图 1 · 摘自论文原文
  • 通过动态抑制高频低语义词,平衡条件分布
  • 重加权注意力图,让关键词主导空间引导
  • 适合文本到视频生成、需要精准语义控制的场景

随着条件扩散模型的快速发展,文本到视频生成已取得显著进展。然而我们发现,这些模型在推理时常忽略语义重要的词,导致分类器自由引导下生成结果出现偏差或不完整。问题源于两个因素:训练数据中长尾词频带来的分布偏差,以及跨注意力中的空间错位,使语义重要词被信息量少的词掩盖。为此,我们提出统一框架DARE,从分布去偏和空间一致性两方面提升扩散模型的语义引导能力。首先,提出分布修正的分类器自由引导(DR-CFG),通过动态抑制高频但低语义密度的词,促使模型更好捕捉稀有语义线索,学习更均衡的条件分布。其次,提出空间表示对齐(SRA),根据词的重要性自适应重加权跨注意力图,强化高语义密度词的空间引导作用,防止低语义密度词主导注意力分配。在多个基准数据集上的实验表明,DARE在生成保真度和语义对齐上均显著优于现有方法。

原文摘要 · Abstract (English)

With the rapid development of conditional diffusion models, significant progress has been made in text-to-video generation. However, we observe that these models often neglect semantically important tokens during inference, leading to biased or incomplete generations under classifier-free guidance. We attribute this issue to two key factors: distributional bias caused by the long-tailed token frequency in training data, and spatial misalignment in cross-attention where semantically important tokens are overshadowed by less informative ones. To address these issues, we propose Distribution-Aware Rectification and Spatial Ensemble (DARE), a unified framework that improves semantic guidance in diffusion models from the perspectives of distributional debiasing and spatial consistency. First, we introduce Distribution-Rectified Classifier-Free Guidance (DR-CFG), which regularizes the training process by dynamically suppressing dominant tokens with low semantic density, encouraging the model to better capture underrepresented semantic cues and learn a more balanced conditional distribution. This design mitigates the risk of the model distribution overfitting to tokens with low semantic density. Second, we propose Spatial Representation Alignment (SRA), which adaptively reweights cross-attention maps according to token importance and enforces representation consistency, enabling semantically important tokens to exert stronger spatial guidance during generation. This mechanism effectively prevents low semantic-density tokens from dominating the attention allocation, thereby avoiding the dilution of the spatial and distributional guidance provided by high semantic-density tokens. Extensive experiments on multiple benchmark datasets demonstrate that DARE consistently improves generation fidelity and semantic alignment, achieving significant gains over existing approaches.

扩散模型文本生成注意力机制语义对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。