Reversible nonlinear transformer for image deblurring
摘要
Although image deblurring methods have achieved impressive results in terms of PSNR in recent years, most existing approaches still suffer from high GPU memory consumption. To address this issue, we develop an approach to reduce GPU memory consumption. Specifically, the conventional encoder-decoder architecture of the Transformer network is restructured into a reversible architecture, reducing memory consumption during backpropagation. Additionally, a Multi-Scale Feature Fusion Module (MSFM) is introduced between the encoder and decoder to fuse low-level details and high-level semantic information. To handle the spatial variability of blur kernels across images, a nonlinear feature processing mechanism is presented, improving blur removal at varying levels. Furthermore, a Channel Importance Learner (CIL) mechanism is designed, which prioritizes the processing of high-value feature channels, thereby enhancing the network’s learning efficiency and overall performance. Experimental results demonstrate that, when processing