Instruct-IPT: All-in-One Image Processing Transformer via Weight Modulation

TL;DR

Instruct-IPT is a versatile image processing transformer that uses weight modulation and task-specific biases to effectively handle diverse low-level vision tasks, including denoising, deblurring, and more, with improved performance and flexibility.

Contribution

The paper introduces a novel weight modulation approach with task-specific biases and low-rank decomposition, enabling a single model to excel across multiple distinct image restoration tasks.

Findings

Effective multi-task performance on various image restoration tasks.

Enhanced cooperation between tasks with distinct characteristics.

Extension of the method to diffusion denoisers.

Abstract

Due to the unaffordable size and intensive computation costs of low-level vision models, All-in-One models that are designed to address a handful of low-level vision tasks simultaneously have been popular. However, existing All-in-One models are limited in terms of the range of tasks and performance. To overcome these limitations, we propose Instruct-IPT -- an All-in-One Image Processing Transformer (IPT) that could effectively address manifold image restoration tasks with large inter-task gaps, such as denoising, deblurring, deraining, dehazing, and desnowing. While most research propose feature adaptation methods, we reveal their failure in addressing highly distinct tasks, and suggest weight modulation that adapts weights to specific tasks. Firstly, we search for task-sensitive weights and introduce task-specific biases on top of them. Secondly, we conduct rank analysis for a good compression strategy and perform low-rank decomposition on the biases. Thirdly, we propose synchronous training that updates the task-general backbone model and the task-specific biases simultaneously. In this way, the model is instructed to learn both general and task-specific knowledge. Via our simple yet effective method that instructs the IPT to be task experts, Instruct-IPT could better cooperate between tasks with distinct characteristics at humble costs. As an additional feature, we enable Instruct-IPT to receive human prompts. We have conducted experiments on Instruct-IPT to demonstrate the effectiveness of our method on manifold tasks, and we have effectively extended our method to diffusion denoisers as well. The code is available at https://github.com/huawei-noah/Pretrained-IPT.