arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.01273cs.CV
收藏

Q-DiT4SR: 探索细节保留的扩散变换器量化以实现实景图像超分辨率

Q-DiT4SR: Exploration of Detail-Preserving Diffusion Transformer Quantization for Real-World Image Super-Resolution

Shanghai Jiao Tong University(上海交通大学);Huawei(华为)

Xun Zhang, Kaicheng Yang, Hongliang Lu, Haotong Qin, Yong Guo, Yulun Zhang

AI总结:

本文提出Q-DiT4SR,一种专门针对基于扩散变换器的实现实景图像超分辨率的后训练量化框架,通过引入层次化SVD和变异性感知时空混合精度方法,在保持细节的同时实现高效的模型压缩和加速。

摘要:

近年来,扩散变换器(DiTs)在实现实景图像超分辨率(Real-ISR)中崭露头角,能够生成高质量的纹理,但其沉重的推理负担阻碍了实际应用。尽管后训练量化(PTQ)是加速的有希望的解决方案,但现有超分辨率方法大多集中在U-Net架构上,而通用的DiT量化通常针对文本到图像任务设计。直接将这些方法应用于基于DiT的超分辨率模型会导致局部纹理严重退化。因此,我们提出了Q-DiT4SR,这是首个专门针对基于DiT的Real-ISR的PTQ框架。我们提出了H-SVD,一种层次化SVD,它在匹配的参数预算下集成了一个全局低秩分支和一个局部块状秩-1分支。我们进一步提出了变异性感知时空混合精度:VaSMP在无数据的情况下基于率-失真理论分配跨层权重位宽,而VaTMP通过动态规划(DP)在最小校准下调度跨扩散时间步的层内激活精度。在多个实现实景数据集上的实验表明,我们的Q-DiT4SR在W4A6和W4A4设置下均实现了SOTA性能。值得注意的是,W4A4量化配置将模型大小减少了5.8倍,并将计算操作减少了6.14倍。我们的代码和模型将在https://github.com/xunzhang1128/Q-DiT4SR上提供。

英文摘要:

Recently, Diffusion Transformers (DiTs) have emerged in Real-World Image Super-Resolution (Real-ISR) to generate high-quality textures, yet their heavy inference burden hinders real-world deployment. While Post-Training Quantization (PTQ) is a promising solution for acceleration, existing methods in super-resolution mostly focus on U-Net architectures, whereas generic DiT quantization is typically designed for text-to-image tasks. Directly applying these methods to DiT-based super-resolution models leads to severe degradation of local textures. Therefore, we propose Q-DiT4SR, the first PTQ framework specifically tailored for DiT-based Real-ISR. We propose H-SVD, a hierarchical SVD that integrates a global low-rank branch with a local block-wise rank-1 branch under a matched parameter budget. We further propose Variance-aware Spatio-Temporal Mixed Precision: VaSMP allocates cross-layer weight bit-widths in a data-free manner based on rate-distortion theory, while VaTMP schedules intra-layer activation precision across diffusion timesteps via dynamic programming (DP) with minimal calibration. Experiments on multiple real-world datasets demonstrate that our Q-DiT4SR achieves SOTA performance under both W4A6 and W4A4 settings. Notably, the W4A4 quantization configuration reduces model size by 5.8$\times$ and computational operations by 6.14$\times$. Our code and models will be available at https://github.com/xunzhang1128/Q-DiT4SR.

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

↑