## Contribution-based Low-Rank Adaptation with Pre-training Model for Real Image Restoration

Dongwon Park , Hayeon Kim , and Se Young Chun 1 2 1 2 , ⋆

1 IPAI &amp; INMC, 2 Dept. of Electrical and Computer Engineering, Seoul National University, Republic of Korea, {dong1park, khy5630, sychun}@snu.ac.kr

Abstract. Recently, pre-trained model and efficient parameter tuning have achieved remarkable success in natural language processing and high-level computer vision with the aid of masked modeling and prompt tuning. In low-level computer vision, however, there have been limited investigations on pre-trained models and even efficient fine-tuning strategy has not yet been explored despite its importance and benefit in various real-world tasks such as alleviating memory inflation issue when integrating new tasks on AI edge devices. Here, we propose a novel efficient parameter tuning approach dubbed contribution-based low-rank adaptation (CoLoRA) for multiple image restorations along with effective pre-training method with random order degradations (PROD). Unlike prior arts that tune all network parameters, our CoLoRA effectively finetunes small amount of parameters by leveraging LoRA (low-rank adaptation) for each new vision task with our contribution-based method to adaptively determine layer by layer capacity for that task to yield comparable performance to full tuning. Furthermore, our PROD strategy allows to extend the capability of pre-trained models with improved performance as well as robustness to bridge synthetic pre-training and real-world fine-tuning. Our CoLoRA with PROD has demonstrated its superior performance in various image restoration tasks across diverse degradation types on both synthetic and real-world datasets for known and novel tasks. Project page: https://janeyeon.github.io/colora/ .

Keywords: Efficient fine-tuing · Low-rank adaptation · Pre-training

## 1 Introduction

Image restoration (IR) is a fundamental low-level computer vision task that aims to recover the original clean image from the input data that was degraded by noise [29, 39, 92, 101], blur [36, 37, 55, 70, 80], and / or bad weather conditions [40, 81, 88, 95]. It does not only enhance the visual quality of images, but also improves the performance of mid to high-level vision downstream tasks such as classification [26,35], object detection [67,69], and autonomous driving [3,51].

⋆ Corresponding author.

However, existing IR methods require expensive training data per degradation and individual training of a separate model from scratch per restoration task.

Natural language processing (NLP) and high-level computer vision also have similar problems of high cost data collection and demanding training for diverse tasks. However, large-scale pre-trained models [4,17,21,59] and efficient parameter tuning methods [6,18,27,28,32,61,62,85,85] have been recently proposed to address this issue for effectively adapting to new tasks. Large-scale pre-training models are constructed, they resolve the data scarcity problem, reduce training time, and improve generalization performance through powerful learning capability and scalability for diverse tasks. Moreover, the efficient parameter tuning methods adjust only a small number of parameters during fine-tuning for each task, so they can significantly reduce memory and storage cost. In IR, efficient parameter tuning has not been investigated yet, but pre-training approaches for IR [7,13,43,48] have recently emerged as a noteworthy development.

The existing pre-training approaches [7,43,48] for IR usually construct synthetic training data with known multiple degradations (called synthetic degradation functions), and then fine-tune the full network parameters to the real world data with novel degradation in Fig. 1(a). This pre-training technique offers the benefit of optimizing performance using limited data in real IR tasks where obtaining a real-world dataset containing pairs of degraded and clean images can be challenging. However, these methods have the disadvantage of requiring a large number of network parameters and storage memory as the number of tasks increases because the entire network parameters must be trained for new tasks, as illustrated in Fig. 1. Therefore, when various functions exist, operation may be difficult in an intelligent edge device with limited computing capacity.

In this paper, we propose an efficient parameter tuning method using a new adapter called contribution-based low-rank adaptation (CoLoRA) with a Pretraining with Random Order Degradation (PROD). These approaches enable the pre-trained network to adapt to IR tasks with novel degradations, as depicted in Fig. 1(b). Unlike previous works that fine-tune the entire pre-trained network, our proposed method introduces adapters for efficient parameter tuning in novel IR tasks. While prior works must have a separate full-size fine-tuned network for each novel IR task, our proposed method only need to store the small tunable adaptor for each task (approximately 7% of the entire network parameters for comparable performance to full-size tuning), so that it will be advantageous in terms of memory for multiple target tasks. The inefficiency with full fine-tuning is especially severe in modern network architectures for IR such as Restormer [86] and IPT [7]. Our proposed PROD strengthens the pre-train model by generating low-quality training images using synthetic degradation functions and their random combinations. This is achieved by synergistically leveraging both random single degradations [7, 43] and deterministic multiple degradations [48] so that the pre-text task can be expanded for adapting IR to novel real-world complex degradation. Here is the summary of our contributions:

- -We propose a novel efficient parameter tuning method with pre-training for IR, dubbed CoLoRA with PROD, using synthetic degradation functions

Fig. 1: Illustrations of tuning strategies for novel image restoration tasks. (a) Existing strategies [7, 43, 48] for fully fine-tuning a pre-trained model for a new task. (b) Our proposed CoLoRA method enables parameter-efficient fine-tuning by freezing the pretrained model and adjusting the additional adapter for novel image restoration tasks.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-4a5529b6-5a20-4278-ab02-9e76ecabf504.jpg)

for pre-training and small amount of real data for parameter tuning. Our proposed method demonstrates flexible enough to work with diverse network architectures like CNNs and vision transformers, achieving state-of-the-art performance on 6 IR tasks with real data.

- -Our CoLoRA allows efficient parameter tuning by freezing the pre-trained model and tuning the additional adaptor with adaptive layer by layer capacity per task, still achieving state-of-the-art performance on real 6 IR tasks.
- -Our PROD has trained an excellent pre-training network with synthetic data that outperformed other state-of-the-art methods on 6 IR tasks with real data after full fine-tuning.

## 2 Related works

## 2.1 NLP and High-level Vision Tasks

Pre-training. In NLP, self-supervised pre-training utilizing large models and billions of data [4, 17, 66] is a fundamental option. These methods train the model to predict missing content by hiding part of the input sequence. Various self-supervised pre-training methodologies [19, 21] have also been proposed in high-level computer vision fields. Recently, contrastive learning [9, 24] and transformer-based masked autoencoder methods [23,83] have emerged, enhancing semantic information learning. These pre-trained networks provide better

generalization performance in various downstream tasks through efficient representation learning and extensive data utilization, even with less training data. Parameter efficient fine-tuning. To adapt the aforementioned pre-trained models to various downstream tasks, a fine-tuning process is necessary. However, a major drawback of fine-tuning is that it comes with long training time [84] and memory discomfort [5], as the learning parameters for the new task are the same as the pre-trained model. To alleviate this, the topic of parameter efficient finetuning methods have been actively researched in the NLP [18,32,62,85] and highlevel computer vision [31, 89, 97] fields. These methodologies propose adjusting only specific parameters of the model, such as the adaptor [27,61], biases [6,85], prompts [31,34,44], etc. Recently in image generation tasks [52,63,72,94], ControlNet [94] and LoRA [28] have been proposed. The ControlNet [94] is a method to adapt to new tasks by additionally using parameters from the decoder and middle block. However, this method has a disadvantage of increasing inference time. On the other hand, LoRA does not increase inference time by fixing all weights of pre-trained models and efficiently adjusting only learnable rank decomposition matrix parameters for downstream tasks. Efficient parameter tuning has not been studied in IR, but recently pre-training methods [7, 13, 43, 48] for IR have begun to be proposed. We propose a novel efficient parameter tuning method, CoLoRA with PROD method specifically designed for IR tasks.

## 2.2 Low-Level Vision Tasks

Image restoration for multiple degradations. Recently, methodologies [8, 14,54,75,78,86,87] have been proposed that use a single network architecture to build multiple independent restoration models, each trained on different degradation datasets, demonstrating high performance in various degradation tasks. However, these methods require numerous network parameters as it necessitates an independent network trained for each degradation tasks. To address this, an all-in-one IR methods [12, 41, 42, 42, 46, 53, 57, 57, 76, 90, 99] have been proposed. Firstly, there was a method [12] of learning a unified network for various degradations through knowledge distillation techniques. Secondly, there were methods [41,46,53,76,90] of using an adaptor to enable the unified network to adapt to various degradations. Lastly, there were methods [42, 57, 99] of using an additional module corresponding to a specific degradation in a unified model, including a classifier that selects the additional module. All-in-one IR methods achieved high performance in various degradation tasks. However, these methods have the limitation of requiring to re-train the unified model and adaptor or classifier to extend to new tasks. Furthermore, all-in-one methods train from scratch without any pre-trained model.

Pre-training with synthetic data. Constructing a pre-trained model for IR tasks requires a significant quantity of low-quality images paired with their highquality counterparts [38, 93]. However, obtaining such image pairs in the real world presents considerable cost and difficulty [48]. To deal with such problems, pre-training methods using synthetic degradation functions have been proposed as illustrated in Fig. 1. IPT [7] and EDT [43] proposed methods that select

single degradation from multiple synthetic degradation functions such as superresolution, Gaussian noise, and rain, thus generate low-quality images. HAT [13] focuses on selecting a single degradation from down-scale degradation functions. DegAE [48] employs pre-determined Gaussian blur, noise, and JPEG degradation functions in a fixed order and introduces a degradation autoencoder to integrate features. However, DegAE [48] does not utilize global residual learning to aggregate features into a single representation. This omission may lead to a decrease in performance, as the original network's capabilities are not fully utilized. These methods may not be well-suited for addressing real-world and complex degradation tasks because their limited representation during pre-training. Moreover, there exists a significant constraint that the entire network must be fine-tuned for each new task. Consequently, there arises a challenge of storing and deploying distinct copies of base parameters for individual tasks, induces the issue of cost and memory. To alleviate it, we propose novel CoLoRA, an efficient fine-tuning methodology, and PROD, an effective pre-training method.

## 3 Method

We propose a Contribution based efficient LoRA (CoLoRA) with Pre-training with Random Order Degradation (PROD) for IR, as illustrated in Fig. 2. Section 3.1 introduces the pre-training method PROD, and Section 3.2 investigates the quantified contribution to each layer. In Section 3.3, we propose CoLoRA that adjusts the ratio of learnable network parameter based on contributions.

Fig. 2: The overview of our proposed CoLoRA with PROD. (a) Our PROD leverages high-quality clean images and synthetic degraded low-quality images for pre-training the model. (b) Our proposed Contribution based efficient LoRA (CoLoRA) for new IR tasks. The proposed CoLoRA is configured to have different ratio of learnable network parameter ( δ ) for each layer based on quantified contributions (Sec 3.2), enabling efficient fine-tuning for new tasks. (c) CoLoRA can be adjusted according to contribution.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-16274cc1-f34f-422f-8b3a-68336134c4e9.jpg)

## 3.1 Pre-training with Random Order Degradation (PROD)

We propose a PROD that randomly applies synthetic degradation to clean images during the pre-training, as illustrated in Fig. 2. Detailed information on distortion functions and magnitudes are in the supplementary materials. Applying 1 to N synthetic degradations to a clean image, PROD can represent a total of ( H N +1 -1) / H ( -1) different types of degraded images where N is set to 6 and H (= 5) represents the number of degradation functions. This PROD method enables about 137 K kinds of degradation representations, which is about 4 K times more than the previous single [7,43] and fixed order [48] synthetic degradation methods. Note that a similar random degradation idea was proposed in [91] and [15], but it was designed for super resolution and classification, not for pretraining for multiple tasks. The experimental results are in the supplementary material. The PROD significantly expands the scalability and generalization of the pre-train model, bringing excellent performance in IR with real data.

## 3.2 The Key Components of Fine-Tuning for a New Task

We utilized Filter Attribution method based on Integral Gradient (FAIG) [82] scores to quantify the major contributing network parts for new IR tasks. FAIG is measured through Integrated Gradients (IG) calculations using pre-trained and fine-tuned models. The FAIG calculation for each layer involves the baseline model ( θ ba ) and the target model ( θ ta ), defined as follows:

$$\text{FAIG} _ { i } ( \theta _ { b a } ^ { i }, \theta _ { t a } ^ { i }, x ) \approx \sum _ { j = 1 } \left | \frac { 1 } { M } [ \theta _ { b a } ^ { i, j } - \theta _ { t a } ^ { i, j } ] \sum _ { t = 0 } ^ { M - 1 } \left [ \frac { \partial \mathcal { L } ( \rho ( \beta _ { t } ), x ) } { \partial \rho ( \beta _ { t } ) } \right ] _ { i, j } \right |, \quad ( 1 )$$

where M represents the total number of steps in the integral approximation, setting to 100 as in FAIG. β t , j , i and ρ are t/M , the kernel index, layer index

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-971ce579-b129-416e-a9ba-c488fd17f9bb.jpg)

(b Results from fine-tuning specific layers

Fig. 3: (a) For each layer, we measured the FAIG score using a pre-trained model and a fine-tuned model specifically tuned for the specific task to observe its contribution. (b) Experimental results according to fine-tuning location for a blur task. The encoder and decoder occupy 18% of the total network parameters, while the middle layers account for 80%. The encoder and decoder have higher FAIG scores than the middle. Bias and normalization have higher FAIG values compared to the weight layer (Conv).

and interpolation between models, respectively. The high value of FAIG indicates a significant importance for new tasks [57]. Conversely, lower values indicate high similarity between the baseline and target model, implying low importance for the new task. In Fig. 3, we measure the FAIG score of each block using the full fine-tuned NAFNet [8] on real IR tasks after PROD execution. The FAIG values are an average of the 6 real IR tasks. Fig. 3 (a) demonstrates that the encoder and the decoder part of the network have significantly higher FAIG values than the middle layer. This indicates that the middle layers, accounting for a substantial fraction (around 80%) of the total network parameters, have a small contribution compared to the encoder and decoder, which account for a small portion (around 18%). Bias and Normalization layers have higher FAIG values than weight layers, despite a smaller portion of the total network parameters.

Based on these observations, we conducted experiments on partially finetuning for the new task, as shown in Fig. 3 (b). Fine-tuning only the encoder and decoder demonstrated superior performance with fewer network parameters compared to other methods. We observed that by fine-tuning the encoder and decoder parts with high FAIG scores, the network performance can be improved without unnecessary tuning to the middle layers. These experiments suggest that FAIG can be helpful in quantifying how specific parts of the network contribute to new tasks. These observations lead to our proposed CoLoRA.

## 3.3 Contribution-based Low-Rank Adaptation (CoLoRA)

Hu et al. [28] proposed a method known as LoRA aimed at fine-tuning only small network parameters. The training weight matrix △ W is represented as follows:

$$W _ { 0 } + \triangle W = W _ { 0 } + B A,$$

where W 0 ∈ R d × k , B ∈ R d × r , A ∈ R r × k , and the r ≪ min( d, k ) is pre-trained weight matrix, projection weight matrices, and rank, respectively. In the finetuning phase, A and B consist of learnable parameters while W 0 remains frozen and does not undergo gradient updates. The output vectors are coordinate-wise sum of W 0 and △ W = BA , which have the same input. The conventional LoRA employs a fixed rank value of r for all layers of the network because optimizing the low-rank parameter for each layer is infeasible, thus imposes constraints on the efficient utilization of parameters and limits its overall performance.

To address these limitations, we propose CoLoRA, a method that flexibly adjusts the ratio of learnable network parameter based on contributions. In Section 3.2, we adjust the ratio of learnable parameters δ by applying different values of r for each layer based on the quantified FAIG score. The δ in CoLoRA, according to the size of r , is defined as δ i = r i ( d i + k i ) / d ( i × k i ) , where i denotes the network layer index, and d and k represent the size of W 0 . For the FAIG scores measured for each block as in Fig. 3 (a), excluding the intro and end layers, we normalize to ensure the maximum value = 1. Using the normalized FAIG score, we propose a threshold-based scaling to determine δ without increasing

complexity as follows:

$$\delta ^ { s } = \begin{cases} \text{Norm} ( F A I G ^ { s } ) \times \alpha & \text{if Norm} ( F A I G ^ { s } ) > 0. 5, \\ \text{Norm} ( F A I G ^ { s } ) \times \beta & \text{otherwise}, \end{cases}$$

where α β , and Norm are two scale factors and a normalization function. FAIG s denotes the average value of FAIG i values for each stage and s is the stage index. We used scale factors ( α, β ) to optimize training parameters. The δ values are applied uniformly across all tasks. Thanks to the proposed CoLoRA, only a small fraction of learnable network parameters (approximately 7%) is allocated for each IR task, resulting in a significant reduction in memory usage compared to full fine-tuning (100%) for additional tasks. Our proposed CoLoRA with PROD method yields high performance using less memory compared to the previous work [48] with full-tuning in Section 4. Our proposed CoLoRA can merge with fixed pre-trained model weights with a simple linear model. Therefore, there is no inference delay compared to the full fine-tuning method [28]. Detailed δ values for each network are in the supplementary material.

Bias and Normalization layer. The parameters of Bias and Normalization layers, despite a small portion of the entire network, have high FAIG values. Therefore, we adjust the bias and normalization layer together during finetuning. Note that LoRA [28] does not update the Bias and Normalization layers.

## 4 Experiments

Experiment setups. We evaluated the proposed CoLoRA with PROD on 6 real IR tasks using CNN-based NAFNet [8] and transformer-based Restormer [86]. To evaluate our method, we compare it with the previously pre-trained methodology, DegAE [48]. In Restormer, the DegAE [48] uses publicly released pretrained weights, and in NAFNet, the DegAE [48] was reproduced based on the publicly released code. In CoLoRA, we experimentally used α = 1 and β = 0.2 for NAFNet, and α = 0.75 and β = 0.1 for Restormer. The learning iteration for the pre-training model and fine-tuning were set to 200K and 10K, respectively. The learning rate starts at 1e-3 for NAFNet and 3e-4 for Restormer and gradually decreases to 1e-6 according to the cosine annealing schedule. The evaluation of experiment results was performed using PSNR. In pre-training, PSNR loss was used, and in the fine-tuning process, NAFNet and Restormer used PSNR loss and L1, respecitvely. Detailed information is in the supplementary material. Six real-world image restoration task datasets: To evaluate our proposed method, we conducted experiments using 6 real-world Image Restoration (IR) datasets: Real Rain, Raindrop, Rain&amp;Raindrop (RainDS [65]), Noise (SIDD [1]), Haze (SMOKE [33]), and Blur (BSD [96]). RainDS [65] consists of 120 training data and 100 test data for each task, Rain and Raindrop, Rain and Raindrop. SMOKE [33] includes of 120 training images and 12 test images. BSD [96] comprises of 18,000 training images and 3,000 test images (2ms-16ms). SIDD [96] data comprises of 160 training images and 1,280 validation samples.

Table 1: Benchmark results for real Rain, Raindrop, Raind&amp;Raindrop, Haze and Blur test datasets in PSNR (dB). Our CoLoRA with PROD achieved comparable performance on NAFNet and Retsomer, respectively, by updating only a small 2M and 2.9M of tuned network parameters, in contrast to the full fine-tuning of 29M and 26M.

| Method                        |   Rain |   Raindrop |   Rain&Raindrop | Tuned par.   |
|-------------------------------|--------|------------|-----------------|--------------|
| SPANet [77]                   |  22.17 |      20.43 |           19.43 | -            |
| PReNet [68]                   |  24.56 |      22.33 |           21.2  | -            |
| DRDN [16]                     |  23.83 |      21.14 |           20.1  | -            |
| CCN [65]                      |  26.83 |      24.81 |           23.09 | -            |
| NAFNet (NAF.) + Full [8]      |  27.09 |      24.86 |           23.79 | 29M          |
| DegAE (NAF.) + Full [48]      |  27.22 |      24.96 |           24.09 | 29M          |
| PROD (NAF.) + Full            |  27.42 |      25.02 |           24.37 | 29M          |
| CoLoRA w. PROD (NAF.)         |  27.37 |      25.22 |           24.62 | 2M           |
| Restormer (Rest.) + Full [86] |  26.93 |      24.73 |           23.53 | 26M          |
| DegAE (Rest.) + Full [48]     |  27.11 |      24.95 |           23.97 | 26M          |
| PROD (Rest.) + Full           |  27.48 |      25.07 |           24.34 | 26M          |
| CoLoRA w. PROD (Rest.)        |  27.35 |      24.96 |           24.04 | 2.9M         |

| Method                    |   Haze | Method                    |   Blur |
|---------------------------|--------|---------------------------|--------|
| DCP [25]                  |  11.25 | IFI-RNN [56]              |  31.53 |
| GDN [47]                  |  15.19 | ESTRNN [96]               |  31.95 |
| MSBDN [20]                |  13.19 | CDVD-TSP [73]             |  32.16 |
| DeHamer [22]              |  13.31 | STFAN [98]                |  32.19 |
| SMOKE [33]                |  18.83 | PVDNet [74]               |  32.22 |
| NAF. + Full [8]           |  19.27 | NAF. + Full [8]           |  32.3  |
| DegAE (NAF.) + Full [48]  |  20.23 | DegAE (NAF.) + Full [48]  |  32.19 |
| PROD(NAF.) + Full         |  20.33 | PROD (NAF.) + Full        |  32.84 |
| CoLoRA w. PROD(NAF.)      |  20.17 | CoLoRA w. PROD (NAF.)     |  32.08 |
| Rest. + Full [86]         |  19.67 | Rest. + Full [86]         |  30.42 |
| DegAE (Rest.) + Full [48] |  20.28 | DegAE (Rest.) + Full [48] |  31.47 |
| PROD (Rest.) + Full       |  20.16 | PROD (Rest.) + Full       |  32.38 |
| CoLoRA w. PROD (Rest.)    |  20.04 | CoLoRA w. PROD (Rest.)    |  31.9  |

## 4.1 Benchmark Results for Real Various IR Tasks

In Table 1, we summarized the benchmark results on the real Rain, Raindrop, Rain&amp;Raindrop, Haze and Blur datasets to evaluate our proposed CoLoRA with PROD . To make a fair comparison with previous methods, we trained the model using the entire training dataset from the real various IR task datasets. For each task, all methods except DegAE and Our PROD are initialized randomly without a pre-trained model. All methods except Our CoLoRA perform full fine-tuning on new tasks. In full fine-tuning, NAFNet and Restormer have 29 M and 26 M (100%) trainable parameters, respectively. The CoLoRA method have 2M and 2.9 M (7% and 11% of the total) trainable parameter in NAFNet and Retormer, respectively. Our PROD achieved state-of-the-art PSNR on datasets containing Real Rain, Raindrop, Rain&amp;Raindrop, Haze and Blur. Our CoLoRA

Fig. 4: Performance comparison based on the scale of training data for 6 IR tasks. In the graph, the results of the 6 IR tasks are averaged for comparison. The x-axis represents the number of training data, and the y-axis is the average PSNR. In the radar graph, we compare the results of 6 IR tasks with Normalized PSNR at a training data size of 128. (a) and (b) present experimental results corresponding to pre-training and fine-tuning methods, respectively. (c) and (d) experimental results for the Our CoLoRA with PROD in NAFNet and Restormer. Our proposed CoLoRA (7%) has much fewer tuned network parameters compared to the full fine-tuning (100%) of NAFNet.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-e2b68423-c2c4-463a-9901-2e9380949546.jpg)

with PROD approach achieved similar performance on NAFNet and Retsomer, respectively, by updating only small 2M and 2.9M (7% and 11% of the total) network parameters compared to full fine-tuning 29M and 26M (100%). These results demonstrate that our proposed PROD and CoLoRA with PROD methods effectively and efficiently improve various IR performances, with abundant data.

## 4.2 Comparing Proposed Methods in Limited Real-world Scenario

In Fig. 4, we conducted a comparative study to demonstrate the performance and efficiency of the parameter-efficient fine-tuning method, CoLoRA, and the proposed pre-training method PROD, depending on the number of training data. To summarize each experiment, (a) 'Pre-training Methods' presents the experimental results based on different pre-training methods, (b) 'Tuning Strategies' presents the results based on efficient parameter tuning methods, and (c)&amp;(d) 'Different Network Architectures' provides the experimental results based on pre-training and tuning methods on CNN and vision transformer architectures.

Fig. 4 summarizes PSNR results for six real-world IR tasks with respect to the number of training data during fine-tuning. The number of training data for each tasks were set to [16, 32, 64, 128]. These small datasets reflect the reality of IR tasks: acquiring paired real world datasets is often difficult and expensive (e.g., weather, medical imaging). In Fig. 4 (line graph), the y-axis represents the average PSNR results of six IR tasks, while the x-axis represents the number of training data during fine-tuning. Each vertical line in the line graph represents results from experiments fine-tuning a total of 6 IR tasks with 4 different amounts of training data. Fig. 4 (radar graph) represents the normalized PSNR results when number of training data is 128 for six IR tasks with real test data.

Comparison of pre-training methods. To validate the effectiveness of our proposed PROD method, we conduct a comparative study based on pre-training methods, using the CNN-based NAFNet, as illustrated in Fig. 4 (a). Random Init refers to the method of randomly initializing the network parameters. Single denotes methods like IPT [7] and EDT [43], where various synthetic degradations are randomly applied once. Fixed represents methods like DegAE [48], where diverse synthetic degradations are applied in a fixed order. PROD (Ours) is a method that involves randomly applying various synthetic degradations multiple times to generate low-quality images. We used full fine-tuning to evaluate the performance of the pre-trained model. Our PROD yielded substantially higher performance than other methods. Using 32 training data, our PROD outperformed Random Init and Fixed , which utilized 128 training data, by a significant improvement of 0.05 dB and 0.10 dB, respectively. Our PROD method effectively improves performance with limited training data.

Comparison of fine-tuning strategies. In the Fig. 4 (b), we validated our proposed CoLoRA strategy by comparing different tuning methods for a new task using CNN-based NAFNet and the PROD pre-trained weight. We investigated the following tuning methods : Full fine-tuning ( Full ), Only tunes only the decoder ( Only Decoder ), apply LoRA to all layers ( LoRA ), and our CoLoRA . The Full method has 29 M trainable parameters. The Only Decoder , LoRA , and CoLoRA (Ours) have approximately 2 M trainable parameters. Using 64 training data, our CoLoRA outperformed Only Deconv and LoRA , which utilized 128 training data, by a significant improvement of 0.08 dB and 0.15 dB, respectively. Our CoLoRA method enhances performance efficiently, even with limited training data and a sparse number of learnable parameters.

Comparison of different network architectures. In Fig.4 (c) and (d), we evaluate the efficiency of the proposed CoLoRA with PROD using the CNNbased NAFNet and Transformer-based Restormer network architectures. We investigated the following pre-train and tuning methods: Random initialization with full fine-tuning ( Random-init+Full ), DegAE [48] with full fine-tuning ( DegAE+Full ), the proposed full fine-tuning with PROD( PROD+Full ), and the proposed CoLoRA with PROD method ( CoLoRA with PROD ). Full fine-tuning has 29M (NAFNet) and 26M (Restormer) trainable parameters, while CoLoRA has 2M (NAFNet) and 2.9M (Restormer). Our proposed methods ( PROD+Full and CoLoRA with PROD ) outperforms than previous methods in 6 IR tasks

Fig. 5: Qualitative results evaluated on the 6 IR tasks for our proposed method, generic Random initial + Full tuning and DegAE + Full tuning. Our methods with partial and full tuning yielded visually excellent results for the real IR task, outperforming others.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-0d3cfa67-594b-4de0-9620-b9a807641ec7.jpg)

for both NAFNet and Restormer. Using 64 training data, our proposed CoLoRA with PROD outperformed DegAE [48] and Random Init which utilized 128 training data, by a significant improvement of 0.22 dB and 0.36 dB in NAFNet, respectively. Furthermore, in Restormer, our proposed CoLoRA with PROD showed superiority over the Full with DegAE [48] in most IR tasks. Our CoLoRA with PROD improves performance efficiently with limited training data and fewer learnable parameters. Fig. 5 illustrates the results for 6 IR task with test data according to previous methods and our proposed methods, with the experiments conducted using the NAFNet. Our methods seems to yield better restoration results than Full with Random and Full with DegAE [48].

Additional experiments on low-light and underwater tasks. We have further evaluated on more datasets, underwater (LSUI [60] dataset) and lowlight enhancement (LoL [79] dataset) tasks. All experiments were performed with NAFNet. The numerical results for each task are as follows: (1) for LSUI data, the PSNR (dB) values of {NAFNet(Random), DegAE, PROD, CoLoRA with PROD} are {22.29, 22.69, 23.32, 22.98}, respectively. (2) for LoL data, the PSNR (dB) values of {NAFNet(Random), DegAE, PROD, CoLoRA with PROD} are {22.79, 22.77, 22.96, 22.94}, respectively. Our methods achieved high performance, demonstrating the generality.

Table 2: Ablation study according to CoLoRA scale parameters α and β for Blur task. We conducted the experiment using NAFNet and used all training data.

| α / β        | 1.0 / 1.0   | 1.0 / 0.2   | 1.0 / 0.1   | 0.5 / 0.2   | 0.25 /   | 0.1 / 0.1   |    |
|--------------|-------------|-------------|-------------|-------------|----------|-------------|----|
| PSNR (dB)    | 32.26       | 32.08       | 31.98       | 31.90       | 31.86    | 31.54       |    |
| Tuned param. | 5.2 M       | 1.9 M       | 1.6 M       | 1.6 M       | 1.4 M    | 0.8 M       |    |

Table 3: We evaluate the performance comparison in terms of PSNR (dB) using the pre-trained model without any fine-tuning on 6 real IR tasks.

| Method   |   Rain |   Raindrop |   Rain&Raindrop | Noise Blur   |   Haze |   Avg. |
|----------|--------|------------|-----------------|--------------|--------|--------|
| Fixed    |  21.76 |      18.81 |           17.42 | 23.68 27.49  |  13.06 |  20.37 |
| Single   |  22.11 |      18.77 |           17.44 | 23.84 27.47  |  13.03 |  20.44 |
| PROD     |  24.11 |      18.85 |           18.14 | 25.10 27.75  |  13.06 |  21.17 |

## 4.3 Ablation Study On Scale Values ( α and β ) of CoLoRA

In Table 2, we investigated the PSNR results based on CoLoRA's scaling factors ( α β , ) for NAFNet and Restormer, to optimize network parameters. Selecting α and β depends on the trade-off between performance and memory. We found that using small values for β (0.1-0.2) was efficient while using large values for α (0.71.0) was effective. We selected α = 1 , β = 0 2 . for NAFNet and α = 0 75 . , β = 0 1 . for Restormer for the highest performance efficiently. With these criteria, α and β can be practically chosen depending on the target device and the applied task.

## 5 Discussion

Empirical analysis on pre-train methods without any fine-tuning: To observe the scalability and generalization capabilities of our pre-training method, PROD, we conduct an empirical analysis on 6 real IR tasks without any finetuning. The investigated pre-training methods consist of untrained Fixed, Single, DeGAE, and PROD. Table 3 summarizes the PSNR for 6 real IR tasks based on pre-training methods without any fine-tuning. Our proposed PROD achieves better results even when not trained for degradation, surpassing previous approaches. Fig. 6 (a) illustrates the degradation representations of various IR tasks using t-SNE, based on pre-training methods without fine-tuning. Features are extracted from the last layer of the middle block of the pre-trained NAFNet for t-SNE plotting, followed by global average pooling, and t-SNE was plotted for 100 samples. We believe that the embedded scalability and generalization ability of PROD shows promising performance on the above IR results.

Why CoLoRA, practical deployment: Considering that IR can be executed immediately after image acquisition and used for on-device AI, the efficiency, robustness, and scalability of CoLoRA are very important. Compared to fulltuning, CoLoRA excels in efficiency, robustness, and scalability. For six tasks, CoLoRA has fewer parameters than Full and LoRA (41M for CoLoRA, 41M

Fig. 6: (a) Visualizing degradation representations for 6 real IR tasks using t-SNE without fine-tuning, our PROD method generates more discriminative clusters for untrained data than other pre-training methods. (b) Average FAIG value to compare the similarity between LoRA / CoLoRA and Full-tuning (the lower, the more similar).

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-34086b59-ded4-4719-939d-5280530426b5.jpg)

for LoRA, 174M for Full-tuning), and requires less memory to store (167MB for CoLoRA, 176MB for LoRA, 696MB for Full-tuning), providing significant advantages to on-device AI applications with limited memory [6]. In scenarios with increasing learning rates (2e-4 to 8e-3), CoLoRA demonstrates robust performance over Full-tuning (CoLoRA 25.3 to 22.3dB, Full-tuning 25.3 to 20.8dB). Additionally, CoLoRA is easier to scale to new tasks compared to all-in-one IR methods. CoLoRA can be used in a variety of industries with AI edge devices. Empirical Analysis of CoLoRA and LoRA: In Fig. 6(b), we measured the FAIG value to analyze the correlation between Full Fine-tuning and CoLoRA/LoRA. Note that lower FAIG score in Eq.(1) indicates more similarity between two networks, reflecting smaller differences and gradients. CoLoRA has a lower FAIG value compared to LoRA, which explains its higher similarity to full-fine tuning and superior performance.

Limitation: While our CoLoRA has been successfully validated for various downstream IR tasks, our CoLoRA faces the limitation of increasing parameters as the tasks expand even though they are still more efficient than full-tuning.

## 6 Conclusion

We proposed CoLoRA with PROD, pre-training with synthetic data and efficient parameter tuning in both abundant and limited real-data scenarios for image restoration. Our PROD yielded excellent pre-trained model as compared to prior arts and our CoLoRA enabled efficient fine-tuning only about 7% learnable parameters. We demonstrate that our proposed method is flexible enough to work with diverse network architectures and achieves state-of-the-art performance on various IR tasks with real-world datasets.

## Acknowledgements

This work was supported by the New Faculty Startup Fund from Seoul National University, Institute of Information &amp; communications Technology Planning &amp; Evaluation (IITP) grant funded by the Korea government(MSIT) [NO.RS-2021II211343, Artificial Intelligence Graduate School Program (Seoul National University)], the National Research Foundation of Korea(NRF) grant funded by the Korea government(MSIT) (No. NRF-2022R1A4A1030579), and CreativePioneering Researchers Program through Seoul National University. Also, the authors acknowledged the financial support from the BK21 FOUR program of the Education and Research Program for Future ICT Pioneers, Seoul National University.

## References

- 1. Abdelhamed, A., Lin, S., Brown, M.S.: A high-quality denoising dataset for smartphone cameras. In: CVPR (2018)
- 3. Bojarski, M., Del Testa, D., Dworakowski, D., Firner, B., Flepp, B., Goyal, P., Jackel, L.D., Monfort, M., Muller, U., Zhang, J., et al.: End to end learning for self-driving cars. NIPS Deep Learning Symposium (2016)
- 2. Agustsson, E., Timofte, R.: Ntire 2017 challenge on single image super-resolution: Dataset and study. In: CVPRW (2017)
- 4. Brown, T., Mann, B., Ryder, N., Subbiah, M., Kaplan, J.D., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., et al.: Language models are fewshot learners. NeurIPS (2020)
- 6. Cai, H., Gan, C., Zhu, L., Han, S.: Tinytl: Reduce memory, not parameters for efficient on-device learning. NeurIPS (2020)
- 5. Bulo, S.R., Porzi, L., Kontschieder, P.: In-place activated batchnorm for memoryoptimized training of dnns. In: CVPR. pp. 5639-5647 (2018)
- 7. Chen, H., Wang, Y., Guo, T., Xu, C., Deng, Y., Liu, Z., Ma, S., Xu, C., Xu, C., Gao, W.: Pre-trained image processing transformer. In: CVPR (2021)
- 9. Chen, T., Kornblith, S., Norouzi, M., Hinton, G.: A simple framework for contrastive learning of visual representations. In: ICML (2020)
- 8. Chen, L., Chu, X., Zhang, X., Sun, J.: Simple baselines for image restoration. ECCV (2022)
- 10. Chen, W.T., Fang, H.Y., Ding, J.J., Tsai, C.C., Kuo, S.Y.: Jstasr: Joint size and transparency-aware snow removal algorithm based on modified partial convolution and veiling effect removal. In: ECCV (2020)
- 12. Chen, W.T., Huang, Z.K., Tsai, C.C., Yang, H.H., Ding, J.J., Kuo, S.Y.: Learning multiple adverse weather removal via two-stage knowledge learning and multicontrastive regularization: Toward a unified model. In: CVPR (2022)
- 11. Chen, W.T., Fang, H.Y., Hsieh, C.L., Tsai, C.C., Chen, I., Ding, J.J., Kuo, S.Y., et al.: All snow removed: Single image desnowing algorithm using hierarchical dual-tree complex wavelet representation and contradict channel loss. In: ICCV (2021)
- 13. Chen, X., Wang, X., Zhou, J., Qiao, Y., Dong, C.: Activating more pixels in image super-resolution transformer. In: CVPR (2023)
- 14. Chu, X., Chen, L., Chen, C., Lu, X.: Improving image restoration by revisiting global information aggregation. ECCV (2022)

- 15. Cubuk, E.D., Zoph, B., Shlens, J., Le, Q.: Randaugment: Practical automated data augmentation with a reduced search space. In: NeurIPS. pp. 18613-18624 (2020)
- 17. Devlin, J., Chang, M.W., Lee, K., Toutanova, K.: Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv:1810.04805 (2018)
- 16. Deng, S., Wei, M., Wang, J., Feng, Y., Liang, L., Xie, H., Wang, F.L., Wang, M.: Detail-recovery image deraining via context aggregation networks. In: CVPR (2020)
- 18. Ding, N., Qin, Y., Yang, G., Wei, F., Yang, Z., Su, Y., Hu, S., Chen, Y., Chan, C.M., Chen, W., et al.: Parameter-efficient fine-tuning of large-scale pre-trained language models. Nature Machine Intelligence (2023)
- 20. Dong, H., Pan, J., Xiang, L., Hu, Z., Zhang, X., Wang, F., Yang, M.H.: Multi-scale boosted dehazing network with dense feature fusion. In: CVPR (2020)
- 19. Doersch, C., Gupta, A., Efros, A.A.: Unsupervised visual representation learning by context prediction. In: ICCV (2015)
- 21. Gidaris, S., Singh, P., Komodakis, N.: Unsupervised representation learning by predicting image rotations. arXiv:1803.07728 (2018)
- 23. He, K., Chen, X., Xie, S., Li, Y., Dollár, P., Girshick, R.: Masked autoencoders are scalable vision learners. In: CVPR (2022)
- 22. Guo, C.L., Yan, Q., Anwar, S., Cong, R., Ren, W., Li, C.: Image dehazing transformer with transmission-aware 3d position embedding. In: CVPR (2022)
- 24. He, K., Fan, H., Wu, Y., Xie, S., Girshick, R.: Momentum contrast for unsupervised visual representation learning. In: CVPR (2020)
- 26. He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: CVPR (2016)
- 25. He, K., Sun, J., Tang, X.: Single image haze removal using dark channel prior. IEEE TPMAI (2010)
- 27. Houlsby, N., Giurgiu, A., Jastrzebski, S., Morrone, B., De Laroussilhe, Q., Gesmundo, A., Attariyan, M., Gelly, S.: Parameter-efficient transfer learning for nlp. In: ICML (2019)
- 29. Huang, T., Li, S., Jia, X., Lu, H., Liu, J.: Neighbor2neighbor: Self-supervised denoising from single noisy images. In: CVPR (2021)
- 28. Hu, E.J., Shen, Y., Wallis, P., Allen-Zhu, Z., Li, Y., Wang, S., Wang, L., Chen, W.: Lora: Low-rank adaptation of large language models. arXiv:2106.09685 (2021)
- 30. Jaw, D.W., Huang, S.C., Kuo, S.Y.: Desnowgan: An efficient single image snow removal framework using cross-resolution lateral connection and gans. IEEE Transactions on Circuits and Systems for Video Technology (2020)
- 32. Jiang, H., He, P., Chen, W., Liu, X., Gao, J., Zhao, T.: Smart: Robust and efficient fine-tuning for pre-trained natural language models through principled regularized optimization. arXiv:1911.03437 (2019)
- 31. Jia, M., Tang, L., Chen, B.C., Cardie, C., Belongie, S., Hariharan, B., Lim, S.N.: Visual prompt tuning. In: ECCV. pp. 709-727 (2022)
- 33. Jin, Y., Yan, W., Yang, W., Tan, R.T.: Structure representation network and uncertainty feedback learning for dense non-uniform fog removal. In: ACCV (2022)
- 35. Krizhevsky, A., Sutskever, I., Hinton, G.E.: Imagenet classification with deep convolutional neural networks. NeurIPS (2012)
- 34. Khattak, M.U., Rasheed, H., Maaz, M., Khan, S., Khan, F.S.: Maple: Multi-modal prompt learning. In: CVPR. pp. 19113-19122 (2023)
- 36. Kupyn, O., Budzan, V., Mykhailych, M., Mishkin, D., Matas, J.: Deblurgan: Blind motion deblurring using conditional adversarial networks. In: CVPR (2018)

- 37. Lai, W.S., Huang, J.B., Hu, Z., Ahuja, N., Yang, M.H.: A comparative study for single image blind deblurring. In: CVPR (2016)
- 39. Lehtinen, J., Munkberg, J., Hasselgren, J., Laine, S., Karras, T., Aittala, M., Aila, T.: Noise2noise: Learning image restoration without clean data. ICML (2018)
- 38. Ledig, C., Theis, L., Huszár, F., Caballero, J., Cunningham, A., Acosta, A., Aitken, A., Tejani, A., Totz, J., Wang, Z., et al.: Photo-realistic single image super-resolution using a generative adversarial network. In: CVPR (2017)
- 40. Li, B., Peng, X., Wang, Z., Xu, J., Feng, D.: Aod-net: All-in-one dehazing network. In: ICCV (2017)
- 42. Li, R., Tan, R.T., Cheong, L.F.: All in one bad weather removal using architectural search. In: CVPR (2020)
- 41. Li, B., Liu, X., Hu, P., Wu, Z., Lv, J., Peng, X.: All-in-one image restoration for unknown corruption. In: CVPR (2022)
- 43. Li, W., Lu, X., Lu, J., Zhang, X., Jia, J.: On efficient transformer and image pre-training for low-level vision. arXiv:2112.10175 (2021)
- 45. Lim, B., Son, S., Kim, H., Nah, S., Lee, K.M.: Enhanced deep residual networks for single image super-resolution. In: CVPRW (2017)
- 44. Li, X.L., Liang, P.: Prefix-tuning: Optimizing continuous prompts for generation. ACL (2021)
- 46. Liu, L., Xie, L., Zhang, X., Yuan, S., Chen, X., Zhou, W., Li, H., Tian, Q.: Tape: Task-agnostic prior embedding for image restoration. In: ECCV (2022)
- 48. Liu, Y., He, J., Gu, J., Kong, X., Qiao, Y., Dong, C.: Degae: A new pretraining paradigm for low-level vision. In: CVPR (2023)
- 47. Liu, X., Ma, Y., Shi, Z., Chen, J.: Griddehazenet: Attention-based multi-scale network for image dehazing. In: ICCV (2019)
- 49. Liu, Y.F., Jaw, D.W., Huang, S.C., Hwang, J.N.: Desnownet: Context-aware deep network for snow removal. IEEE TIP (2018)
- 51. Luo, C., Yang, X., Yuille, A.: Self-supervised pillar motion learning for autonomous driving. In: CVPR (2021)
- 50. Loshchilov, I., Hutter, F.: Decoupled weight decay regularization. arXiv:1711.05101 (2017)
- 52. Luo, S., Tan, Y., Patil, S., Gu, D., von Platen, P., Passos, A., Huang, L., Li, J., Zhao, H.: Lcm-lora: A universal stable-diffusion acceleration module. arXiv:2311.05556 (2023)
- 54. Mou, C., Wang, Q., Zhang, J.: Deep generalized unfolding networks for image restoration. In: CVPR (2022)
- 53. Luo, Z., Gustafsson, F.K., Zhao, Z., Sjölund, J., Schön, T.B.: Controlling visionlanguage models for universal image restoration. arXiv:2310.01018 (2023)
- 55. Nah, S., Hyun Kim, T., Mu Lee, K.: Deep multi-scale convolutional neural network for dynamic scene deblurring. In: CVPR (2017)
- 57. Park, D., Lee, B.H., Chun, S.Y.: All-in-one image restoration for unknown degradations using adaptive discriminative filters for specific degradations. In: CVPR (2023)
- 56. Nah, S., Son, S., Lee, K.M.: Recurrent neural networks with intra-frame iterations for video deblurring. In: CVPR (2019)
- 58. Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., et al.: Pytorch: An imperative style, highperformance deep learning library. NeurIPS (2019)
- 59. Pathak, D., Girshick, R., Dollár, P., Darrell, T., Hariharan, B.: Learning features by watching objects move. In: CVPR (2017)

- 60. Peng, L., Zhu, C., Bian, L.: U-shape transformer for underwater image enhancement. IEEE TIP (2023)
- 62. Pfeiffer, J., Rücklé, A., Poth, C., Kamath, A., Vulić, I., Ruder, S., Cho, K., Gurevych, I.: Adapterhub: A framework for adapting transformers. EMNLP (2020)
- 61. Pfeiffer, J., Kamath, A., Rücklé, A., Cho, K., Gurevych, I.: Adapterfusion: Nondestructive task composition for transfer learning. arXiv:2005.00247 (2020)
- 63. Podell, D., English, Z., Lacey, K., Blattmann, A., Dockhorn, T., Müller, J., Penna, J., Rombach, R.: Sdxl: Improving latent diffusion models for high-resolution image synthesis. arXiv:2307.01952 (2023)
- 65. Quan, R., Yu, X., Liang, Y., Yang, Y.: Removing raindrops and rain streaks in one go. In: CVPR (2021)
- 64. Potlapalli, V., Zamir, S.W., Khan, S., Khan, F.: Promptir: Prompting for all-inone image restoration. In: NeurIPS (2023)
- 66. Radford, A., Wu, J., Child, R., Luan, D., Amodei, D., Sutskever, I., et al.: Language models are unsupervised multitask learners. OpenAI blog (2019)
- 68. Ren, D., Zuo, W., Hu, Q., Zhu, P., Meng, D.: Progressive image deraining networks: A better and simpler baseline. In: CVPR (2019)
- 67. Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: Unified, real-time object detection. In: CVPR (2016)
- 69. Ren, S., He, K., Girshick, R., Sun, J.: Faster r-cnn: Towards real-time object detection with region proposal networks. NeurIPS (2015)
- 71. Rim, J., Lee, H., Won, J., Cho, S.: Real-world blur dataset for learning and benchmarking deblurring algorithms. In: ECCV (2020)
- 70. Rim, J., Kim, G., Kim, J., Lee, J., Lee, S., Cho, S.: Realistic blur synthesis for learning image deblurring. ECCV (2022)
- 72. Rombach, R., Blattmann, A., Lorenz, D., Esser, P., Ommer, B.: High-resolution image synthesis with latent diffusion models. In: CVPR (2022)
- video deblurring with non-consecutively blurry frames. In: ICCV (2021)
- 73. Shang, W., Ren, D., Zou, D., Ren, J.S., Luo, P., Zuo, W.: Bringing events into
- 74. Son, H., Lee, J., Lee, J., Cho, S., Lee, S.: Recurrent video deblurring with blurinvariant motion estimation and pixel volumes. ACM TOG (2021)
- 76. Valanarasu, J.M.J., Yasarla, R., Patel, V.M.: Transweather: Transformer-based restoration of images degraded by adverse weather conditions. In: CVPR (2022)
- 75. Tu, Z., Talebi, H., Zhang, H., Yang, F., Milanfar, P., Bovik, A., Li, Y.: Maxim: Multi-axis mlp for image processing. In: CVPR (2022)
- 77. Wang, T., Yang, X., Xu, K., Chen, S., Zhang, Q., Lau, R.W.: Spatial attentive single-image deraining with a high quality real rain dataset. In: CVPR (2019)
- 79. Wei, C., Wang, W., Yang, W., Liu, J.: Deep retinex decomposition for low-light enhancement. arXiv preprint arXiv:1808.04560 (2018)
- 78. Wang, Z., Cun, X., Bao, J., Zhou, W., Liu, J., Li, H.: Uformer: A general u-shaped transformer for image restoration. In: CVPR (2022)
- 80. Whang, J., Delbracio, M., Talebi, H., Saharia, C., Dimakis, A.G., Milanfar, P.: Deblurring via stochastic refinement. In: CVPR (2022)
- 82. Xie, L., Wang, X., Dong, C., Qi, Z., Shan, Y.: Finding discriminative filters for specific degradations in blind super-resolution. NeurIPS (2021)
- 81. Wu, H., Qu, Y., Lin, S., Zhou, J., Qiao, R., Zhang, Z., Xie, Y., Ma, L.: Contrastive learning for compact single image dehazing. In: CVPR (2021)
- 83. Xie, Z., Zhang, Z., Cao, Y., Lin, Y., Bao, J., Yao, Z., Dai, Q., Hu, H.: Simmim: A simple framework for masked image modeling. In: CVPR (2022)

- 84. Yang, Y., Kim, K.S., Kim, M., Park, J.: Gram: Fast fine-tuning of pre-trained language models for content-based collaborative filtering. arXiv:2204.04179 (2022)
- 86. Zamir, S.W., Arora, A., Khan, S., Hayat, M., Khan, F.S., Yang, M.H.: Restormer: Efficient transformer for high-resolution image restoration. In: CVPR (2022)
- 85. Zaken, E.B., Ravfogel, S., Goldberg, Y.: Bitfit: Simple parameter-efficient finetuning for transformer-based masked language-models. ACL (2021)
- 87. Zamir, S.W., Arora, A., Khan, S., Hayat, M., Khan, F.S., Yang, M.H., Shao, L.: Multi-stage progressive image restoration. In: CVPR (2021)
- 89. Zhang, J.O., Sax, A., Zamir, A., Guibas, L., Malik, J.: Side-tuning: a baseline for network adaptation via additive side networks. In: ECCV (2020)
- 88. Zhang, H., Patel, V.M.: Densely connected pyramid dehazing network. In: CVPR (2018)
- 90. Zhang, J., Huang, J., Yao, M., Yang, Z., Yu, H., Zhou, M., Zhao, F.: Ingredientoriented multi-degradation learning for image restoration. In: CVPR (2023)
- 92. Zhang, K., Zuo, W., Chen, Y., Meng, D., Zhang, L.: Beyond a gaussian denoiser: Residual learning of deep cnn for image denoising. IEEE TIP (2017)
- 91. Zhang, K., Liang, J., Van Gool, L., Timofte, R.: Designing a practical degradation model for deep blind image super-resolution. In: ICCV (2021)
- 93. Zhang, K., Zuo, W., Zhang, L.: Ffdnet: Toward a fast and flexible solution for cnn-based image denoising. IEEE TIP (2018)
- 95. Zheng, Z., Ren, W., Cao, X., Hu, X., Wang, T., Song, F., Jia, X.: Ultra-highdefinition image dehazing via multi-guided bilateral learning. In: CVPR (2021)
- 94. Zhang, L., Rao, A., Agrawala, M.: Adding conditional control to text-to-image diffusion models. In: ICCV (2023)
- 96. Zhong, Z., Gao, Y., Zheng, Y., Zheng, B.: Efficient spatio-temporal recurrent neural network for video deblurring. In: ECCV (2020)
- 98. Zhou, S., Zhang, J., Pan, J., Xie, H., Zuo, W., Ren, J.: Spatio-temporal filter adaptive network for video deblurring. In: ICCV (2019)
- 97. Zhou, K., Yang, J., Loy, C.C., Liu, Z.: Learning to prompt for vision-language models. ICCV (2022)
- 99. Zhu, Y., Wang, T., Fu, X., Yang, X., Guo, X., Dai, J., Qiao, Y., Hu, X.: Learning weather-general and weather-specific features for image restoration under multiple adverse weather conditions. In: CVPR (2023)
- 101. Zhussip, M., Soltanayev, S., Chun, S.Y.: Extending stein's unbiased risk estimator to train deep denoisers with correlated pairs of noisy images. NeurIPS (2019)
- 100. Zhu, Y., Wang, T., Fu, X., Yang, X., Guo, X., Dai, J., Qiao, Y., Hu, X.: Learning weather-general and weather-specific features for image restoration under multiple adverse weather conditions. In: CVPR (2023)
- 102. Zou, Z., Lei, S., Shi, T., Shi, Z., Ye, J.: Deep adversarial decomposition: A unified framework for separating superimposed images. In: CVPR (2020)

## Supplementary Material

## S1 Details on the proposed CoLoRA with PROD

Fig. S1: Illustrations of pre-train methods and tuning strategies for image restoration. From synthetic degradation functions, (a) while prior works either randomly select a single degradation [7,43] or pre-determine multiple degradations in a fixed order [48], (b) our proposed PROD pre-trains with generated low-quality images by random order degradations. (c) While utilizing the full fine-tuning strategy is common among prior works, (d) our CoLoRA enables parameter-efficient fine-tuning by freezing the pretrained model and only adjusting the additional adapter for each task.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-619bd7e7-5a72-487f-8ba0-83aef3ce4800.jpg)

In Figure S1, we compare our CoLoRA - PROD method with existing methods [7, 43, 48]. Figure S1 illustrates the differences between PROD, which introduces degradation during pre-training, and other prior works, as well as the distinctions between CoLoRA, a efficient tuning approach, and simple full finetuning. Specifically, in simple full fine-tuning, the entire network is tuned. In contrast, our CoLoRA approach is a parameter-efficient tuning methodology that flexibly adjusts the ratio of learnable network parameters based on layer positions. While prior works must have a separate full fine-tuned network for each novel IR task, our proposed CoLoRA only need to store the small tunable adaptor for each task (approximately 7% of the entire network parameters for comparable performance to full-size tuning), so that it will be advantageous in terms of memory for multiple target tasks.

## S2 Details on synthetic degradation functions of PROD

The environmental settings for synthetic blur, noise and JPEG functions were similar to DegAE [48]. In the synthetic blur function ( f Blur ), we employ Gaussian

kernels, generalized Gaussian kernels, and plateau-shaped kernels. The kernel size is randomly chosen from the set 7, 9, ..., 21. For the generalized Gaussian and plateau-shaped kernels, the shape parameters are sampled from the intervals [0.5, 4] and [1, 2], respectively. In the synthetic noise function ( f Noise ), we use Gaussian and Poisson, the range for the noise sigma is set to [1, 30], and the Poisson noise scale is set to [0.05, 3]. In the synthetic JPEG function ( f JPEG ), the quality factor is specified within the range [30, 95]. In the synthetic motion blur function ( f Motion ), we use the centered motion kernel proposed by Rim [71]. The kernel size is randomly chosen from the set 5, 7, 9, ..., 31. In the synthetic rain function ( f Rain ), the rain noise is set to a range of [10, 1000], the rain length is set to [10, 90], the alpha value is chosen between [0.3, 1.3], and the rain angle is set to [-80, 80]. Figure S2 illustrates synthetic degraded images ( I syn deg ) using PROD.

## S3 Ablation studies for CoLoRA

In Table. S1, we performed an additional ablation study on bias and normalization for deblurring. 'Bias &amp; Norm' improves performance by 0.08 dB with additionally tuned parameter (0.1M), which is consistent with our FAIG analysis. The level of importance could be different for various tasks.

Table S1: Ablation study according to 'Bias and Norm.' (B&amp;N) and 'Adaptation'(A.) of CoLoRA using NAFNet in Blur task.

| Method       | w/o (Adaptation) w/o   | (Bias&Norm)   | CoLoRA   |
|--------------|------------------------|---------------|----------|
| PSNR         | 30.88 dB               | 32.00 dB      | 32.08 dB |
| Tund. Param. | 0.1M                   | 1.8M          | 1.9M     |

Fig. S2: Example of degraded image generation using PROD with synthetic degradation functions such as Gaussian noise, Gaussian blur, JPEG artifact, and so on.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-d6687392-dcfb-47a5-a1b4-492642892533.jpg)

Table S2: In Real 6 IR tasks, these are the Norm(FAIG) average and standard deviation values according to location in NAFNet and Restormer networks.

| Network            | NAFNet    | NAFNet    | NAFNet    | NAFNet    | NAFNet    | NAFNet    | NAFNet    | NAFNet    | NAFNet     |
|--------------------|-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------|------------|
| Location           | Encoder   | Encoder   | Encoder   | Encoder   | Middle    | Decoder   | Decoder   | Decoder   | Decoder    |
|                    | 1         | 2         | 3         | 4         | Middle    | 1         | 2         | 3         | 4          |
| Norm(FAIG)         | 0.794     | 1.0       | 0.563     | 0.352     | 0.089     | 0.391     | 0.831     | 0.932     | 0.920      |
| Standard deviation | 0.229     | 0.241     | 0.172     | 0.144     | 0.037     | 0.153     | 0.158     | 0.108     | 0.201      |
| Network            | Restormer | Restormer | Restormer | Restormer | Restormer | Restormer | Restormer | Restormer | Restormer  |
| Location           | Encoder   | Encoder   | Encoder   | Middle    | Decoder   | Decoder   | Decoder   |           | Refinement |
|                    | 1         | 2         | 3         | Middle    | 1         | 2         | 3         |           | Refinement |
| Norm(FAIG)         | 0.968     | 1.0       | 0.415     | 0.005     | 0.240     | 0.723     | 0.850     |           | 0.901      |
| Standard deviation | 0.118     | 0.119     | 0.131     | 0.001     | 0.095     | 0.104     | 0.041     |           | 0.150      |

## S4 Details on the δ of the proposed CoLoRA.

The LoRA [28] method uses a fixed value of r for all layers. On the other hand, our proposed CoLoRA adjusts the ratio of learnable parameters ( δ ) by using different values of r depending on the layer location, as shown in Figure S3. The δ in CoLoRA, according to the size of r , is defined as follows: δ i = ( r i ( d i + k i )) / d ( i × k i ) , where i denotes the network layer index, and d and k represent the size of W 0 . The proposed method has designed the size of r to use a small ratio of learnable network parameters (approximately 7%) in the entire network for each task. Thanks to CoLoRA, additional IR operations use only small network parameters, significantly reducing memory usage. Table S2 shows Norm(FAIG) values according to location in NAFNet and Restormer. For real 6 IR tasks, Norm(FAIG) values show the average value and standard deviation. The middle part has a very small value compared to the encoder part and decoder part.

## S5 Details of experiment setups

For a fair comparison, we evaluate the proposed method under the same conditions using PyTorch [58] on an NVIDIA A100 GPU. All our experiments were based on the code published by NAFNet [8] and Restormer [86]. In all our experiments, the channel dimension of NAFNet [8] is set to 32, the channel dimension of Restormer [86] is configured to 48. In NAFNet, the Encoder block is configured as [2,2,4,8], the Middle block has 12 layers, and the Decoder block is structured as [2,2,2,2]. In Restormer, the Encoder block is composed of [4,6,6], with 8 Middle blocks, and the Decoder block is structured as [6,6,4]. We trained the model with the AdamW [50] optimizer ( β 1 = 0.9, β 2 = 0.9, weight decay 1e-3), and the training patch size is set to 256 × 256 . In NAFNet, the batch size is set to 32, and in Restormer, the batch size is set to 4. The initial learning rate starts at 3 e -4 and gradually decreases to 1 e -6 according to the cosine annealing schedule. Data augmentation for training such as random crop, horizontal flip, and 90-degree rotation were used. In the case of DegAE [48], we used

the publicly weight files for Restormer, and for NAFNet, we implemented the method based on the publicly code.

Pre-training: In the pre-training, we generate degraded low-quality image using synthetic degradation functions, a combination of two open datasets, DIV2K [2] and Flickr2K [45].

The number of training iterations for all pre-train models is set to 200K. To train the pre-train model, PSNR loss was used except for the DegAE [48].

Fine-tuning: In Section 4.1 of the paper, fine-tuning is conducted using 10K iterations. In Section 4.2 of the paper, Fine-tuning is performed with a different number of training data, and the number of training data are composed of [16, 32, 64, 128], 100K training iterations are used, and all publicly available training data are used. In fine-tuning, NAFNet uses PSNR Loss, and Restormer uses L1 loss. Due to variations in experimental setups, slight gap in results between the original paper may arise, but this does not impact the validation of our method. We perform experiments in the same environment, including learning rate, decay method, data augmentation and so on, for all image restoration tasks to ensure consistency across all experiments.

## S6 Details of evaluation metrics

To ensure a fair comparison of our proposed method, we evaluated its performance using PSNR and SSIM evaluation metrics. Since evaluation metrics on the Y component in the YUV domain or in the RGB domain tends to exhibit similar trends, we exclusively perform certain measurements, such as Rain, Raindrop, and Rain&amp;Raindrop, on the Y component. Specifically in the case of benchmark, Rain, Raindrop, and Rain&amp;Raindrop, we applied evaluation metrics only to the Y component in the YUV domain. In the benchmark results presented in

Fig. S3: Illustration of our proposed reparametrization method, CoLoRA. The existing LoRA method uses a fixed value of r for all layers since it is challenging to optimize each value at each layer. On the other hand, our proposed CoLoRA adjusts the ratio of learnable parameters ( δ ) by using different values of r layer by layer without much increasing the degree of freedom.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-c9028d0f-b979-4097-94b5-8ac6157e0a92.jpg)

| Number of training data   | Number of training data   | 2     | 2        | 2             | 2     | 2     | 2     | Avg.   | Tuned Parm.   |
|---------------------------|---------------------------|-------|----------|---------------|-------|-------|-------|--------|---------------|
| Pre-training              | Tunning                   | Rain  | Raindrop | Rain&Raindrop | Blur  | Noise | Haze  | Avg.   | Tuned Parm.   |
| Random                    | Full                      | 20.76 | 18.33    | 17.05         | 27.58 | 33.35 | 13.31 | 21.73  |               |
| DegAE [48]                | Full                      | 20.62 | 19.40    | 18.90         | 28.02 | 34.72 | 15.52 | 22.86  | 174 M         |
| PROD                      | Full                      | 21.09 | 19.14    | 18.97         | 28.67 | 35.42 | 16.18 | 23.25  |               |
| PROD                      | CoLoRA                    | 21.77 | 18.78    | 18.70         | 29.26 | 34.39 | 15.10 | 23.00  | 41 M          |

| Number of training data   | Number of training data   | 4     | 4        | 4             | 4     | 4     | 4     | Avg.   | Tuned Parm.   |
|---------------------------|---------------------------|-------|----------|---------------|-------|-------|-------|--------|---------------|
| Pre-training              | Tunning                   | Rain  | Raindrop | Rain&Raindrop | Blur  | Noise | Haze  | Avg.   | Tuned Parm.   |
| Random                    | Full                      | 21.70 | 19.23    | 18.52         | 27.55 | 35.62 | 17.34 | 23.33  |               |
| DegAE [48]                | Full                      | 22.72 | 20.59    | 19.56         | 28.22 | 35.97 | 17.74 | 24.13  | 174 M         |
| PROD                      | Full                      | 23.30 | 20.94    | 20.61         | 29.78 | 36.62 | 17.43 | 24.78  |               |
| PROD                      | CoLoRA                    | 23.25 | 20.84    | 20.66         | 29.70 | 36.71 | 17.32 | 24.74  | 41 M          |

| Number of training data   | Number of training data   | 8     | 8        | 8             | 8     | 8     | 8     | Avg.   | Tuned Parm.   |
|---------------------------|---------------------------|-------|----------|---------------|-------|-------|-------|--------|---------------|
| Pre-training              | Tunning                   | Rain  | Raindrop | Rain&Raindrop | Blur  | Noise | Haze  | Avg.   | Tuned Parm.   |
| Random                    | Full                      | 22.97 | 20.99    | 19.73         | 27.72 | 36.96 | 17.40 | 24.30  |               |
| DegAE [48]                | Full                      | 23.50 | 21.46    | 19.83         | 28.58 | 36.89 | 18.36 | 24.77  | 174 M         |
| PROD                      | Full                      | 24.21 | 21.97    | 21.37         | 30.24 | 37.72 | 18.31 | 25.64  |               |
| PROD                      | CoLoRA                    | 24.34 | 21.91    | 21.20         | 30.26 | 37.70 | 18.15 | 25.49  | 41 M          |

Table S3: We evaluated the performance comparison based on pre-training methods and tuning strategies for 6 IR tasks with real validation data in terms of PSNR (dB) in RGB domain and tuned network parameter size (Million). 'Random' and 'Full' denote randomly initializing the network parameters without a pre-trained model and full fine-tuning, respectively. Our proposed CoLoRA requires a significantly smaller number of network parameters compared to the full tuning method, as it only tunes 7% of the network parameters.

Table 1 of the paper, to ensure a fair comparison with previous existing methods [16,65,68,77], we applied evaluation metrics to the Y component in the YUV domain for Rain, Raindrop, and Rain + Raindrop. Conversely, for Blur, Noise, and Haze, we applied evaluation metrics in the RGB domain. For the ablation study, we measured evaluation metrics in the RGB domain. The ablation study results, presented in Figure 4, Figure S.4, Figure S.5, Figure S.6, and Table S.1 within the paper, entailed applying evaluation metrics in the RGB domain for 6 IR tasks.

Table S4: Ablation study according to CoLoRA scale parameters α and β for Blur task. We conducted the experiment using Restormer and used all training data.

| α               | Full   |     1 |   1.0 |   0.75 |   1.0 |   0.75 |   0.5 |   0.25 |
|-----------------|--------|-------|-------|--------|-------|--------|-------|--------|
| β               | Full   |  1    |  0.2  |   0.2  |  0.1  |    0.1 |   0.1 |   0.1  |
| PSNR (dB)       | 32.38  | 32.14 | 32.01 |  31.88 | 31.94 |   31.9 |  31.8 |  31.82 |
| Tuned parameter | 26.2   |  8.7  |  4    |   3.6  |  3.5  |    2.9 |   2.5 |   1.9  |

## S7 Comparison of pre-train methods and tuning strategies for extremely limited training data

In this section, we evaluated the proposed CoLoRA with PROD on 6 IR datasets extremely limited training data and a training iteration is 10K. Table S3 summarizes the PSNR results for six real IR tasks with extremely small number of training data during fine-tuning in NAFNet. The extremely small number of training data for each tasks were set to [2, 4, 8]. The full fine-tuning method ( Full ) has 29 M trainable parameters in NAFNet [8]. In NAFNet, the full finetuning ( Full ) method requires 6 times more network parameters 174 M ( 29 M × 6 ) than the original network parameters 29 M due to explicitly separated structures for multiple degradations, and our proposed CoLoRA uses an additional 2 M parameters for a single task, resulting in a total of 41 M ( 29 M +2 M × 6 ) parameter for the network required in the 6 IR tasks. These results demonstrate that the our proposed PROD + Full and PROD + CoLoRA method effectively and efficiently improves various IR performances, with extremely limited data.

## S7.1 Ablation study on scales ( α and β ) of CoLoRA (Restormer)

In Table S4, we investigated the PSNR results based on the scaling factors ( α β , ) of CoLoRA to further optimize the tuned network parameters. All experiments are conducted using the transformer-based Restormer with full training data. As the values of α and β increase, the performance improves, and accordingly, the tuned network parameters also increase. We have experimentally selected ( α = 0 75 . , β = 0 1 . ) that efficient and brings the highest performance. Through scale adjustment, it is possible to use the optimal parameters for each task.

## S8 Details on tuning strategies in NAFNet

Table S5 provides detailed tuned parameters for experiments based on tuning strategies. The tuned parameter are in millions (M). The Full method has 29 M trainable parameters. The Only Decoder , LoRA , and CoLoRA (Ours) have approximately 2 M trainable parameters. The LoRA [28] is constructed based on open-source code, with the value of r set to 16 for all layers.

Table S5: In Section 3.1, 'tuning parameters' of this paper refers to the number of tuned parameters in millions (M).

| Tuning strategies     |   Tuned parmeter (M) |
|-----------------------|----------------------|
| Only Deconv LoRA [28] |                1.942 |
|                       |                2.442 |
| CoLoRA (Ours)         |                1.993 |
| Full                  |               29.16  |

## S9 Our proposed PROD vs BSRGAN

Indeed, BSRGAN developed another pre-trained method with multiple degradations only for SR, while our PROD is a method for multiple IR tasks. Nevertheless, we performed experiments using 64 fine-tuning datasets on rain/drop and blur, demonstrating the superiority of our work over BSRGAN (PROD 31.4dB vs BSRGAN 30.8dB for deblur and PROD 22.7dB vs BSRGAN 22.1dB for derain).

## S10 Benchmark results

We evaluated our proposed CoLoRA with PROD on the mixed Snow&amp;Haze [11] benchmark dataset in Table S6. The mixed Snow&amp;Haze [11] benchmark dataset contains 10,000 training images and 3,000 test images. "All-in-One image restoration" may experience a decrease in performance as the number of tasks increases, and there are inherent limitations in extending it to new tasks. On the other hand, "Efficient Fine-tuning" maintains consistent results even as the number of tasks increases, and it is easily scalable to accommodate new tasks. In Park [57], the results are presented for All-in-One image restoration methods when handling only three tasks. Thanks to CoLoRA with PROD, the proposed method in NAFNet architecture provides the highest PSNR compared to existing methods, even though only a small ratio of tuned parameters are learned in the entire network.

Table S6: Benchmark results for mixed Snow&amp;Haze [11] dataset in PSNR (dB). Our CoLoRA with PROD achieved state-of-the art performance on NAFNet, by updating only a small 2M of tuned network parameters.

| Task                  | Method                                                              | PSNR for mixed Snow&Haze   |
|-----------------------|---------------------------------------------------------------------|----------------------------|
| Independent methods   | DesnowNet [49] JSTASR [10] DesnowGAN [30] HDCW-Net [11] DAD-S [102] | 25.63 27.52 28.63 29.11    |
| Independent methods   |                                                                     | 32.28                      |
| Independent methods   |                                                                     | 29.29                      |
| Independent methods   | MPRNet-S [87]                                                       | 31.53                      |
| All-In-One methods    | Chen [12]                                                           |                            |
| All-In-One methods    | Li [41]                                                             | 26.91                      |
| All-In-One methods    | Park [57]                                                           | 32.41                      |
| All-In-One methods    | PromptIR [64]                                                       | 29.81                      |
| All-In-One methods    | WGWS-Net [100]                                                      | 29.16                      |
| Efficient Fine-tuning | Our CoLoRA (NAFNet)                                                 | 33.09                      |
|                       |                                                                     |                            |

## S11 Comparison of tuning strategies in DeGAE

In this section, we evaluated the proposed CoLoRA on 6 IR datasets with DegAE and a training iteration is 10K. Figure S4 summarizes the experimental results based on DegAE with CoLoRA. All experiments are conducted using the CNNbased NAFNet architecture. We investigated the following tuning methods: Full fine-tuning ( Full ), and the proposed CoLoRA method. The Full method has 29 M trainable parameters. CoLoRA (Ours) have approximately 2 M trainable parameters. Our proposed 'CoLoRA + DegAE' method produces results similar to 'Full + DeGAE' while tuning only very small network parameters.

## S12 Details on comparison of pre-train methods and tuning strategies

We conducted a detailed comparative study to demonstrate the performance and efficiency of the proposed pre-training method PROD, and the parameterefficient fine-tuning method, CoLoRA. To summarize each experiment, 'pretraining methods' presents the experimental results based on different pretraining methods in Figure S5 (top), 'tuning strategies' presents the results based on efficient parameter tuning methods in Figure S5 (bottom), and 'different network architectures' provides the experimental results based on pretraining and tuning methods on CNN and vision transformer architectures in Figure S6. Figure S5 and S6 illustrate a detailed summary of the PSNR (dB) results for 6 IR tasks with real data in relation to the number of training data during fine-tuning. The number of training data for each tasks were set to [16, 32, 64, 128]. These results demonstrate that the our proposed PROD + Full and PROD + CoLoRA method effectively and efficiently improves various IR performances, even with limited training data and a sparse number of learnable parameters, underscoring its effectiveness.

Figure S7, Figure S8 and S9 illustrates the results for 6 IR task with test data tasks using test data, comparing outcomes between previous methods and our proposed CoLoRA with PROD. Our methods ( PROD + Full and PROD + CoLoRA ) appear to produce superior restoration results compared to Random + Full and DegAE + Full [48].

## Comparision between DegAE and CoLoRA with Prod

Fig. S4: Performance comparison of applying CoLoRA to DegAE method according to number of training data for 6 IR tasks. In the graph (a), the results of the 6 IR tasks are averaged for comparison. The x-axis represents the number of training data, and the y-axis is the average PSNR in the RGB domain. In the radar graph, we compare the results of 6 IR tasks with Normalized PSNR at a training data size of 128. In (b), the PSNR in the RGB domain is reported for each of the six tasks. The x-axis represents the number of training data, and the y-axis is the PSNR in the RGB domain for each test data.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-a1943cb2-c815-4f8e-a00e-f8bf922d414e.jpg)

Fig. S5: In section 4.1 of the paper, 'pre-training methods' and 'fine-tuning strategies', we conducted a performance comparison based on the scale of training data for 6 IR tasks with real data. The PSNR in the RGB domain is reported for each of the 6 tasks in Figure 5 of the paper. The x-axis represents the number of training data, and the y-axis is the PSNR (dB) for each test data.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-9351c403-4638-4419-bf73-a757e62e0bc9.jpg)

## Pre-train method and tuning strategies CNN based NAFNet

Fig. S6: In section 4.1 of the paper, 'different network architectures', we conducted a performance comparison based on the scale of training data for 6 IR tasks with real data. The PSNR in the RGB domain is reported for each of the six tasks in Figure 5 of the paper. The x-axis represents the number of training data, and the y-axis is the PSNR for each test data.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-6ec366e6-e6dd-45d6-abaa-e7844f24c6da.jpg)

Fig. S7: Qualitative results evaluated on the 6 IR tasks for our proposed CoLoRA with PROD, generic Random initial + Full tuning and DegAE + Full tuning. Our proposed methods with partial and full tuning yielded visually excellent results for the real IR tasks, outperforming others.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-f960f63f-78cb-4401-9c82-cfb3e6e7c50b.jpg)

Fig. S8: Qualitative results evaluated on the 6 IR tasks for our proposed CoLoRA with PROD, generic Random initial + Full tuning and DegAE + Full tuning. Our proposed methods with partial and full tuning yielded visually excellent results for the real IR tasks, outperforming others.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-69874881-6eb6-4949-89b0-de3409124544.jpg)

Fig. S9: Qualitative results evaluated on the 6 IR tasks for our proposed CoLoRA with PROD, generic Random initial + Full tuning and DegAE + Full tuning. Our proposed methods with partial and full tuning yielded visually excellent results for the real IR tasks, outperforming others.

![image](https://ac.marcodsn.me/arxiv-markdown-images/2408.01099v1-figure-50230ad2-7326-45d5-b96f-aafbe6b78652.jpg)
Donwon Park Hayeon Kim Se Young Chun
Contribution-based Low-Rank Adaptation with Pre-training Model for Real Image Restoration
Recently, pre-trained model and efficient parameter tuning have achieved
remarkable success in natural language processing and high-level computer
vision with the aid of masked modeling and prompt tuning. In low-level computer
vision, however, there have been limited investigations on pre-trained models
and even efficient fine-tuning strategy has not yet been explored despite its
importance and benefit in various real-world tasks such as alleviating memory
inflation issue when integrating new tasks on AI edge devices. Here, we propose
a novel efficient parameter tuning approach dubbed contribution-based low-rank
adaptation (CoLoRA) for multiple image restorations along with effective
pre-training method with random order degradations (PROD). Unlike prior arts
that tune all network parameters, our CoLoRA effectively fine-tunes small
amount of parameters by leveraging LoRA (low-rank adaptation) for each new
vision task with our contribution-based method to adaptively determine layer by
layer capacity for that task to yield comparable performance to full tuning.
Furthermore, our PROD strategy allows to extend the capability of pre-trained
models with improved performance as well as robustness to bridge synthetic
pre-training and real-world fine-tuning. Our CoLoRA with PROD has demonstrated
its superior performance in various image restoration tasks across diverse
degradation types on both synthetic and real-world datasets for known and novel
tasks.