PGP-DiffSR: Phase-Guided Progressive Pruning for Efficient Diffusion-based Image Super-Resolution
By: Zhongbao Yang , Jiangxin Dong , Yazhou Yao and more
Potential Business Impact:
Makes blurry pictures sharp with less computer power.
Although diffusion-based models have achieved impressive results in image super-resolution, they often rely on large-scale backbones such as Stable Diffusion XL (SDXL) and Diffusion Transformers (DiT), which lead to excessive computational and memory costs during training and inference. To address this issue, we develop a lightweight diffusion method, PGP-DiffSR, by removing redundant information from diffusion models under the guidance of the phase information of inputs for efficient image super-resolution. We first identify the intra-block redundancy within the diffusion backbone and propose a progressive pruning approach that removes redundant blocks while reserving restoration capability. We note that the phase information of the restored images produced by the pruned diffusion model is not well estimated. To solve this problem, we propose a phase-exchange adapter module that explores the phase information of the inputs to guide the pruned diffusion model for better restoration performance. We formulate the progressive pruning approach and the phase-exchange adapter module into a unified model. Extensive experiments demonstrate that our method achieves competitive restoration quality while significantly reducing computational load and memory consumption. The code is available at https://github.com/yzb1997/PGP-DiffSR.
Similar Papers
TinySR: Pruning Diffusion for Real-World Image Super-Resolution
CV and Pattern Recognition
Makes blurry pictures sharp, super fast.
MRI super-resolution reconstruction using efficient diffusion probabilistic model with residual shifting
CV and Pattern Recognition
Makes MRI scans clearer and faster.
NeuralRemaster: Phase-Preserving Diffusion for Structure-Aligned Generation
CV and Pattern Recognition
Keeps pictures' shapes while changing them.