Towards Effective Multiple-in-One Image Restoration: A Sequential and Prompt Learning Strategy
CoRR(2024)
Abstract
While single task image restoration (IR) has achieved significant successes,
it remains a challenging issue to train a single model which can tackle
multiple IR tasks. In this work, we investigate in-depth the multiple-in-one
(MiO) IR problem, which comprises seven popular IR tasks. We point out that MiO
IR faces two pivotal challenges: the optimization of diverse objectives and the
adaptation to multiple tasks. To tackle these challenges, we present two simple
yet effective strategies. The first strategy, referred to as sequential
learning, attempts to address how to optimize the diverse objectives, which
guides the network to incrementally learn individual IR tasks in a sequential
manner rather than mixing them together. The second strategy, i.e., prompt
learning, attempts to address how to adapt to the different IR tasks, which
assists the network to understand the specific task and improves the
generalization ability. By evaluating on 19 test sets, we demonstrate that the
sequential and prompt learning strategies can significantly enhance the MiO
performance of commonly used CNN and Transformer backbones. Our experiments
also reveal that the two strategies can supplement each other to learn better
degradation representations and enhance the model robustness. It is expected
that our proposed MiO IR formulation and strategies could facilitate the
research on how to train IR models with higher generalization capabilities.
MoreTranslated text
AI Read Science
Must-Reading Tree
Example
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
Generate MRT to find the research sequence of this paper
Chat Paper
Summary is being generated by the instructions you defined