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

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

30 Jun 2024arXiv:2407.00676archive 2025-07-28

Yuchuan Tian, Jianhong Han, Hanting Chen, Yuanyuan Xi, Ning Ding, Jie Hu, Chao Xu, Yunhe Wang

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.

PaperPDFConference PDFCode

Code

huawei-noah/Pretrained-IPT officialmentioned in paperpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

AllDeblurringDenoisingImage RestorationRain Removal

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Color Image Denoising CBSD68 sigma50 Instruct-IPT PSNR 28.61 #6 of 18 Archive leaderboard report
Image Deblurring GoPro Instruct-IPT PSNR 33.86 #11 of 55 Archive leaderboard report
Image Deblurring GoPro Instruct-IPT SSIM 0.967 #11 of 55 Archive leaderboard report
Image Dehazing SOTS Outdoor Instruct-IPT PSNR 39.95 #4 of 31 Archive leaderboard report
Image Dehazing SOTS Outdoor Instruct-IPT SSIM 0.992 #4 of 31 Archive leaderboard report
Single Image Deraining Rain100L Instruct-IPT PSNR 39.35 #5 of 19 Archive leaderboard report
Single Image Deraining Rain100L Instruct-IPT SSIM 0.977 #5 of 19 Archive leaderboard report
Single Image Desnowing CSD Instruct-IPT Average PSNR (dB) 40.12 #1 of 6 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Absolute Position EncodingsAdamAttentionBPEDense ConnectionsDiffusionDropoutLabel SmoothingLayer NormalizationLinear LayerMulti-Head AttentionPosition-Wise Feed-Forward LayerResidual ConnectionSoftmaxTransformer

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections