Adamer: Adapting Transformer Modules For Training-Efficient Image Restoration
Ouyang Sun, Xinyu Fan, Zhan Yang, Jun Bo Long · 2023
Recently, Transformer model shines in the field of image restoration due to its unique global modeling capabilities. However, as Transformer models get larger and larger, training from scratch becomes expensive on some edge devices. In addition, due to the low computing power of edge devices, complete finetuning of the Transformer model becomes unaffordable (becomes very expensive). In our paper, we innovatively utilize the already trained partially structured network to Adapt pretrained Transformer modules (Adamer) for training-efficient image restoration. We froze some weights and introduced fast and memory-friendly Adapters. We propose Adaptation Transformer Block (ATB) to fusion of local information from CNN and global information from MDTA. Adamer achieving performance comparable to prior arts with lower computing resources in some major image restoration tasks, including image deraining, image deblurring, and image denoising.