arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.27829cs.CV

CSD: 内容感知的投机解码用于高效图像生成

CSD: Content-aware Speculative Decoding for Efficient Image Generation

East China Normal University(东华大学);Huawei Foundation(华为基金会);Key Laboratory of Advanced Theory and Application in Statistics and Data Science-MOE(统计与数据科学高级理论与应用关键实验室-教育部)

Mingcheng Wang, Junbo Qiao, Yunchen Li, Lingfu Jiang, Wei Li, Jie Hu, Jiao Xie, Zhou Yu, Xinghao Chen, Guixu Zhang, Shaohui Lin

AI总结:

提出内容感知投机解码算法CSD,通过熵驱动的概率松弛和最优重采样策略,在保持生成质量的同时加速自回归图像生成。

摘要:

投机解码(SD)已成为加速自回归模型推理的关键解决方案。然而,在图像生成领域,它面临接受率低的挑战,直接放宽标准会导致图像质量下降。本文提出了一种新颖的内容感知投机解码算法,称为CSD,该算法将基于熵的概率松弛机制与最优重采样策略相结合,以提高自回归图像生成的推理效率。通过利用图像不同区域固有的信息不确定性,CSD动态调整候选令牌的接受概率,增加低细节区域的接受率以加速生成。此外,引入分布对齐滤波器以确保输出分布与目标模型对齐,从而显著提高生成质量。在Lumina-mGPT和Janus-Pro上进行的实验证明了所提出的CSD的优越性。我们的源代码可在以下网址获取:https://this https URL。

英文摘要:

Speculative decoding (SD) has emerged as a key solution to accelerate the inference of autoregressive models. However, in the field of image generation, it faces the challenge of low acceptance rates, and directly relaxing its criteria leads to degradation in image quality. In this paper, we propose a novel content-aware speculative decoding algorithm, termed CSD, which integrates an entropy-based probability relaxation mechanism with an optimal resampling strategy to enhance the inference efficiency for autoregressive image generation. By leveraging the informational uncertainty inherent in different regions of an image, CSD dynamically adjusts the acceptance probability of candidate tokens, increasing the acceptance rate in low-detail areas to accelerate generation. Moreover, a distribution alignment filter is introduced to ensure the output distribution to be aligned with the target model, which significantly improves the generative quality. Experiments conducted on Lumina-mGPT and Janus-Pro demonstrate that the superiority of the proposed CSD. Our source code is available at https://github.com/aderfebr/CSD.

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

↑