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

用于图像超分辨率的二值化扩散模型

Binarized Diffusion Model for Image Super-Resolution

Shanghai Jiao Tong University(上海交通大学);ETH Zürich(苏黎世联邦理工学院);Max Planck Institute for Informatics(马克斯·普朗克信息学研究所);Westlake University(西湖大学)

Zheng Chen, Haotong Qin, Yong Guo, Xiongfei Su, Xin Yuan, Linghe Kong, Yulun Zhang

AI总结:

针对现有二值化方法应用于图像超分辨率扩散模型时性能下降显著的问题,提出适配二值化的UNet结构与时间步感知激活调整策略的BI-DiffSR模型,实现了更优的二值化超分辨率扩散模型性能。

摘要:

先进的扩散模型(DMs)在图像超分辨率(SR)任务中表现出色,但其高内存占用和计算成本阻碍了实际部署。二值化作为一种极致压缩算法,具备有效加速扩散模型的潜力。然而,受扩散模型的结构特性及多步迭代属性影响,现有二值化方法会导致显著的性能下降。本文提出一种用于图像超分辨率的新型二值化扩散模型BI-DiffSR。首先在模型结构层面,设计了适配二值化优化的UNet架构:提出一致像素下采样(CP-Down)与一致像素上采样(CP-Up),以维持维度一致性并促进全精度信息传递;同时设计通道混洗融合(CS-Fusion)模块,增强跳跃连接中的特征融合。其次针对不同时间步的激活差异,设计时间步感知重分布(TaR)与时间步感知激活函数(TaA),二者可根据不同时间步动态调整激活分布,提升二值化模块的灵活性与表征能力。综合实验表明,本文提出的BI-DiffSR性能优于现有二值化方法。代码已发布于:https://github.com/zhengchen1999/BI-DiffSR。

英文摘要:

Advanced diffusion models (DMs) perform impressively in image super-resolution (SR), but the high memory and computational costs hinder their deployment. Binarization, an ultra-compression algorithm, offers the potential for effectively accelerating DMs. Nonetheless, due to the model structure and the multi-step iterative attribute of DMs, existing binarization methods result in significant performance degradation. In this paper, we introduce a novel binarized diffusion model, BI-DiffSR, for image SR. First, for the model structure, we design a UNet architecture optimized for binarization. We propose the consistent-pixel-downsample (CP-Down) and consistent-pixel-upsample (CP-Up) to maintain dimension consistent and facilitate the full-precision information transfer. Meanwhile, we design the channel-shuffle-fusion (CS-Fusion) to enhance feature fusion in skip connection. Second, for the activation difference across timestep, we design the timestep-aware redistribution (TaR) and activation function (TaA). The TaR and TaA dynamically adjust the distribution of activations based on different timesteps, improving the flexibility and representation alability of the binarized module. Comprehensive experiments demonstrate that our BI-DiffSR outperforms existing binarization methods. Code is released at: https://github.com/zhengchen1999/BI-DiffSR.

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

↑