arXiv:2606.23682cs.CV2026-06

通过删减参考图像的冗余令牌,实现高效可控图像生成。

Keep The Essentials: Efficient Reference Conditioned Generation via Token Dropping

论文配图:Keep The Essentials: Efficient Reference Conditioned Generation via Token Dropping
图 1 · 摘自论文原文
  • 仅保留参考图像的关键令牌,构建稀疏表示
  • 单参考生成提速2倍,多参考生成提速4倍
  • 支持任务自适应选择,适合高效率图像编辑场景

参考引导的扩散模型通过利用输入图像中的元素实现高度可控的图像生成,但其运行时计算成本随参考数量显著增加。尽管提示驱动生成的效率已有广泛研究,参考引导模型的效率问题仍缺乏关注。尤其,将参考信息表示为密集令牌网格造成资源浪费。本文提出稀疏上下文(Sparse Context)方法,通过保留参考令牌的子集来构建稀疏表示。我们发现,仅在推理时随机丢弃大量参考令牌,即可基本保持生成能力。为此,我们在训练中引入不同比例的随机令牌丢弃,提升模型对部分参考的鲁棒性。关键在于,该策略使模型摆脱特定令牌选择规则的依赖,从而在推理时可灵活采用任务感知的选择策略,优先保留参考图像中最关键区域的令牌,按需分配令牌预算。大量实验表明,本方法在多参考生成中实现4倍加速,在单参考生成中实现2倍加速,且在空间对齐编辑与主体驱动生成任务中均保持视觉质量。

原文摘要 · Abstract (English)

Reference-based diffusion models enable highly controllable image generation by leveraging elements from input images to guide prompt-driven synthesis. However, these models are computationally expensive in runtime, and their cost scales severely with the number of input references. While the efficiency of diffusion models has been extensively studied in the context of prompt-driven generation, it remains largely under-explored in the realm of reference-based models. This setting presents unique challenges not addressed by methods focusing solely on generation. In particular, the wasteful representation of references as dense token grids offers significant opportunities for improvement. In this work, we present Sparse Context, a method for constructing sparse reference representations by retaining only a reduced subset of reference tokens. We observe that even without modifying the model, dropping a significant portion of reference tokens at inference time largely preserves its generation capabilities. To fully realize this potential, we fine-tune the model with random token dropping at varying ratios, encouraging robustness to partial reference representations. Crucially, this training strategy decouples the model from any specific token selection rule, allowing flexible control at inference time. At inference time, instead of random dropping, we apply task-aware token selection strategies that prioritize the most informative regions of the reference images, adapting the token budget to the input and task requirements. Extensive experiments show our method achieves a 4x increase in inference speed for multi-reference generation and an 2x for single reference generation. Importantly, this efficiency is achieved without compromising visual quality across both spatially-aligned editing and subject-driven generation.

扩散模型图像生成效率优化参考引导

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